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

fixes to ruff format

Browse source
This commit is contained in:
vasilije 2026-01-07 16:26:36 +01:00
parent 34c6652939
commit af72dd2fc2
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,
), ),