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

Fix formatiing

Browse source
This commit is contained in:
Andrej Milicevic 2025-09-17 10:45:36 +02:00
parent 30df102656
commit 293a0e0053
3 changed files with 19 additions and 23 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: