kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

bug: --

Browse Source
This commit is contained in:
littlewwwhite 2025-02-15 23:23:59 +08:00
parent 31bbaa307d
commit 5bea2d5f44
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/routers/chat_router.py
View File

@ -72,12 +72,7 @@ def chat_post(
chunk = make_chunk("", "searching", history=None) chunk = make_chunk("", "searching", history=None)
yield chunk yield chunk
<<<<<<< HEAD
meta["config"] = startup.config
new_query, refs = startup.retriever(query, history_manager.messages, meta)
=======
modified_query, refs = startup.retriever(modified_query, history_manager.messages, meta) modified_query, refs = startup.retriever(modified_query, history_manager.messages, meta)
>>>>>>> feature/web_search
refs_pool[cur_res_id] = refs refs_pool[cur_res_id] = refs
messages = history_manager.get_history_with_msg(modified_query, max_rounds=meta.get('history_round')) messages = history_manager.get_history_with_msg(modified_query, max_rounds=meta.get('history_round'))

2
src/utils/web_search.py
View File

@ -7,7 +7,7 @@ logger = setup_logger("web-search")
class WebSearcher: class WebSearcher:
def __init__(self): def __init__(self):
api_key = os.getenv("TAVILY_API_KEY", "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") api_key = os.getenv("TAVILY_API_KEY")
if not api_key: if not api_key:
raise ValueError("TAVILY_API_KEY environment variable is not set") raise ValueError("TAVILY_API_KEY environment variable is not set")
self.client = TavilyClient(api_key) self.client = TavilyClient(api_key)