Fix linting

This commit is contained in:
yangdx
2025-01-29 22:14:18 +08:00
parent 5c7b2d7c9f
commit 90c765c724
2 changed files with 7 additions and 3 deletions

View File

@@ -87,7 +87,9 @@ class NanoVectorDBStorage(BaseVectorStorage):
vector_db_kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
self.cosine_better_than_threshold = vector_db_kwargs.get(
"cosine_better_than_threshold",
self.global_config.get("cosine_better_than_threshold", self.cosine_better_than_threshold)
self.global_config.get(
"cosine_better_than_threshold", self.cosine_better_than_threshold
),
)
async def upsert(self, data: dict[str, dict]):
@@ -137,7 +139,9 @@ class NanoVectorDBStorage(BaseVectorStorage):
async def query(self, query: str, top_k=5):
embedding = await self.embedding_func([query])
embedding = embedding[0]
logger.info(f"Query: {query}, top_k: {top_k}, cosine_better_than_threshold: {self.cosine_better_than_threshold}")
logger.info(
f"Query: {query}, top_k: {top_k}, cosine_better_than_threshold: {self.cosine_better_than_threshold}"
)
results = self._client.query(
query=embedding,
top_k=top_k,

View File

@@ -1028,7 +1028,7 @@ async def _build_query_context(
# not necessary to use LLM to generate a response
if not entities_context.strip() and not relations_context.strip():
return None
return f"""
-----Entities-----
```csv