updates

Dockerfile

@@ -4,7 +4,9 @@ WORKDIR /app
 
 COPY ./requirements.txt /app/requirements.txt
     
-RUN pip3 install --no-cache-dir -r /app/requirements.txt
+RUN pip3 install --no-cache-dir --upgrade pip
+RUN pip3 install --no-cache-dir wheel setuptools build
+RUN pip3 install --no-cache-dir --use-pep517 -r /app/requirements.txt
 
 # User
 RUN useradd -m -u 1000 user

agent.py

@@ -48,11 +48,11 @@ def create_assistant_tools(cfg):
             },
             {
                 "type": "mmr",
-                "diversity_bias": 0.4,
+                "diversity_bias": 0.2,
                 "limit": 30
             }
         ],
-        n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
+        n_sentences_before = 2, n_sentences_after = 3, lambda_val = 0.005,
         vectara_summarizer = summarizer,
         include_citations = True,
         verbose=False
@@ -80,6 +80,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
     - For questions about customers' complaints (the text of the complaint), use the ask_complaints tool.
       You only need the query parameter to use this tool, but you can supply other parameters if provided.
       Do not include the "References" section in your response.
+    - When using a company name with a tool, if the tool returns no information, try a difference variation of that company names, as well as different capitalization.
     - Never discuss politics, and always respond politely.
     """
 
@@ -89,7 +90,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
         custom_instructions=cfpb_complaints_bot_instructions,
         agent_progress_callback=agent_progress_callback
     )
-    agent.report()
+    agent.report(detailed=False)
     return agent
 
 

app.py

@@ -1,6 +1,7 @@
 import os
 
 import streamlit as st
+import torch
 from st_app import launch_bot
 
 import nest_asyncio
@@ -10,6 +11,8 @@ import uuid
 import sqlite3
 from datasets import load_dataset
 
+torch.classes.__path__ = []
+
 # Setup for HTTP API Calls to Amplitude Analytics
 if 'device_id' not in st.session_state:
     st.session_state.device_id = str(uuid.uuid4())

requirements.txt

@@ -1,10 +1,10 @@
 omegaconf==2.3.0
 python-dotenv==1.0.1
-streamlit==1.41.1
-streamlit_pills==0.3.0
+streamlit==1.43.2
 streamlit-feedback==0.1.3
 langdetect==1.0.9
 langcodes==3.4.0
 datasets==2.19.2
 uuid==1.30
-vectara-agentic==0.2.5
+vectara-agentic==0.2.9
+torch==2.6.0

st_app.py

@@ -3,7 +3,6 @@ import sys
 import re
 
 import streamlit as st
-from streamlit_pills import pills
 from streamlit_feedback import streamlit_feedback
 
 from utils import thumbs_feedback, escape_dollars_outside_latex, send_amplitude_data
@@ -15,7 +14,7 @@ initial_prompt = "How can I help you today?"
 
 def show_example_questions():        
     if len(st.session_state.example_messages) > 0 and st.session_state.first_turn:            
-        selected_example = pills("Queries to Try:", st.session_state.example_messages, index=None)
+        selected_example = st.pills("Queries to Try:", st.session_state.example_messages, default=None)
         if selected_example:
             st.session_state.ex_prompt = selected_example
             st.session_state.first_turn = False