revert: re-enable identify_customer node in agent graph
- Move test.ipynb to tests/notebooks for RPC testing
This commit is contained in:
@@ -135,12 +135,12 @@ class Agent:
|
||||
"""Construye el grafo de estado completo."""
|
||||
agent_builder = StateGraph(MessagesState)
|
||||
|
||||
# agent_builder.add_node("identify_customer", self._identify_customer_node)
|
||||
agent_builder.add_node("identify_customer", self._identify_customer_node)
|
||||
agent_builder.add_node("llm_call", self._llm_call_node)
|
||||
agent_builder.add_node("tool_node", self._tool_node)
|
||||
|
||||
agent_builder.add_edge(START, "llm_call")
|
||||
# agent_builder.add_edge("identify_customer", "llm_call")
|
||||
agent_builder.add_edge("identify_customer", "llm_call")
|
||||
agent_builder.add_conditional_edges(
|
||||
"llm_call", self._should_continue, {"tool_node": "tool_node", END: END}
|
||||
)
|
||||
|
||||
41
tests/notebooks/test.ipynb
Normal file
41
tests/notebooks/test.ipynb
Normal file
@@ -0,0 +1,41 @@
|
||||
{
|
||||
"cells": [
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "f614f47a",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"import asyncio\n",
|
||||
"from fastmcp import Client\n",
|
||||
"\n",
|
||||
"MCP_SERVER_URL = \"http://192.168.58.109:3001/mcp\"\n",
|
||||
"_MCP_CLIENT = Client(MCP_SERVER_URL)\n",
|
||||
"\n",
|
||||
"\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
"metadata": {
|
||||
"kernelspec": {
|
||||
"display_name": "naliiabot-X7BrBtpI-py3.13",
|
||||
"language": "python",
|
||||
"name": "python3"
|
||||
},
|
||||
"language_info": {
|
||||
"codemirror_mode": {
|
||||
"name": "ipython",
|
||||
"version": 3
|
||||
},
|
||||
"file_extension": ".py",
|
||||
"mimetype": "text/x-python",
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.13.11"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 5
|
||||
}
|
||||
Reference in New Issue
Block a user