Update OllamaEmbeddingEngine.py (#1330)
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:
commit
3a073ca462
1 changed files with 1 additions and 0 deletions
|
|
@ -97,6 +97,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
|
|||
payload = {
|
||||
"model": self.model,
|
||||
"prompt": prompt,
|
||||
"input": prompt
|
||||
}
|
||||
headers = {}
|
||||
api_key = os.getenv("LLM_API_KEY")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue