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

Update OllamaEmbeddingEngine.py (#1330)

Browse source 
Compatible private embeddeding model (openai format)

<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
This commit is contained in:
Vasilije 2025-09-15 14:43:34 -07:00 • committed by GitHub
commit 3a073ca462
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")