NA

app.py

@@ -55,17 +55,16 @@ def create_tools(cfg):
     query_tool = tools_factory.create_rag_tool(
         tool_name = "justice_harvard_query",
         tool_description = """
-        Given a user query, returns a response (str) based on the content of the Justice Harvard lecture transcripts.
-        It can answer questions about the justice, morality, politics and related topics,
+        Answer questions about the justice, morality, politics and related topics,
         based on transcripts of recordings from the Justice Harvard class that includes a lot of content on these topics.
         When using the tool it's best to ask simple short questions. You can break complex questions into sub-queries.
         """,
         tool_args_schema = JusticeHarvardArgs,
-        tool_filter_template = '',
         reranker = "multilingual_reranker_v1", rerank_k = 100, 
-        n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.01,
+        n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
         summary_num_results = 10,
         vectara_summarizer = 'vectara-summary-ext-24-05-med-omni',
+        include_citations = True,
     )
 
     return (tools_factory.get_tools(
@@ -79,15 +78,15 @@ def create_tools(cfg):
     )
 
 def initialize_agent(_cfg):
-    date = datetime.datetime.now().strftime("%Y-%m-%d")
     bot_instructions = f"""
     - You are a helpful teacher assistant, with expertise in education in various teaching styles.
-    - Today's date is {date}.
+    - Obtain information using tools to answer the user's query.
+    - If the tool cannot provide information relevant to the user's query, tell the user that you are unable to provide an answer.
+    - If the tool can provide relevant information, use the adjust_response_to_student tool 
+      to rephrase the response to ensure it fits the student's age of {_cfg.student_age}, 
+      the {_cfg.style} teaching style and the {_cfg.language} language.
     - Response in a concise and clear manner, and provide the most relevant information to the student.
-    - Use tools when available instead of depending on your own knowledge.
-    - Always use the rephrase tool at the end in order to ensure it fits the student's age of {_cfg.student_age}, 
-      the {_cfg.style} teaching style and the {_cfg.language} language
-    - Always use any guardrails tools to ensure your responses are polite and do not discuss politices.
+    - Never discuss politics, and always respond politely.
     """
 
     def update_func(status_type: AgentStatusType, msg: str):
@@ -96,12 +95,15 @@ def initialize_agent(_cfg):
 
     agent = Agent(
         tools=create_tools(_cfg),
-        topic="An educator with expertise in philosophy",
+        topic="justice, morality, politics, and philosophy",
         custom_instructions=bot_instructions,
         update_func=update_func
     )
     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
@@ -109,6 +111,7 @@ def launch_bot():
         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="Justice Harvard Teaching Assistant", layout="wide")
@@ -150,7 +153,7 @@ def launch_bot():
 
         st.markdown("\n") 
         cfg.student_age = st.number_input(
-            'Student age:',  min_value=13, value=cfg.student_age,
+            'Student age:',  min_value=13, max_value=99, value=cfg.student_age,
             step=1, format='%i'
         )
         if st.session_state.student_age != cfg.student_age:
@@ -158,13 +161,10 @@ def launch_bot():
             reset()
 
         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(
@@ -177,7 +177,7 @@ def launch_bot():
 
     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"]):
@@ -186,27 +186,33 @@ 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)
-
-    if st.session_state.messages[-1]["role"] != "assistant":
+        
+    # Generate a new response if last message is not from 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)
-                cleaned = re.sub(r'\[\d+\]', '', res).replace('$', '\\$')
-            message = {"role": "assistant", "content": cleaned, "avatar": '🤖'}
+                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()
-  
-    # Display log messages in an expander
-    if st.session_state.show_logs:
-        with st.expander("Agent Log Messages", expanded=True):
+            st.markdown(res)
+            st.session_state.prompt = None
+
+    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()