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

Handle multiple response formats in OllamaEmbeddingEngine

Browse source 
The OllamaEmbeddingEngine is compatible with OpenAI
This commit is contained in:
Gao,Wei 2025-11-05 17:44:08 +08:00 • committed by GitHub
parent 1e531385a6
commit 72c20c256d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
View file

@ -124,7 +124,10 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
self.endpoint, json=payload, headers=headers, timeout=60.0 self.endpoint, json=payload, headers=headers, timeout=60.0
) as response: ) as response:
data = await response.json() data = await response.json()
return data["embeddings"][0] if "embeddings" in data:
return data["embeddings"][0]
else:
return data["data"][0]["embedding"]
def get_vector_size(self) -> int: def get_vector_size(self) -> int:
""" """