update

agent.py

@@ -1,7 +1,5 @@
 import os
-from typing import Optional
 from omegaconf import OmegaConf
-
 from vectara_agentic.agent import Agent
 
 from dotenv import load_dotenv
@@ -12,8 +10,7 @@ initial_prompt = "How can I help you today?"
 
 def initialize_agent(_cfg, agent_progress_callback=None):
     agent = Agent.from_corpus(
-        vectara_customer_id=_cfg.customer_id,
-        vectara_corpus_id=_cfg.corpus_id,
+        vectara_corpus_key=_cfg.corpus_key,
         vectara_api_key=_cfg.api_key,
         tool_name="ask_ucsf_ortho",
         data_description="UCSF Orthopedic Website",
@@ -24,6 +21,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
         vectara_rerank_k=25,
         vectara_summary_num_results=15,
         verbose=False,
+        agent_progress_callback=agent_progress_callback,
     )
     agent.report()
     return agent
@@ -31,8 +29,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': "UCSF Ortho Demo",

requirements.txt

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