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

format: fix ruff format

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-09-16 15:26:45 +01:00
parent 3a073ca462
commit cccb84ce90
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
View file

@ -94,11 +94,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
""" """
Internal method to call the Ollama embeddings endpoint for a single prompt. Internal method to call the Ollama embeddings endpoint for a single prompt.
""" """
payload = { payload = {"model": self.model, "prompt": prompt, "input": prompt}
"model": self.model,
"prompt": prompt,
"input": prompt
}
headers = {} headers = {}
api_key = os.getenv("LLM_API_KEY") api_key = os.getenv("LLM_API_KEY")
if api_key: if api_key: