ofermend commited on
Commit
b8315ca
·
1 Parent(s): 8ca5093
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -16,7 +16,7 @@ load_dotenv(override=True)
16
 
17
  from pydantic import Field, BaseModel
18
  from vectara_agent.agent import Agent, AgentStatusType
19
- from vectara_agent.tools import ToolsFactory
20
  from vectara_agent.tools_catalog import summarize_text
21
 
22
  initial_prompt = "How can I help you today?"
@@ -35,10 +35,12 @@ def create_tools(cfg):
35
  class QueryHackerNews(BaseModel):
36
  query: str = Field(..., description="The user query.")
37
 
38
- tools_factory = ToolsFactory(vectara_api_key=cfg.api_key,
39
- vectara_customer_id=cfg.customer_id,
40
- vectara_corpus_id=cfg.corpus_id)
41
- ask_hackernews = tools_factory.create_rag_tool(
 
 
42
  tool_name = "ask_hackernews",
43
  tool_description = """
44
  Responds to query based on information and stories in hacker news from the last 6-9 months.
 
16
 
17
  from pydantic import Field, BaseModel
18
  from vectara_agent.agent import Agent, AgentStatusType
19
+ from vectara_agent.tools import ToolsFactory, VectaraToolFactory
20
  from vectara_agent.tools_catalog import summarize_text
21
 
22
  initial_prompt = "How can I help you today?"
 
35
  class QueryHackerNews(BaseModel):
36
  query: str = Field(..., description="The user query.")
37
 
38
+ vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_key,
39
+ vectara_customer_id=cfg.customer_id,
40
+ vectara_corpus_id=cfg.corpus_id)
41
+ tools_factory = ToolsFactory()
42
+
43
+ ask_hackernews = vec_factory.create_rag_tool(
44
  tool_name = "ask_hackernews",
45
  tool_description = """
46
  Responds to query based on information and stories in hacker news from the last 6-9 months.