Wi" ddlZddlmZmZddlZejejdejeZ GddZ dS)N)DictAnyz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatcheZdZdefdZdZdZdZdefdZ dZ d Z d ed e ee fdefd Zd S)AgrotaMcpClient server_urlc~|dd|_tjd|_d|_dS)Nz/ssez/mcpg^@)timeoutF)replacer httpx AsyncClientclient _is_connected)selfr s yc:\Users\Dell Inspiron 16\Desktop\tws\github tws\Agrota\back-agrota-powerbi\mcp-client-agrota\app\services\mcp_service.py__init__zAgrotaMcpClient.__init__ s<$,,VV<<'666 "c>K|d{V|SN)connectrs r __aenter__zAgrotaMcpClient.__aenter__s+llnn rc>K|d{VdSr) disconnect)rexc_typeexc_valexc_tbs r __aexit__zAgrotaMcpClient.__aexit__s.oorcK|jri |jd{Vn4#t$r'}td|Yd}~nd}~wwxYwd|_n #d|_wxYwtddS)zCierra el cliente HTTP.NzError al cerrar cliente HTTP: Fz Desconectado.)raclose Exceptionloggererrorrinfo)res rrzAgrotaMcpClient.disconnects ; + +k((********** C C C AaAABBBBBBBB C&+""U"**** O$$$$$s++A' AAA'AA'' A0returnc|jS)u$Verifica si el cliente está activo.)rrs r is_connectedzAgrotaMcpClient.is_connected$s !!rcnK|jrdStd|jd|_dS)uFMarca el cliente como conectado (sin verificación real de conexión).NzCliente HTTP listo para T)rr#r%r rs rrzAgrotaMcpClient.connect(sA    F @t@@AAA!rcxKddlm}|ddg}|dddg}||dd g S) 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_tools0s{ +*****J{WI66 z&6="9::y* v    r tool_name argumentsc@K|dkrtd| td|j|j|jd|ddiddid {V}|jd kr*|}|d d St d |jd|j d|jS#t$r,}t d|d|cYd }~Sd }~wwxYw)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@sj ) ) )F9FFGG G < KKPtPP Q Q Q![-- h ; ;<');<.H #s**}}xx/::: Vx/CVVx}VVWWWDh.BDDD < < < LLDDD E E E;;; ; ; ; ; ; ; rQs  - V  8 $ $J<J<J<J<J<J<J<J<J<J