瀏覽代碼

refac: ollama cloud web search count support

Timothy Jaeryang Baek 2 周之前
父節點
當前提交
fe65fe0b97
共有 1 個文件被更改,包括 2 次插入4 次删除
  1. 2 4
      backend/open_webui/retrieval/web/ollama.py

+ 2 - 4
backend/open_webui/retrieval/web/ollama.py

@@ -26,11 +26,9 @@ def search_ollama_cloud(
         filter_list (Optional[list[str]]): List of domains to filter results by
     """
     log.info(f"Searching with Ollama for query: {query}")
-    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
 
-    payload = {
-        "query": query,
-    }
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+    payload = {"query": query, "max_results": count}
 
     try:
         response = requests.post(f"{url}/api/web_search", headers=headers, json=payload)