AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 16:05:55 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats 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