102 lines
2.9 KiB
TypeScript
102 lines
2.9 KiB
TypeScript
import { defineStore } from 'pinia'
|
|
import { ref } from 'vue'
|
|
import type { SearchResult, SearchRequest } from '@/contracts'
|
|
import * as searchService from '@/services/searchService'
|
|
import { ApiErrorClass } from '@/services/apiClient'
|
|
|
|
const VECTOR_ERROR_CODES = new Set([
|
|
'VECTOR_UNAVAILABLE', 'EMBEDDING_UNAVAILABLE', 'INDEX_UNAVAILABLE',
|
|
'MODEL_NOT_FOUND', 'MODEL_CAPABILITY_MISMATCH', 'PROVIDER_UNAVAILABLE',
|
|
])
|
|
|
|
export const useSearchStore = defineStore('search', () => {
|
|
const query = ref('')
|
|
const mode = ref<'fts' | 'vector' | 'hybrid'>('hybrid')
|
|
const results = ref<SearchResult[]>([])
|
|
const total = ref(0)
|
|
const isSearching = ref(false)
|
|
const selectedIndex = ref(0)
|
|
const recentQueries = ref<string[]>(['红黑树', '死锁', 'TCP三次握手'])
|
|
const error = ref<string | null>(null)
|
|
const vectorUnavailable = ref(false)
|
|
|
|
async function doSearch(request: SearchRequest) {
|
|
query.value = request.query
|
|
mode.value = request.mode || 'hybrid'
|
|
isSearching.value = true
|
|
error.value = null
|
|
vectorUnavailable.value = false
|
|
|
|
try {
|
|
const resp = await searchService.search(request)
|
|
results.value = resp.results
|
|
total.value = resp.total
|
|
selectedIndex.value = 0
|
|
} catch (reason) {
|
|
const canFallback = mode.value !== 'fts' && reason instanceof ApiErrorClass && VECTOR_ERROR_CODES.has(reason.code)
|
|
if (canFallback) {
|
|
try {
|
|
const fallback = await searchService.search({ ...request, mode: 'fts' })
|
|
results.value = fallback.results
|
|
total.value = fallback.total
|
|
mode.value = 'fts'
|
|
vectorUnavailable.value = true
|
|
selectedIndex.value = 0
|
|
} catch (fallbackError) {
|
|
error.value = fallbackError instanceof Error ? fallbackError.message : '全文检索降级失败'
|
|
results.value = []
|
|
total.value = 0
|
|
}
|
|
} else {
|
|
error.value = reason instanceof Error ? reason.message : '搜索失败'
|
|
results.value = []
|
|
total.value = 0
|
|
}
|
|
} finally {
|
|
isSearching.value = false
|
|
}
|
|
|
|
if (request.query && !recentQueries.value.includes(request.query)) {
|
|
recentQueries.value.unshift(request.query)
|
|
if (recentQueries.value.length > 10) recentQueries.value.pop()
|
|
}
|
|
}
|
|
|
|
function clearResults() {
|
|
results.value = []
|
|
query.value = ''
|
|
total.value = 0
|
|
selectedIndex.value = 0
|
|
error.value = null
|
|
}
|
|
|
|
function selectNext() {
|
|
if (selectedIndex.value < results.value.length - 1) selectedIndex.value++
|
|
}
|
|
|
|
function selectPrev() {
|
|
if (selectedIndex.value > 0) selectedIndex.value--
|
|
}
|
|
|
|
function setMode(m: 'fts' | 'vector' | 'hybrid') {
|
|
mode.value = m
|
|
}
|
|
|
|
return {
|
|
query,
|
|
mode,
|
|
results,
|
|
total,
|
|
isSearching,
|
|
selectedIndex,
|
|
recentQueries,
|
|
error,
|
|
vectorUnavailable,
|
|
doSearch,
|
|
clearResults,
|
|
selectNext,
|
|
selectPrev,
|
|
setMode,
|
|
}
|
|
})
|