Просмотр исходного кода

refac: ollama cloud web search count support

Timothy Jaeryang Baek 2 недель назад
Родитель
Сommit
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)