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

Co-authored-by: claude <[email protected]>
2026-09-29 22:27:16 -04:00
parent 1a3963f306
commit 92219878d9
13 changed files with 803 additions and 74 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)