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

reformat

Browse source
This commit is contained in:
vasilije 2026-01-06 18:05:34 +01:00
parent 555eef69e3
commit aa4ab1ed8a
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -134,7 +134,9 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
litellm.aembedding( litellm.aembedding(
model=self.model, model=self.model,
input=text, input=text,
api_key=self.api_key if self.api_key and self.api_key.strip() != "" else "EMPTY", api_key=self.api_key
if self.api_key and self.api_key.strip() != ""
else "EMPTY",
api_base=self.endpoint, api_base=self.endpoint,
api_version=self.api_version, api_version=self.api_version,
), ),