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

Update OllamaEmbeddingEngine.py

Browse source 
Compatible private embeddeding model (openai format)
This commit is contained in:
Gao,Wei 2025-09-04 15:27:36 +08:00 • committed by GitHub
parent 8b6aaff554
commit ac8f30f45c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -97,6 +97,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
payload = { payload = {
"model": self.model, "model": self.model,
"prompt": prompt, "prompt": prompt,
"input": prompt
} }
headers = {} headers = {}
api_key = os.getenv("LLM_API_KEY") api_key = os.getenv("LLM_API_KEY")