Yi" ddlZddlmZmZddlZej ej dejeZ GddZ y)N)DictAnyz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatc`eZdZdefdZdZdZdZdefdZ dZ d Z d ed e ee fdefd Zy )AgrotaMcpClient server_urlcv|jdd|_tjd|_d|_y)Nz/ssez/mcpg^@)timeoutF)replacer httpx AsyncClientclient _is_connected)selfr s :/opt/agrota/powerbi/client-mcp/app/services/mcp_service.py__init__zAgrotaMcpClient.__init__ s0$,,VV<''6 "cBK|jd{|S7wN)connectrs r __aenter__zAgrotaMcpClient.__aenter__slln  s c@K|jd{y7wr) disconnect)rexc_typeexc_valexc_tbs r __aexit__zAgrotaMcpClient.__aexit__soos cK|jr* |jjd{d|_tj dy7!#t$r"}tj d|Yd}~Ed}~wwxYw#d|_wxYww)zCierra el cliente HTTP.NzError al cerrar cliente HTTP: Fz Desconectado.)raclose Exceptionloggererrorrinfo)res rrzAgrotaMcpClient.disconnectsv ;; +kk((***&+" O$ + C =aSABB C&+"sI B AAAB A A<A72A?7A<<A?? BB returnc|jS)u$Verifica si el cliente está activo.)rrs r is_connectedzAgrotaMcpClient.is_connected$s!!!rcxK|jrytjd|jd|_yw)uFMarca el cliente como conectado (sin verificación real de conexión).NzCliente HTTP listo para T)rr#r%r rs rrzAgrotaMcpClient.connect(s3     .t.?@A!s8:cdKddlm}|ddg}|dddg}||dd g Sw) zBLista las herramientas disponibles (simulado para compatibilidad).r) namedtuple ToolsListtoolsToolname descriptionconsultarAgrotazTTransforma una solicitud en lenguaje natural a DAX y consulta el Power BI de Agrota.)r0r1)r.) collectionsr,)rr,r-r/s r list_toolszAgrotaMcpClient.list_tools0sJ +{WI6 &6="9:* v  s.0 tool_name argumentscFK|dk7rtd| tjd|j|jj |jd|j ddiddid {}|jd k(r"|j}|j d d Stjd |jd|jd|jS7s#t$r'}tjd|d|cYd }~Sd }~wwxYww)z$Llama al endpoint HTTP directamente.r2zHerramienta desconocida: u$🌐 Enviando solicitud HTTP POST a promptz Content-Typezapplication/json)jsonheadersNresultz Sin resultadou❌ Error HTTP z: zError del servidor: u%❌ Error llamando al endpoint HTTP: z"Error al ejecutar la herramienta: ) RuntimeErrorr#r%r rpostget status_coder:r$textr")rr5r6responsedatar&s r call_toolzAgrotaMcpClient.call_tool@s ) )!:9+FG G < KK>t>OP Q![[-- h ;<');<.H ##s*}}xx/:: x/C/C.DBx}}oVW-h.B.B-CDD < LL@D E7s; ; rQsH  -- V   8 $J<J