updates

agent.py

@@ -79,8 +79,8 @@ def create_assistant_tools(cfg):
 
     class QueryTranscriptsArgs(BaseModel):
         query: str = Field(..., description="The user query, always in the form of a question", examples=["what are the risks reported?", "who are the competitors?"])
-        year: int = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)}.")
-        ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {tickers.keys()}.")
+        year: int | str = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)} or a string specifying a condition on the year (example: '>2020').")
+        ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {list(tickers.keys())}.")
 
     vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_key,
                                      vectara_customer_id=cfg.customer_id,

requirements.txt

@@ -1,9 +1,10 @@
 omegaconf==2.3.0
 python-dotenv==1.0.1
-streamlit==1.39.0
+streamlit==1.41.1
 streamlit_pills==0.3.0
 streamlit_feedback==0.1.3
 uuid==1.30
 langdetect==1.0.9
 langcodes==3.4.0
-vectara-agentic==0.1.19
+vectara-agentic==0.1.21
+anthropic==0.40.0

utils.py

@@ -31,8 +31,11 @@ def thumbs_feedback(feedback, **kwargs):
 
 def send_amplitude_data(user_query, bot_response, demo_name, feedback=None):
     # Send query and response to Amplitude Analytics
+    amplitude_token = os.environ.get('AMPLITUDE_TOKEN', None)
+    if amplitude_token is None:
+        return
     data = {
-        "api_key": os.getenv('AMPLITUDE_TOKEN'),
+        "api_key": amplitude_token,
         "events": [{
             "device_id": st.session_state.device_id,
             "event_type": "submitted_query",