gig0TddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddl m!Z!ee!jDde!jFZ$dZ%defdZ&dededfdZ'defdZ(defdZ)defdZ*defdZ+dededfdZ,e eZ-e-j]de&e-j]de(e-j]de)e-j]dee-j]d ee-j]d!ee-j]d"ee-j]d#e*e-j]d$e+e-j_de-jade'dede-jcdddd d!d"d$dZ2e-jade,e2e-jade,e2e-jad e,e2e-jad!e,e2e-jad"e,e2e-jcd$ee-jgZ4y)%N)Literal) ChatOpenAI) SystemMessage AIMessage) StateGraphEND) AgentState) SYSTEM_PROMPTget_classification_prompt)ub_presupuesto_skill_node)cartera_skill_node)talento_humano_skill_node)resumen_pdf_skill_node)get_active_archetypesget_cached_resultsave_archetype_to_backend)settings)model temperatureapi_keyc|jjdi}|jdd|jdd|jdddS)zExtrae el consumo de tokens. token_usage prompt_tokensrcompletion_tokens total_tokens)prompt completiontotal)response_metadataget)messageusages -/opt/agrota/powerbi/client-mcp/agent/graph.py extract_usager%sO  % % ) )- 1- statecK|jdg}|sddiS|dj}tj}|jd}|jdd}|r ||z dkrddiS|rctjd|j }|r-t d|dd |jd td gd Sdtd gdSt ddtd gdSw)uUVerifica si la sesión de 12 horas ha caducado y solicita la cédula si es necesario.messagesis_authenticatingFauth_validated_atiz^\d{10}$u;🔒 [SECURITY] Cédula validada. Sesión iniciada por 12h.securityru@✅ Acceso validado correctamente. ¿En qué puedo ayudarte hoy?content)r,r* intencion user_cedular)Tuk⚠️ El número de cédula ingresado no es válido (debe tener 10 dígitos). Por favor, intenta de nuevo.)r*r)uD🔒 [SECURITY] Sesión expirada o inexistente. Solicitando cédula.uuHola, por motivos de seguridad tu sesión ha expirado (12h). Por favor, ingresa tu número de cédula para continuar.) r!r/timerematchstripprintgroupr)r'r)last_msgnowlast_valis_auth cedula_matchs r$security_check_noder=&syyR(H #U++|##H ))+Cyy,-Hii+U3GS8^e+#U++xx X^^-=>  O P%(%*'+11!4&/qrs  &*&0]^_   PQ!(_`a sC-C/return)classifyrcJ|jds|jdsyy)uBDecide si continuar con el flujo o detenerse a esperar la cédula.r*r,rr?)r!)r's r$security_routerrAOs! yy$%UYY7J-K r&cK|d}|sdddStd{}|r+dj|Dcgc]}d|dd|d c}nd }|d d}tjt t | g|d{} |j j}tjd |tj}|r$tj|jdni} tt| j!dg} | s#| j!dr| j!dg} | j!di} | | j#| r| dnd| j!d| | j!dxs|j!d| j!dxs|j!dt%|dgdddddS7cc}w7R#t&$r'} t)d| ddt%|dcYd} ~ Sd} ~ wwxYww)uHNodo para clasificar la intención y detectar arquetipos pre-calculados.r) GENERAL_CHATN)r0 archetype_key z- archetypeKeyz: sampleQuestionzNinguno disponibler.z\{.*\}r intencionesr0entitiesUNKNOWNrD user_name user_positionF)rIintenciones_pendientesr0rD last_entitiesrLrMr#is_precalculatedresultados_tecnicos mensaje_skill dashboard dax_queryinteractive_datau⚠️ Error clasificación: )r0rDr#)rjoinllmainvokerr r/r5r3searchDOTALLjsonloadsr7listsetr!copyr% Exceptionr6) r'r) archetypesaarchetypes_txtrecent_messagesresponser/ json_matchdatarIrJes r$classify_intent_noderjUsZ H +dCC-..JisTYYYcdTU"Q~%6$7r!X"8, %#% $  +/dB a -aS12& V^H_``as]H G H GH HH c K|jd}|sddiStd|t|d{}|r5tdtj|dd|jd d Std ddiS7Jw) zRVerifica si existe un resultado pre-calculado en Java para el arquetipo detectado.rDrQFu0🔍 [CACHE] Buscando resultado para arquetipo: Nu8✅ [CACHE] Hit! Usando resultado pre-calculado de Java.) ensure_asciiTr0)ultimo_resultadorQr0u1❌ [CACHE] Miss. Procediendo a consulta en vivo.)r!r6rr\dumps)r'rD cache_results r$check_cache_noderpsIIo.M "E** <]O LM*=99L HI!%  &&:s4BBA BcK|jds|jdr|jdrgd}|jd|vriSd|jddjvriS|dd j}|jd}|jd}td |d d d t |||d {iS7w)u[Nodo que registra automáticamente consultas exitosas como nuevos arquetipos en el backend.rQrmrU)UB_PRESUPUESTOCARTERATALENTO_HUMANOr0u¡HOLArNr)r+u@🧠 [LEARNING] Auto-registrando nueva consulta como arquetipo: N2z...)r!upperr/r6r)r'allowed_intentsquestiondaxcategorys r$register_archetype_noder{s 99' (UYY7I-JuyyYdOeJ 99[ ! 8I uyyb1779 9I$R(00ii $99[) PQYZ][]Q^P__bcd'#x@@@ I AsCC C CcK|d}tjd}tjj|j d}t j |}d|jdd}|jdr|d |jddz }|jd r|d |jd dz }|jd g}|r(|d z }t|D]\}}|d|dzd|dz }n(|jdr|d|jdz }tjt|g|ddtd|d{} t| jgt| dS7'w)z@Genera la respuesta final persuasiva y ejecutiva usando Felizia.r)zAmerica/Guayaquilz%Y-%m-%d) current_dateuIntención Detectada: r0rErLz Usuario: rMzCargo: rRu$Datos Técnicos Consolidados (MCP): z--- Resultado r z --- rmuDato Técnico (MCP): r.iNuContexto técnico actual: )r)r#)pytztimezonedatetimer9strftimer formatr! enumeraterXrYrrr/r%) r'r)tzr} sys_promptcontext resultadosires response_objs r$ response_nodersZ H * +B$$((,55jAL%%<@J&uyy'=&>bAG yyYuyy56b99 yy!WUYY78;;0"5J:: + ;FAs !uF3%r: :G ; % &*5995G+H*IJJj)& 34.&  ;G9EF&L|';';<=|,  sE!F #F $(F )ub_presupuesto_skill cartera_skilltalento_humano_skillresumen_pdf_skillrc|jdry|jdg}|s tdy|d}td|d||dk(ry |d k(ry |d k(ry |dk(ryy)zARouter secuencial que maneja la cola de intenciones (Dashboards).rQrrOuE🏁 [ROUTER] No hay más tareas pendientes. Yendo a respuesta final.ru!🔄 [ROUTER] Tareas pendientes: z. Ejecutando: rrrrsrrtrRESUMEN_EJECUTIVO_PDFr)r!r6)r' pendientesproximas r$unified_routerrs yy#$3R8J  UVmG -j\y QR""% I  $ $% + +" r&security_checkr? check_cacherrrrregister_archetyper)5r\rr~r3r2typingrlangchain_openairlangchain_core.messagesrrlanggraph.graphrrr'r promptsr r skills.ub_presupuesto_skillr skills.cartera_skillrskills.talento_humano_skillrskills.resumen_pdf_skillrutils.precalc_helperrrrconfig.settingsr MODEL_NAMEOPENAI_API_KEYrXr%r=rArjrpr{rrworkflow_builderadd_nodeset_entry_pointadd_conditional_edgesadd_edgedestinos_routercompileworkflow_compiledr&r$rs\  '<+=B4B<ee$     # # 'Z'R:'2C*D .aj.a`'*',. z F*2W*X<j)*,?@*&:;-)9:02KL/+=>02KL-/EF.0GH/=9 !12&&'7 K *m43$2,$ &&}noV&&'=~_&&X&&'=~_&&':NO\/3/$,,.r&