Spaces:
Running
Running
updates
Browse files
agent.py
CHANGED
@@ -24,7 +24,7 @@ def create_assistant_tools(cfg):
|
|
24 |
)
|
25 |
|
26 |
summarizer = 'vectara-summary-table-md-query-ext-jan-2025-gpt-4o'
|
27 |
-
|
28 |
tool_name = "ask_cona",
|
29 |
tool_description = """
|
30 |
Given a user query,
|
@@ -38,12 +38,12 @@ def create_assistant_tools(cfg):
|
|
38 |
include_citations = True,
|
39 |
verbose = False
|
40 |
)
|
41 |
-
return [
|
42 |
|
43 |
def initialize_agent(_cfg, agent_progress_callback=None):
|
44 |
bot_instructions = """
|
45 |
- You are a helpful assistant, with expertise in products from coca cola and other bottling companies.
|
46 |
-
- Use the
|
47 |
"""
|
48 |
|
49 |
agent = Agent(
|
|
|
24 |
)
|
25 |
|
26 |
summarizer = 'vectara-summary-table-md-query-ext-jan-2025-gpt-4o'
|
27 |
+
ask_cona = vec_factory.create_rag_tool(
|
28 |
tool_name = "ask_cona",
|
29 |
tool_description = """
|
30 |
Given a user query,
|
|
|
38 |
include_citations = True,
|
39 |
verbose = False
|
40 |
)
|
41 |
+
return [ask_cona] + ToolsFactory().guardrail_tools()
|
42 |
|
43 |
def initialize_agent(_cfg, agent_progress_callback=None):
|
44 |
bot_instructions = """
|
45 |
- You are a helpful assistant, with expertise in products from coca cola and other bottling companies.
|
46 |
+
- Use the ask_cona tool to answer most questions about any products related to coca cola.
|
47 |
"""
|
48 |
|
49 |
agent = Agent(
|