xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-02 03:55:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: chroma ruff lint

Browse Source
This commit is contained in:
youhuanghe
2026-03-11 04:07:21 +00:00
parent fe5ce68436
commit bd15b630b0
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/langbot/pkg/vector/vdbs/chroma.py
View File

@@ -79,7 +79,9 @@ class ChromaVectorDatabase(VectorDatabase):
if filter:
query_kwargs['where'] = filter
results = await asyncio.to_thread(col.query, **query_kwargs)
self.ap.logger.info(f"Chroma vector search in '{collection}' returned {len(results.get('ids', [[]])[0])} results.")
self.ap.logger.info(
f"Chroma vector search in '{collection}' returned {len(results.get('ids', [[]])[0])} results."
)
return results
async def _full_text_search(
@@ -149,9 +151,7 @@ class ChromaVectorDatabase(VectorDatabase):
fused_ids = [doc_id for doc_id, _ in fused]
# Fetch full metadata and documents for fused results
fetched = await asyncio.to_thread(
col.get, ids=fused_ids, include=['metadatas', 'documents']
)
fetched = await asyncio.to_thread(col.get, ids=fused_ids, include=['metadatas', 'documents'])
# col.get returns results in arbitrary order; re-order to match fused ranking
fetched_map: dict[str, tuple] = {}
@@ -187,7 +187,7 @@ class ChromaVectorDatabase(VectorDatabase):
self.ap.logger.info(
f"Chroma hybrid search in '{collection}' returned {len(ordered_ids)} results "
f"(vector={len(vector_ids)}, text={len(text_ids)})."
f'(vector={len(vector_ids)}, text={len(text_ids)}).'
)
return {
'ids': [ordered_ids],