chore: refurbish Python code by applying refurb linter rules (#8296)

2024-09-12 15:50:49 +08:00
parent c69f5b07ba
commit 40fb4d16ef
105 changed files with 220 additions and 276 deletions
--- a/api/core/rag/datasource/vdb/milvus/milvus_vector.py
+++ b/api/core/rag/datasource/vdb/milvus/milvus_vector.py
@@ -141,7 +141,7 @@ class MilvusVector(BaseVector):
        for result in results[0]:
            metadata = result["entity"].get(Field.METADATA_KEY.value)
            metadata["score"] = result["distance"]
-            score_threshold = kwargs.get("score_threshold") if kwargs.get("score_threshold") else 0.0
+            score_threshold = kwargs.get("score_threshold", 0.0)
            if result["distance"] > score_threshold:
                doc = Document(page_content=result["entity"].get(Field.CONTENT_KEY.value), metadata=metadata)
                docs.append(doc)