Fix web_search in Web UI

hlohaus · Jan 15, 2025 · 78fa745 · 78fa745
1 parent ae978ad
commit 78fa745
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/g4f/tools/run_tools.py b/g4f/tools/run_tools.py
@@ -33,7 +33,8 @@ async def async_iter_run_tools(async_iter_callback, model, messages, tool_calls:
     if web_search:
         try:
             messages = messages.copy()
-            messages[-1]["content"] = await do_search(messages[-1]["content"], web_search if web_search != "true" else None)
+            web_search = web_search if isinstance(web_search, str) and web_search != "true" else None
+            messages[-1]["content"] = await do_search(messages[-1]["content"], web_search)
         except Exception as e:
             debug.log(f"Couldn't do web search: {e.__class__.__name__}: {e}")
             # Keep web_search in kwargs for provider native support
@@ -83,7 +84,8 @@ def iter_run_tools(
     if web_search:
         try:
             messages = messages.copy()
-            messages[-1]["content"] = asyncio.run(do_search(messages[-1]["content"], web_search if web_search != "true" else None))
+            web_search = web_search if isinstance(web_search, str) and web_search != "true" else None
+            messages[-1]["content"] = asyncio.run(do_search(messages[-1]["content"], web_search))
         except Exception as e:
             debug.log(f"Couldn't do web search: {e.__class__.__name__}: {e}")
             # Keep web_search in kwargs for provider native support