topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 03:58:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement](chat) fix filter_condition size ==0 error in python service and update version to '0.8.0-SNAPSHOT' (#437)

Browse Source
This commit is contained in:
lexluo09
2023-11-28 00:27:21 +08:00
committed by GitHub
parent b5daf04c96
commit 4222d7e2b5
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
chat/python/utils/chromadb_utils.py
View File

@@ -68,7 +68,7 @@ def query_chroma_collection(collection:Collection, query_texts:List[str]=None, q
outer_opt = '$and'
inner_opt = '$eq'
if filter_condition is not None:
if filter_condition is not None and len(filter_condition) > 0:
if len(filter_condition)==1:
outer_filter = filter_condition
else:
@@ -79,10 +79,8 @@ def query_chroma_collection(collection:Collection, query_texts:List[str]=None, q
logger.info('outer_filter: {}'.format(outer_filter))
res = collection.query(query_texts=query_texts, query_embeddings=query_embeddings,
n_results=n_results, where=outer_filter)
return res
return collection.query(query_texts=query_texts, query_embeddings=query_embeddings,
n_results=n_results, where=outer_filter)
def parse_retrieval_chroma_collection_query(res:List[Mapping[str, Any]]):
parsed_res = [[] for _ in range(0, len(res['ids']))]