justin/seed-mcp
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Port docs-mcp-template upgrades (eval, citations, prefixes) (#26)
Image rebuild (skip scrape) / build (push) Successful in 3m58s
Details

Browse Source 
Co-authored-by: claude <[email protected]>
This commit was merged in pull request #26.
This commit is contained in:
claude
2026-09-30 08:22:55 -04:00
committed by claude
parent 4de5aaa02f
commit 817c0d6c20
12 changed files with 509 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
eval/retrievers.py
+9 -3
View File
@@ -71,8 +71,10 @@ class DenseRetriever:
def retrieve(self, query: str, k: int, filters: dict | None) -> list[str]:
where = _build_where(filters)
try:
from rag.embeddings import EMBED_QUERY_PREFIX, embed_texts
qvec = embed_texts([query], prefix=EMBED_QUERY_PREFIX)[0]
r = self.col.query(
query_texts=[query], n_results=max(k, self.pool), where=where,
query_embeddings=[qvec], n_results=max(k, self.pool), where=where,
)
except Exception:
return []
@@ -105,7 +107,9 @@ class HybridRetriever:
def retrieve(self, query: str, k: int, filters: dict | None) -> list[str]:
where = _build_where(filters)
try:
d = self.col.query(query_texts=[query], n_results=self.pool, where=where)
from rag.embeddings import EMBED_QUERY_PREFIX, embed_texts
qvec = embed_texts([query], prefix=EMBED_QUERY_PREFIX)[0]
d = self.col.query(query_embeddings=[qvec], n_results=self.pool, where=where)
dense_ids = (d.get("ids") or [[]])[0]
except Exception:
dense_ids = []
@@ -138,8 +142,10 @@ class HybridRerankRetriever:
def retrieve(self, query: str, k: int, filters: dict | None) -> list[str]:
where = _build_where(filters)
try:
from rag.embeddings import EMBED_QUERY_PREFIX, embed_texts
qvec = embed_texts([query], prefix=EMBED_QUERY_PREFIX)[0]
d = self.col.query(
query_texts=[query], n_results=self.pool, where=where,
query_embeddings=[qvec], n_results=self.pool, where=where,
include=["documents"],
)
dense_ids = (d.get("ids") or [[]])[0]