107 lines
2.5 KiB
TypeScript
107 lines
2.5 KiB
TypeScript
import { invoke } from '@tauri-apps/api/core'
|
|
|
|
export type SearchMode = 'auto' | 'ann' | 'linear'
|
|
|
|
export interface ChunkInput {
|
|
id?: string
|
|
text: string
|
|
embedding: number[]
|
|
metadata?: Record<string, any>
|
|
}
|
|
|
|
export interface SearchResult {
|
|
id: string
|
|
text: string
|
|
score?: number
|
|
file_id: string
|
|
chunk_file_order: number
|
|
}
|
|
|
|
export interface Status {
|
|
ann_available: boolean
|
|
}
|
|
|
|
export interface AttachmentFileInfo {
|
|
id: string
|
|
name?: string
|
|
path?: string
|
|
type?: string
|
|
size?: number
|
|
chunk_count: number
|
|
}
|
|
|
|
// Events
|
|
// Events are not exported in guest-js to keep API minimal
|
|
|
|
export async function getStatus(): Promise<Status> {
|
|
return await invoke('plugin:vector-db|get_status')
|
|
}
|
|
|
|
export async function createCollection(name: string, dimension: number): Promise<void> {
|
|
// Use camelCase param name `dimension` to match Tauri v2 argument keys
|
|
return await invoke('plugin:vector-db|create_collection', { name, dimension })
|
|
}
|
|
|
|
export async function insertChunks(
|
|
collection: string,
|
|
chunks: ChunkInput[]
|
|
): Promise<void> {
|
|
return await invoke('plugin:vector-db|insert_chunks', { collection, chunks })
|
|
}
|
|
|
|
export async function searchCollection(
|
|
collection: string,
|
|
queryEmbedding: number[],
|
|
limit: number,
|
|
threshold: number,
|
|
mode?: SearchMode,
|
|
fileIds?: string[]
|
|
): Promise<SearchResult[]> {
|
|
return await invoke('plugin:vector-db|search_collection', {
|
|
collection,
|
|
queryEmbedding,
|
|
limit,
|
|
threshold,
|
|
mode,
|
|
fileIds,
|
|
})
|
|
}
|
|
|
|
export async function deleteChunks(collection: string, ids: string[]): Promise<void> {
|
|
return await invoke('plugin:vector-db|delete_chunks', { collection, ids })
|
|
}
|
|
|
|
export async function deleteCollection(collection: string): Promise<void> {
|
|
return await invoke('plugin:vector-db|delete_collection', { collection })
|
|
}
|
|
|
|
export async function chunkText(
|
|
text: string,
|
|
chunkSize: number,
|
|
chunkOverlap: number
|
|
): Promise<string[]> {
|
|
// Use snake_case to match Rust command parameter names
|
|
return await invoke('plugin:vector-db|chunk_text', { text, chunkSize, chunkOverlap })
|
|
}
|
|
|
|
export async function listAttachments(
|
|
collection: string,
|
|
limit?: number
|
|
): Promise<AttachmentFileInfo[]> {
|
|
return await invoke('plugin:vector-db|list_attachments', { collection, limit })
|
|
}
|
|
|
|
export async function getChunks(
|
|
collection: string,
|
|
fileId: string,
|
|
startOrder: number,
|
|
endOrder: number
|
|
): Promise<SearchResult[]> {
|
|
return await invoke('plugin:vector-db|get_chunks', {
|
|
collection,
|
|
fileId,
|
|
startOrder,
|
|
endOrder,
|
|
})
|
|
}
|