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