xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-26 06:16:09 +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
src/langbot/pkg/vector/vdbs/chroma.py
+5 -5
View File
@@ -79,7 +79,9 @@ class ChromaVectorDatabase(VectorDatabase):
if filter: if filter:
query_kwargs['where'] = filter query_kwargs['where'] = filter
results = await asyncio.to_thread(col.query, **query_kwargs) 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 return results
async def _full_text_search( async def _full_text_search(
@@ -149,9 +151,7 @@ class ChromaVectorDatabase(VectorDatabase):
fused_ids = [doc_id for doc_id, _ in fused] fused_ids = [doc_id for doc_id, _ in fused]
# Fetch full metadata and documents for fused results # Fetch full metadata and documents for fused results
fetched = await asyncio.to_thread( fetched = await asyncio.to_thread(col.get, ids=fused_ids, include=['metadatas', 'documents'])
col.get, ids=fused_ids, include=['metadatas', 'documents']
)
# col.get returns results in arbitrary order; re-order to match fused ranking # col.get returns results in arbitrary order; re-order to match fused ranking
fetched_map: dict[str, tuple] = {} fetched_map: dict[str, tuple] = {}
@@ -187,7 +187,7 @@ class ChromaVectorDatabase(VectorDatabase):
self.ap.logger.info( self.ap.logger.info(
f"Chroma hybrid search in '{collection}' returned {len(ordered_ids)} results " 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 { return {
'ids': [ordered_ids], 'ids': [ordered_ids],