updated

agent.py

@@ -45,8 +45,7 @@ def create_assistant_tools(cfg):
         query: str = Field(..., description="The user query, always in the form of a question", examples=["what are the risks reported?", "which drug was use on the and how big was the population?"])        
         
     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-summary-table-md-query-ext-jan-2025-gpt-4o'
     ask_publications = vec_factory.create_rag_tool(
         tool_name = "ask_publications",

config.py

@@ -5,8 +5,7 @@ load_dotenv(override=True)
 
 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'])
     })
     return cfg