fix: use TestClient as context manager to trigger lifespan

This commit is contained in:
2026-03-18 15:05:01 -05:00
parent 9b1108da5c
commit ea3da00b95
5 changed files with 15 additions and 10 deletions

View File

@@ -11,7 +11,7 @@ from .schemes import ScheduleSchema
TIMEZONE = "America/Bogota"
MCP_SERVER_URL = "http://localhost:8001/mcp"
MCP_SERVER_URL = "http://192.168.58.109:3001/mcp"
_MCP_CLIENT = Client(MCP_SERVER_URL)
@@ -76,10 +76,10 @@ class NaliiaTools:
@tool
def find_customer_by_identifier(identifier: str) -> str:
"""
Busca un cliente por su identificador (teléfono, email o número de documento).
Busca un cliente por su identificador (teléfono o número de documento).
Args:
identifier: Identificador del cliente (teléfono, email o número de documento).
identifier: Identificador del cliente (teléfono o número de documento).
Returns:
str: JSON con la información del cliente encontrado.
@@ -97,6 +97,7 @@ class NaliiaTools:
return result
result = asyncio.run(call_tool())
return result.content[0].text
@tool
@@ -122,6 +123,7 @@ class NaliiaTools:
return result
result = asyncio.run(call_tool())
return result.content[0].text
@tool
@@ -150,6 +152,7 @@ class NaliiaTools:
return result
result = asyncio.run(call_tool())
return result.content[0].text
@tool(args_schema=ScheduleSchema)
@@ -216,6 +219,8 @@ class NaliiaTools:
logger.info(result)
asyncio.run(call_tool())
return result
result = asyncio.run(call_tool())
return True