Skip to content

Commit fe45923

Browse files
committed
feat: add KnowledgeSearchRow interface
1 parent d2c49d9 commit fe45923

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

packages/adapter-pglite/src/index.ts

+11-1
Original file line numberDiff line numberDiff line change
@@ -1349,6 +1349,16 @@ export class PGLiteDatabaseAdapter
13491349
searchText?: string;
13501350
}): Promise<RAGKnowledgeItem[]> {
13511351
return this.withDatabase(async () => {
1352+
interface KnowledgeSearchRow {
1353+
id: UUID;
1354+
agentId: UUID;
1355+
content: string;
1356+
embedding: Buffer | null;
1357+
createdAt: string | number;
1358+
vector_score: number;
1359+
keyword_score: number;
1360+
combined_score: number;
1361+
}
13521362
try {
13531363
const cacheKey = `embedding_${params.agentId}_${params.searchText}`;
13541364
const cachedResult = await this.getCache({
@@ -1400,7 +1410,7 @@ export class PGLiteDatabaseAdapter
14001410
LIMIT $5
14011411
`;
14021412

1403-
const { rows } = await this.query<RAGKnowledgeItem>(sql, [
1413+
const { rows } = await this.query<KnowledgeSearchRow>(sql, [
14041414
vectorStr,
14051415
params.agentId,
14061416
`%${params.searchText || ""}%`,

0 commit comments

Comments
 (0)