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

Ollama fixes, missing libs + config fixes

Browse source
This commit is contained in:
Vasilije 2024-03-26 17:52:49 +01:00
parent 4e1b2db8ae
commit 365c7bfc0e
2 changed files with 33 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/InfrastructureConfig.py
View file

@ -24,7 +24,7 @@ class InfrastructureConfig():
self.vector_engine = WeaviateAdapter( self.vector_engine = WeaviateAdapter(
config.weaviate_url, config.weaviate_url,
config.weaviate_api_key, config.weaviate_api_key,
config.openai_key embedding_engine = DefaultEmbeddingEngine()
) )
return { return {

4
cognee/infrastructure/databases/vector/weaviate_db/WeaviateAdapter.py
View file

@ -9,6 +9,7 @@ from ..models.DataPoint import DataPoint
from ..models.ScoredResult import ScoredResult from ..models.ScoredResult import ScoredResult
from ..embeddings.EmbeddingEngine import EmbeddingEngine from ..embeddings.EmbeddingEngine import EmbeddingEngine
class WeaviateAdapter(VectorDBInterface): class WeaviateAdapter(VectorDBInterface):
async_pool: Pool = None async_pool: Pool = None
embedding_engine: EmbeddingEngine = None embedding_engine: EmbeddingEngine = None
@ -44,7 +45,8 @@ class WeaviateAdapter(VectorDBInterface):
return self.client.collections.get(collection_name) return self.client.collections.get(collection_name)
async def create_data_points(self, collection_name: str, data_points: List[DataPoint]): async def create_data_points(self, collection_name: str, data_points: List[DataPoint]):
data_vectors = await self.embed_data(list(map(lambda data_point: data_point.get_embeddable_data(), data_points))) data_vectors = await self.embed_data(
list(map(lambda data_point: data_point.get_embeddable_data(), data_points)))
def convert_to_weaviate_data_points(data_point: DataPoint): def convert_to_weaviate_data_points(data_point: DataPoint):
return DataObject( return DataObject(