Sfoglia il codice sorgente

fix:error when adding the ollama embedding model (#8236)

Co-authored-by: crazywoola <427733928@qq.com>
tags/0.8.1
HowardChan 1 anno fa
parent
commit
82c42b9ec5
Nessun account collegato all'indirizzo email del committer

+ 3
- 8
api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py Vedi File

@@ -77,15 +77,10 @@ class OllamaEmbeddingModel(TextEmbeddingModel):
inputs.append(text)

# Prepare the payload for the request
payload = {
"input": inputs,
"model": model,
}
payload = {"input": inputs, "model": model, "options": {"use_mmap": "true"}}

# Make the request to the OpenAI API
response = requests.post(
endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300), options={"use_mmap": "true"}
)
# Make the request to the Ollama API
response = requests.post(endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300))

response.raise_for_status() # Raise an exception for HTTP errors
response_data = response.json()

Loading…
Annulla
Salva