NA

app.py

@@ -13,6 +13,7 @@ from bs4 import BeautifulSoup
 from pydantic import Field, BaseModel
 from vectara_agent.agent import Agent, AgentStatusType
 from vectara_agent.tools import ToolsFactory
+from vectara_agent.tools_catalog import summarize_text
 
 initial_prompt = "How can I help you today?"
 
@@ -26,6 +27,7 @@ get_headers = {
     "Connection": "keep-alive",
 }
 
+
 def create_tools(cfg):    
     
     class QueryHackerNews(BaseModel):
@@ -128,6 +130,18 @@ def create_tools(cfg):
         text = soup.get_text(" ", strip=True).replace('\n', ' ')
         return text
 
+    def whats_new(
+            n_stories: int = Field(default=10, description="The number of new stories to return.")
+    ) -> list[str]:
+        """
+        Provides a succint summary of what is new in the hackernews community
+        by summarizing the content and comments of top stories.
+        Returns a string with the summary.
+        """
+        stories = get_top_stories(n_stories)
+        texts = [get_story_text(story_id) for story_id in stories[:n_stories]]
+        all_stories = '---------\n\n'.join(texts)
+        return summarize_text(all_stories)
 
     return (
         tools_factory.get_tools(
@@ -137,6 +151,7 @@ def create_tools(cfg):
                     get_ask_stories,
                     get_story_details,
                     get_story_text,
+                    whats_new
                 ]
             ) +
         tools_factory.standard_tools() + 
@@ -145,15 +160,9 @@ def create_tools(cfg):
     )
 
 def initialize_agent(_cfg):
-    date = datetime.datetime.now().strftime("%Y-%m-%d")
-    bot_instructions = f"""
-    - You are a helpful assistant, answering user questions about content from hacker news. 
-    - Today's date is {date}.
+    bot_instructions = """
+    - You are a helpful assistant, with expertise in answering user questions about Hacker News stories and comments.
     - Never discuss politics, and always respond politely.
-    - Use tools when available instead of depending on your own knowledge. 
-    - If a tool provides citations, you can include them in your response to provide more context.
-    - If a tool cannot respond properly, retry with a rephrased question or ask the user for more information.
-    - Be very careful not to report results you are not confident about.
     """
 
     def update_func(status_type: AgentStatusType, msg: str):
@@ -168,13 +177,15 @@ def initialize_agent(_cfg):
     )
     return agent
 
+def toggle_logs():
+    st.session_state.show_logs = not st.session_state.show_logs
+
 def launch_bot():
     def reset():
-        cfg = st.session_state.cfg
         st.session_state.messages = [{"role": "assistant", "content": initial_prompt, "avatar": "🦖"}]
         st.session_state.thinking_message = "Agent at work..."
-        st.session_state.agent = initialize_agent(cfg)
         st.session_state.log_messages = []
+        st.session_state.prompt = None
         st.session_state.show_logs = False
 
     st.set_page_config(page_title="Hacker News Bot", layout="wide")
@@ -186,7 +197,10 @@ def launch_bot():
         })
         st.session_state.cfg = cfg
         reset()
+
     cfg = st.session_state.cfg
+    if 'agent' not in st.session_state:
+        st.session_state.agent = initialize_agent(cfg)
 
     # left side content
     with st.sidebar:
@@ -195,13 +209,10 @@ def launch_bot():
         st.markdown("## Welcome to the hacker news assistant demo.\n\n\n")
 
         st.markdown("\n\n")
-        bc1, bc2 = st.columns([1, 1])
+        bc1, _ = st.columns([1, 1])
         with bc1:
             if st.button('Start Over'):
                 reset()
-        with bc2:
-            if st.button('Show Logs'):
-                st.session_state.show_logs = not st.session_state.show_logs
 
         st.markdown("---")
         st.markdown(
@@ -211,10 +222,9 @@ def launch_bot():
         )
         st.markdown("---")
 
-
     if "messages" not in st.session_state.keys():
         reset()
-
+    
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"], avatar=message["avatar"]):
@@ -223,28 +233,34 @@ def launch_bot():
     # User-provided prompt
     if prompt := st.chat_input():
         st.session_state.messages.append({"role": "user", "content": prompt, "avatar": '🧑‍💻'})
+        st.session_state.prompt = prompt  # Save the prompt in session state
+        st.session_state.log_messages = []
+        st.session_state.show_logs = False
         with st.chat_message("user", avatar='🧑‍💻'):
             print(f"Starting new question: {prompt}\n")
             st.write(prompt)
-    
+        
     # Generate a new response if last message is not from assistant
-    if st.session_state.messages[-1]["role"] != "assistant":
+    if st.session_state.prompt:
         with st.chat_message("assistant", avatar='🤖'):
             with st.spinner(st.session_state.thinking_message):
-                res = st.session_state.agent.chat(prompt)
+                res = st.session_state.agent.chat(st.session_state.prompt)
                 res = res.replace('$', '\\$')  # escape dollar sign for markdown
             message = {"role": "assistant", "content": res, "avatar": '🤖'}
             st.session_state.messages.append(message)
-            st.rerun()
+            st.markdown(res)
+            st.session_state.prompt = None
 
-    # Display log messages in an expander
-    if st.session_state.show_logs:
-        with st.expander("Agent Log Messages", expanded=True):
+    log_placeholder = st.empty()
+    with log_placeholder.container():
+        if st.session_state.show_logs:
+            st.button("Hide Logs", on_click=toggle_logs)
             for msg in st.session_state.log_messages:
                 st.write(msg)
-            if st.button('Close Logs'):
-                st.session_state.show_logs = False
-                st.rerun()
+        else:
+            if len(st.session_state.log_messages) > 0:
+                st.button("Show Logs", on_click=toggle_logs)
+
 
     sys.stdout.flush()