Spaces:
Running
Running
bump to 0.2.0
Browse files- agent.py +2 -4
- 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 |
-
|
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 |
-
'
|
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.
|
|
|
6 |
uuid==1.30
|
7 |
langdetect==1.0.9
|
8 |
langcodes==3.4.0
|
9 |
+
vectara-agentic==0.2.0
|