Spaces:
Running
Running
update
Browse files
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 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
|
|
|
|
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.
|