yangdx
commited on
Commit
·
82a1a8d
1
Parent(s):
4aa6b0f
Add logging for query parameters in NanoVectorDBStorage.query
Browse files
lightrag/kg/nano_vector_db_impl.py
CHANGED
@@ -137,6 +137,7 @@ class NanoVectorDBStorage(BaseVectorStorage):
|
|
137 |
async def query(self, query: str, top_k=5):
|
138 |
embedding = await self.embedding_func([query])
|
139 |
embedding = embedding[0]
|
|
|
140 |
results = self._client.query(
|
141 |
query=embedding,
|
142 |
top_k=top_k,
|
|
|
137 |
async def query(self, query: str, top_k=5):
|
138 |
embedding = await self.embedding_func([query])
|
139 |
embedding = embedding[0]
|
140 |
+
logger.info(f"Query: {query}, top_k: {top_k}, cosine_better_than_threshold: {self.cosine_better_than_threshold}")
|
141 |
results = self._client.query(
|
142 |
query=embedding,
|
143 |
top_k=top_k,
|