chore: update something

app.py

@@ -286,13 +286,13 @@ supported_tools = json.dumps(
                             "description": "Search keywords, rephrase to optimize search results based on questions suitable to the specified search type.",
                             "required": True,
                         },
-                        # "type": {
-                        #     "type": "string",
-                        #     "description": "Search type, based on the question to determine whether to search for it in 'wikipedia' or 'google', prefer to use wikipedia for information about events, history and people.",
-                        #     "enum": ["wikipedia", "google"],
-                        #     "default": "google",
-                        #     "required": True,
-                        # },
+                        "type": {
+                            "type": "string",
+                            "description": "Search type, based on the question to determine whether to search for it in 'wikipedia' or 'google', prefer to use wikipedia for information about events, history and people.",
+                            "enum": ["wikipedia", "google"],
+                            "default": "google",
+                            "required": True,
+                        },
                         "language": {
                             "type": "string",
                             "description": "Search language, is the user language code with 2 letters, e.g: vi = vietnamese, en = english.",
@@ -477,25 +477,25 @@ def generate(
                 and scheduled_tools_runs["name"] == "search_on_internet"
             ):
                 keyword = scheduled_tools_runs["arguments"]["keyword"]
-                # search_type = scheduled_tools_runs["arguments"]["type"]
+                search_type = scheduled_tools_runs["arguments"]["type"]
                 language = scheduled_tools_runs["arguments"]["language"]
                 print("scheduled_tools_runs:", scheduled_tools_runs)
-                # if search_type == "wikipedia":
-                #     gr.Info(
-                #         "Searching for information on the Wikipedia.",
-                #         duration=5,
-                #         visible=True,
-                #     )
-                #     document_references.extend(
-                #         search_with_wikipedia(query=keyword, language=language)
-                #     )
+                if search_type == "wikipedia":
+                    gr.Info(
+                        "Searching for information on the Wikipedia.",
+                        duration=5,
+                        visible=True,
+                    )
+                    document_references.extend(
+                        search_with_wikipedia(query=keyword, language=language)
+                    )
 
                 gr.Info("Searching for information on the Google.")
                 document_references.extend(
                     search_with_google(
                         query=keyword,
                         language=language,
-                        num_results=4,
+                        num_results=3,
                         # num_results=2 if search_type == "wikipedia" else 3,
                     )
                 )