chore: update something

app.py

@@ -279,7 +279,7 @@ if torch.cuda.is_available():
     )
     model.generation_config.max_length = 123392
 
-waiting_tools_timeout = 7.5
+waiting_tools_timeout = 10
 supported_tools = json.dumps(
     [
         {
@@ -302,6 +302,12 @@ supported_tools = json.dumps(
                             "default": "google",
                             "required": True,
                         },
+                        "language": {
+                            "type": "string",
+                            "description": "Search language, is the user language code with 2 letters, e.g: vi = vietnamese, en = english.",
+                            "default": "en",
+                            "required": True,
+                        },
                     },
                 },
             },
@@ -320,10 +326,16 @@ def extract_text_from_webpage(html_content):
     return visible_text
 
 
-def search_with_wikipedia(query: str):
+def search_with_wikipedia(
+    query: str,
+    language: str = "en",
+):
     all_results = []
     try:
-        all_results.append(wikipedia.summary(query))
+        wikipedia.set_lang(language)
+        page = wikipedia.page(query)
+        content = page.content[:5120]
+        all_results.append(content)
     except Exception as e:
         pass
     return all_results
@@ -333,10 +345,11 @@ def search_with_google(
     query: str,
     num_results: int = 3,
     timeout: int = 5,
+    language: str = "en",
     ssl_verify: bool = None,
 ):
     all_results = []
-    max_chars_per_page = 8192
+    max_chars_per_page = 4096
     with requests.Session() as session:
         resp = session.get(
             url="https://www.google.com/search",
@@ -347,6 +360,7 @@ def search_with_google(
                 "q": query,
                 "num": num_results,
                 "udm": 14,
+                "hl": language,
             },
             timeout=timeout,
             verify=ssl_verify,
@@ -461,12 +475,20 @@ def generate(
             ):
                 keyword = scheduled_tools_runs["arguments"]["keyword"]
                 search_type = scheduled_tools_runs["arguments"]["type"]
+                language = scheduled_tools_runs["arguments"]["language"]
                 if search_type == "wikipedia":
                     gr.Info("Searching for information on the Wikipedia.")
-                    document_references = search_with_wikipedia(keyword)
+                    document_references = search_with_wikipedia(
+                        query=keyword, language=language
+                    )
                 else:
                     gr.Info("Searching for information on the Google.")
-                    document_references = search_with_google(keyword)
+                    document_references = search_with_google(
+                        query=keyword, language=language
+                    )
+                print(
+                    "scheduled_tools_runs:", scheduled_tools_runs, document_references
+                )
 
         apply_tools = (
             True if allow_used_tools is True and previous_response is None else False
@@ -516,7 +538,6 @@ def generate(
             and state["respond"] is False
             and state["mark"] + waiting_tools_timeout > time.time()
         ):
-            gr.Info("Searching for information on the internet.")
             previous_response = "".join(outputs)
             yield from generate_chat_responses(previous_response=previous_response)