Port docs-mcp-template upgrades (eval, citations, chunking, prefixes) (#13)

Co-authored-by: claude <[email protected]>
2026-09-30 08:33:16 -04:00
parent 6897fbb19a
commit 169816d98d
13 changed files with 812 additions and 77 deletions
@@ -51,7 +51,9 @@ class DenseRetriever:
        self.pool = pool

    def retrieve(self, query: str, k: int = 10) -> list[tuple[str, str]]:
-        res = self.col.query(query_texts=[query], n_results=self.pool)
+        from rag.embeddings import EMBED_QUERY_PREFIX, embed_texts
+        qvec = embed_texts([query], prefix=EMBED_QUERY_PREFIX)[0]
+        res = self.col.query(query_embeddings=[qvec], n_results=self.pool)
        ids = (res.get("ids") or [[]])[0]
        return _collapse_to_pages(ids, k)