estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix:ollama text embedding 500 error (#8252)

Browse Source
This commit is contained in:
HowardChan
2024-09-11 16:23:19 +08:00
committed by GitHub
parent 75c1a82556
commit 53f37a6704
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py
View File

@@ -77,7 +77,7 @@ class OllamaEmbeddingModel(TextEmbeddingModel):
inputs.append(text) inputs.append(text)
# Prepare the payload for the request # Prepare the payload for the request
payload = {"input": inputs, "model": model, "options": {"use_mmap": "true"}} payload = {"input": inputs, "model": model, "options": {"use_mmap": True}}
# Make the request to the Ollama API # Make the request to the Ollama API
response = requests.post(endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300)) response = requests.post(endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300))