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", {}) vector_db_kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
self.cosine_better_than_threshold = vector_db_kwargs.get( self.cosine_better_than_threshold = vector_db_kwargs.get(
"cosine_better_than_threshold", "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]): async def upsert(self, data: dict[str, dict]):
@@ -137,7 +139,9 @@ class NanoVectorDBStorage(BaseVectorStorage):
async def query(self, query: str, top_k=5): async def query(self, query: str, top_k=5):
embedding = await self.embedding_func([query]) embedding = await self.embedding_func([query])
embedding = embedding[0] 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( results = self._client.query(
query=embedding, query=embedding,
top_k=top_k, top_k=top_k,