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."""
|
"""Construye el grafo de estado completo."""
|
||||||
agent_builder = StateGraph(MessagesState)
|
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("llm_call", self._llm_call_node)
|
||||||
agent_builder.add_node("tool_node", self._tool_node)
|
agent_builder.add_node("tool_node", self._tool_node)
|
||||||
|
|
||||||
agent_builder.add_edge(START, "llm_call")
|
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(
|
agent_builder.add_conditional_edges(
|
||||||
"llm_call", self._should_continue, {"tool_node": "tool_node", END: END}
|
"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