hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-12 04:16:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: ollama text embedding 500 error (#8131)

Browse Source
This commit is contained in:
crazywoola
2024-09-09 15:27:49 +08:00
committed by GitHub
parent 50d92f0fd4
commit a27d4d58ec
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -89,7 +89,8 @@ class OllamaEmbeddingModel(TextEmbeddingModel):
endpoint_url, endpoint_url,
headers=headers, headers=headers,
data=json.dumps(payload), data=json.dumps(payload),
timeout=(10, 300) timeout=(10, 300),
options={"use_mmap": "true"}
) )
response.raise_for_status() # Raise an exception for HTTP errors response.raise_for_status() # Raise an exception for HTTP errors