version upgrade

agent.py

@@ -10,8 +10,9 @@ load_dotenv(override=True)
 
 from pydantic import Field, BaseModel
 from vectara_agentic.agent import Agent
+from vectara_agentic.agent_config import AgentConfig
 from vectara_agentic.tools import ToolsFactory, VectaraToolFactory
-from vectara_agentic.tools_catalog import summarize_text
+from vectara_agentic.tools_catalog import ToolsCatalog
 
 initial_prompt = "How can I help you today?"
 
@@ -23,14 +24,13 @@ get_headers = {
     "Connection": "keep-alive",
 }
 
-def create_assistant_tools(cfg):    
+def create_assistant_tools(cfg, agent_config):
     
     class QueryHackerNews(BaseModel):
         query: str = Field(..., description="The user query.")
 
     vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_key, 
-                                     vectara_customer_id=cfg.customer_id, 
-                                     vectara_corpus_id=cfg.corpus_id)
+                                     vectara_corpus_key=cfg.corpus_key)
     summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
     ask_hackernews = vec_factory.create_rag_tool(
         tool_name = "ask_hackernews",
@@ -121,6 +121,7 @@ def create_assistant_tools(cfg):
         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)
+        summarize_text = ToolsCatalog(agent_config).summarize_text
         return summarize_text(all_stories)
     
     tools_factory = ToolsFactory()
@@ -154,9 +155,9 @@ def initialize_agent(_cfg, agent_progress_callback = None):
     - You can use the tavily_search tool to gain additional information if needed for follow up questions about Hacker News topic.
     - When including information or links provided by the tavily_search tool, make sure to notify the user in your response that this is not based on Hacker News stories.
     """
-
+    agent_config = AgentConfig()
     agent = Agent(
-        tools=create_assistant_tools(_cfg),
+        tools=create_assistant_tools(_cfg, agent_config),
         topic="hacker news",
         custom_instructions=bot_instructions,
         agent_progress_callback=agent_progress_callback
@@ -166,8 +167,7 @@ def initialize_agent(_cfg, agent_progress_callback = None):
 
 def get_agent_config() -> OmegaConf:
     cfg = OmegaConf.create({
-        'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
-        'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
+        'corpus_key': str(os.environ['VECTARA_CORPUS_KEY']),
         'api_key': str(os.environ['VECTARA_API_KEY']),
         'examples': os.environ.get('QUERY_EXAMPLES', None),
         'demo_name': "hacker-news-chat",

requirements.txt

@@ -6,5 +6,5 @@ streamlit_feedback==0.1.3
 uuid==1.30
 langdetect==1.0.9
 langcodes==3.4.0
-vectara-agentic==0.1.22
+vectara-agentic==0.2.0
 

st_app.py

@@ -132,8 +132,8 @@ async def launch_bot():
     if st.session_state.prompt:
         with st.chat_message("assistant", avatar='🤖'):
             st.session_state.status = st.status('Processing...', expanded=False)
-            res = st.session_state.agent.chat(st.session_state.prompt)
-            res = escape_dollars_outside_latex(res)
+            response = st.session_state.agent.chat(st.session_state.prompt)
+            res = escape_dollars_outside_latex(response.response)
             message = {"role": "assistant", "content": res, "avatar": '🤖'}
             st.session_state.messages.append(message)
             st.markdown(res)