Преглед на файлове

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

Co-authored-by: crazywoola <427733928@qq.com>
tags/0.8.1
HowardChan преди 1 година
родител
ревизия
82c42b9ec5
No account linked to committer's email address
променени са 1 файла, в които са добавени 3 реда и са изтрити 8 реда
  1. 3
    8
      api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py

+ 3
- 8
api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py Целия файл

@@ -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…
Отказ
Запис