Update app.py
Browse files
app.py
CHANGED
|
@@ -145,7 +145,7 @@ def ref_context(request: ModelRequest) -> str:
|
|
| 145 |
|
| 146 |
return system_message
|
| 147 |
|
| 148 |
-
contact_agent = (create_agent(
|
| 149 |
|
| 150 |
@tool("contact", description="refer users to WEAI team using links and contact details")
|
| 151 |
def call_contact_agent(query: str):
|
|
@@ -174,7 +174,7 @@ def doc_context(request: ModelRequest) -> str:
|
|
| 174 |
|
| 175 |
return system_message
|
| 176 |
|
| 177 |
-
support_agent = (create_agent(
|
| 178 |
|
| 179 |
|
| 180 |
@tool("support", description="respond to user queries using context in WEAI docs")
|
|
@@ -194,7 +194,7 @@ Steps:
|
|
| 194 |
3. Return only a complete response with included contact and resource information.
|
| 195 |
"""
|
| 196 |
|
| 197 |
-
response_agent = create_agent(model=
|
| 198 |
tools=[call_contact_agent, call_support_agent],
|
| 199 |
system_prompt=support_instructions,
|
| 200 |
)
|
|
|
|
| 145 |
|
| 146 |
return system_message
|
| 147 |
|
| 148 |
+
contact_agent = (create_agent(chat_llm, tools=[], middleware=[ref_context]))
|
| 149 |
|
| 150 |
@tool("contact", description="refer users to WEAI team using links and contact details")
|
| 151 |
def call_contact_agent(query: str):
|
|
|
|
| 174 |
|
| 175 |
return system_message
|
| 176 |
|
| 177 |
+
support_agent = (create_agent(chat_llm, tools=[call_contact_agent], middleware=[doc_context]))
|
| 178 |
|
| 179 |
|
| 180 |
@tool("support", description="respond to user queries using context in WEAI docs")
|
|
|
|
| 194 |
3. Return only a complete response with included contact and resource information.
|
| 195 |
"""
|
| 196 |
|
| 197 |
+
response_agent = create_agent(model=chat_llm,
|
| 198 |
tools=[call_contact_agent, call_support_agent],
|
| 199 |
system_prompt=support_instructions,
|
| 200 |
)
|