ofermend commited on
Commit
adf13ff
·
1 Parent(s): 4792c87

bump to 0.2.0

Browse files
Files changed (2) hide show
  1. agent.py +2 -4
  2. requirements.txt +1 -1
agent.py CHANGED
@@ -89,8 +89,7 @@ def create_assistant_tools(cfg):
89
  ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {list(tickers.keys())}.")
90
 
91
  vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_key,
92
- vectara_customer_id=cfg.customer_id,
93
- vectara_corpus_id=cfg.corpus_id)
94
  summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
95
  ask_transcripts = vec_factory.create_rag_tool(
96
  tool_name = "ask_transcripts",
@@ -172,8 +171,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
172
  def get_agent_config() -> OmegaConf:
173
  companies = ", ".join(tickers.values())
174
  cfg = OmegaConf.create({
175
- 'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
176
- 'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
177
  'api_key': str(os.environ['VECTARA_API_KEY']),
178
  'examples': os.environ.get('QUERY_EXAMPLES', None),
179
  'demo_name': "finance-chat",
 
89
  ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {list(tickers.keys())}.")
90
 
91
  vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_key,
92
+ vectara_corpus_key=cfg.corpus_key)
 
93
  summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
94
  ask_transcripts = vec_factory.create_rag_tool(
95
  tool_name = "ask_transcripts",
 
171
  def get_agent_config() -> OmegaConf:
172
  companies = ", ".join(tickers.values())
173
  cfg = OmegaConf.create({
174
+ 'corpus_key': str(os.environ['VECTARA_CORPUS_KEY']),
 
175
  'api_key': str(os.environ['VECTARA_API_KEY']),
176
  'examples': os.environ.get('QUERY_EXAMPLES', None),
177
  'demo_name': "finance-chat",
requirements.txt CHANGED
@@ -6,4 +6,4 @@ streamlit_feedback==0.1.3
6
  uuid==1.30
7
  langdetect==1.0.9
8
  langcodes==3.4.0
9
- vectara-agentic==0.1.27
 
6
  uuid==1.30
7
  langdetect==1.0.9
8
  langcodes==3.4.0
9
+ vectara-agentic==0.2.0