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

Update cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py

Browse source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Stony 2026-01-05 16:26:02 +08:00 • committed by GitHub
parent 338c9e4366
commit e2f85d70b2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
View file

@ -115,7 +115,7 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
""" """
try: try:
if self.mock: if self.mock:
dim = self.dimensions if self.dimensions is not None else 3072 dim = self.dimensions if self.dimensions is not None else 3072
response = {"data": [{"embedding": [0.0] * dim} for _ in text]} response = {"data": [{"embedding": [0.0] * dim} for _ in text]}
return [data["embedding"] for data in response["data"]] return [data["embedding"] for data in response["data"]]
else: else: