浏览代码

fix: include_domain does NOT exist

Timothy Jaeryang Baek 2 月之前
父节点
当前提交
ba6cde8a87
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 1 2
      backend/open_webui/retrieval/web/tavily.py
  2. 1 0
      backend/open_webui/routers/retrieval.py

+ 1 - 2
backend/open_webui/retrieval/web/tavily.py

@@ -27,8 +27,7 @@ def search_tavily(
     """
     """
     url = "https://api.tavily.com/search"
     url = "https://api.tavily.com/search"
     data = {"query": query, "api_key": api_key}
     data = {"query": query, "api_key": api_key}
-    include_domain = filter_list
-    response = requests.post(url, include_domain, json=data)
+    response = requests.post(url, json=data)
     response.raise_for_status()
     response.raise_for_status()
 
 
     json_response = response.json()
     json_response = response.json()

+ 1 - 0
backend/open_webui/routers/retrieval.py

@@ -1270,6 +1270,7 @@ def search_web(request: Request, engine: str, query: str) -> list[SearchResult]:
                 request.app.state.config.TAVILY_API_KEY,
                 request.app.state.config.TAVILY_API_KEY,
                 query,
                 query,
                 request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
                 request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
+                request.app.state.config.RAG_WEB_SEARCH_DOMAIN_FILTER_LIST,
             )
             )
         else:
         else:
             raise Exception("No TAVILY_API_KEY found in environment variables")
             raise Exception("No TAVILY_API_KEY found in environment variables")