gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

clean up todos from lancedb

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-09-07 20:48:58 +01:00
parent a79ca4a7a4
commit 6db69e635e
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py
View file

@ -57,7 +57,7 @@ class LanceDBAdapter(VectorDBInterface):
def __init__( def __init__(
self, self,
url: Optional[str], # TODO: consider if we want to make this required and/or api_key url: Optional[str],
api_key: Optional[str], api_key: Optional[str],
embedding_engine: EmbeddingEngine, embedding_engine: EmbeddingEngine,
): ):
@ -138,7 +138,7 @@ class LanceDBAdapter(VectorDBInterface):
id: UUID id: UUID
vector: Vector[ vector: Vector[
vector_size vector_size
] # TODO: double check and consider raising this later in Pydantic ]
payload: Dict[str, Any] payload: Dict[str, Any]
if not await self.has_collection(collection_name): if not await self.has_collection(collection_name):