diff --git a/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py index 60b85197be..ffa9ea2f90 100644 --- a/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py +++ b/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()