feat: add vector retrieval API support

2026-10-16 13:30:43 +08:00
parent 751667d2c8
commit 90f6685c31
2 changed files with 6 additions and 3 deletions
@@ -16,3 +16,6 @@ export type RetrievalResult={run_id:string;query:string;mode:'keyword';status:'o
 export function retrieveKnowledge(payload:{query:string;mode?:'keyword';top_k?:number;category?:string}){return apiRequest<RetrievalResult>('/api/retrieval',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify(payload)})}
 export function getRetrievalRun(runId:string){return apiRequest<RetrievalResult>(`/api/retrieval/runs/${encodeURIComponent(runId)}`)}
 export function getRetrievalRuns(page=1,pageSize=20){return apiRequest<{items:RetrievalResult[];total:number;page:number;page_size:number}>(`/api/retrieval/runs?page=${page}&page_size=${pageSize}`)}
+
+export type VectorRetrievalResult=Omit<RetrievalResult,'mode'|'results'|'config'> & {mode:'vector';vector_version_id:string;text_index_version_id:string;generation:number;text_generation:number;results:(RetrievalHit & {chunk_id:string;chunk_ordinal:number})[];config:{scorer:'pgvector-cosine-exact-v1';ranking_unit:'chunk';space:{id:string;dimension:number}}}
+export function retrieveVectorKnowledge(payload:{query:string;top_k?:number;category?:string}){return apiRequest<VectorRetrievalResult>('/api/retrieval/vector',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({mode:'vector',top_k:5,...payload})})}