chore: update something

app.py

@@ -333,9 +333,7 @@ def search_with_wikipedia(
     all_results = []
     try:
         wikipedia.set_lang(language)
-        page = wikipedia.page(query)
-        content = page.content[:8192]
-        all_results.append(content)
+        all_results.append(wikipedia.summary(query))
     except Exception as e:
         pass
     return all_results
@@ -490,6 +488,9 @@ def generate(
                 keyword = scheduled_tools_runs["arguments"]["keyword"]
                 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.",
@@ -504,10 +505,14 @@ def generate(
                     "Searching for information on the Google.", duration=5, visible=True
                 )
                 document_references.extend(
-                    search_with_google(query=keyword, language=language)
+                    search_with_google(
+                        query=keyword,
+                        language=language,
+                        num_results=2 if search_type == "wikipedia" else 3,
+                    )
                 )
                 print(
-                    "scheduled_tools_runs:", scheduled_tools_runs, document_references
+                    "document_references:", document_references
                 )
 
         apply_tools = (