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

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

View file

@ -115,7 +115,7 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
"""
try:
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]}
return [data["embedding"] for data in response["data"]]
else: