fix: lint

langgenius · Sep 11, 2024 · a46c6f9 · a46c6f9
1 parent 4b9b0e2
commit a46c6f9
Showing 1 changed file with 2 additions and 10 deletions.
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
@@ -77,18 +77,10 @@ def _invoke(
                 inputs.append(text)
 
         # 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
-        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))
 
         response.raise_for_status()  # Raise an exception for HTTP errors
         response_data = response.json()