chore: update something

app.py

@@ -23,6 +23,7 @@ import SelfExtend
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from bs4 import BeautifulSoup
 from functools import lru_cache
+from datetime import datetime, timezone
 
 
 MAX_MAX_NEW_TOKENS = 8192
@@ -403,6 +404,7 @@ def generate(
     top_p: float = 0.95,
     top_k: int = 50,
     repetition_penalty: float = 1.0,
+    local_date_time: str = None,
 ) -> Iterator[str]:
     # print()
     # print("allow_used_tools:\n", allow_used_tools)
@@ -412,13 +414,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 = [local_date_time]            
+        else:
+            references.insert(0, local_date_time)
+
         if (
             references is not None
             and isinstance(references, list)
@@ -484,14 +492,20 @@ 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.")
+                    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 = search_with_google(
-                    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
@@ -555,6 +569,8 @@ chatbot = gr.Chatbot(
     height=500, placeholder=PLACEHOLDER, label="Ghost 8B Beta", show_copy_button=True
 )
 
+utc_dt = datetime.now(timezone.utc)
+
 chat_interface = gr.ChatInterface(
     fn=generate,
     chatbot=chatbot,
@@ -599,6 +615,14 @@ chat_interface = gr.ChatInterface(
             step=0.05,
             value=1.0,
         ),
+        gr.Textbox(
+            label="This user's current time",
+            lines=1,
+            value="This user's current time: {}".format(
+                utc_dt.astimezone().isoformat()
+            ),
+            visible=True,
+        ),
     ],
     stop_btn="Stop",
     cache_examples=False,