Skip to content

Commit 2917e6b

Browse files
committed
fix: restore engram strict typecheck coverage
1 parent 6107eb7 commit 2917e6b

4 files changed

Lines changed: 8 additions & 5 deletions

File tree

examples/basic.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,3 +37,4 @@ for (const r of memories) {
3737
const stats = await fetch(`${BASE}/stats`);
3838
console.log("\nStats:", await stats.json());
3939

40+
export {};

retrieval/search.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,10 @@ export class SearchEngine {
1010
constructor(private store: StoreBackend) {}
1111

1212
async query(req: SearchRequest): Promise<SearchResult[]> {
13-
log.debug("Searching", { query: req.query.slice(0, 60), topK: req.topK });
13+
const topK = req.topK ?? 5;
14+
log.debug("Searching", { query: req.query.slice(0, 60), topK });
1415

15-
const raw = await this.store.search(req.query, req.topK * 2, {
16+
const raw = await this.store.search(req.query, topK * 2, {
1617
...(req.category ? { category: req.category } : {}),
1718
...(req.agentId ? { agentId: req.agentId } : {}),
1819
...(req.poolAddress ? { poolAddress: req.poolAddress } : {}),
@@ -26,7 +27,7 @@ export class SearchEngine {
2627
// Rerank: blend similarity score with recency
2728
const reranked = rerank(filtered, { recencyWeight: 0.15 });
2829

29-
return reranked.slice(0, req.topK);
30+
return reranked.slice(0, topK);
3031
}
3132

3233
async queryRaw(query: string, topK = 5): Promise<SearchResult[]> {

schemas/index.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ export const SearchRequestSchema = z.object({
4545
debug: z.boolean().default(false),
4646
});
4747

48-
export type SearchRequest = z.infer<typeof SearchRequestSchema>;
48+
export type SearchRequest = z.input<typeof SearchRequestSchema>;
4949

5050
export const SearchResultSchema = z.object({
5151
memory: MemorySchema,

tsconfig.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010
"noUncheckedIndexedAccess": true,
1111
"skipLibCheck": true,
1212
"declaration": true,
13-
"sourceMap": true
13+
"sourceMap": true,
14+
"types": ["node", "bun-types"]
1415
},
1516
"include": ["**/*.ts"],
1617
"exclude": ["node_modules", "dist"]

0 commit comments

Comments
 (0)