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:
parent
338c9e4366
commit
e2f85d70b2
1 changed files with 1 additions and 1 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue