chore: update something

app.py

@@ -325,7 +325,7 @@ def search_with_wikipedia(
     try:
         wikipedia.set_lang(language)
         page = wikipedia.page(query)
-        content = page.content[:8192]
+        content = page.content[:4096]
         all_results.append(content)
     except Exception as e:
         pass
@@ -334,7 +334,7 @@ def search_with_wikipedia(
 
 def search_with_google(
     query: str,
-    num_results: int = 4,
+    num_results: int = 3,
     timeout: int = 5,
     language: str = "en",
     ssl_verify: bool = None,
@@ -394,6 +394,7 @@ def generate(
     top_p: float = 0.95,
     top_k: int = 50,
     repetition_penalty: float = 1.0,
+    other_client_info: str = None,
 ) -> Iterator[str]:
     # print()
     # print("allow_used_tools:\n", allow_used_tools)
@@ -403,13 +404,19 @@ def generate(
 
     def build_input_ids(
         apply_tools: bool = None,
-        references: list[str] = None,
+        references = None,
     ):
         conversation = []
         if system_prompt:
             conversation.append({"role": "system", "content": system_prompt})
         if apply_tools is True:
             conversation.append({"role": "tools", "content": supported_tools})
+
+        if references is None:
+            references = [other_client_info]
+        else:
+            references.insert(0, other_client_info)
+
         if (
             references is not None
             and isinstance(references, list)
@@ -475,15 +482,21 @@ def generate(
                 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(
-                        query=keyword, language=language
+                    gr.Info(
+                        "Searching for information on the Wikipedia.",
+                        duration=5,
+                        visible=True,
                     )
-                else:
-                    gr.Info("Searching for information on the Google.")
-                    document_references = search_with_google(
-                        query=keyword, language=language
+                    document_references.extend(
+                        search_with_wikipedia(query=keyword, language=language)
                     )
+
+                gr.Info(
+                    "Searching for information on the Google.", duration=5, visible=True
+                )
+                document_references.extend(
+                    search_with_google(query=keyword, language=language)
+                )
                 print(
                     "scheduled_tools_runs:", scheduled_tools_runs, document_references
                 )
@@ -590,6 +603,12 @@ chat_interface = gr.ChatInterface(
             step=0.05,
             value=1.0,
         ),
+        gr.Textbox(
+            label="Other client information",
+            lines=1,
+            value="This user's current time: {}".format(time.strftime("%Y-%m-%d")),
+            visible=False,
+        ),
     ],
     stop_btn="Stop",
     cache_examples=False,