api-server/test/mocks/qdrant.mock.ts

41 lines
968 B
TypeScript
Raw Normal View History

export class QdrantClient {
constructor(_opts?: any) {}
getCollection(_name: string) {
return Promise.resolve({
config: { params: { vectors: { size: 1024, distance: 'Cosine' } } },
status: 'green',
})
}
createCollection(_name: string, _opts?: any) {
return Promise.resolve(true)
}
createPayloadIndex(_name: string, _field: string, _opts?: any) {
return Promise.resolve({})
}
upsert(_name: string, _opts?: any) {
return Promise.resolve({ status: 'completed' })
}
delete(_name: string, _opts?: any) {
return Promise.resolve({ status: 'completed' })
}
search(_name: string, opts?: any) {
return Promise.resolve(
Array.from({ length: opts?.limit || 10 }, (_, i) => ({
id: `vec-${i}`,
score: 0.9 - i * 0.05,
payload: { text: `Chunk ${i}`, userId: 'u1', knowledgeBaseId: 'kb1' },
})),
)
}
count(_name: string) {
return Promise.resolve({ count: 0 })
}
}