o $iQ@shdZddlZddlZddlZddlmZmZmZmZm Z ddl m Z m Z m Z eeZGdddZdS)z" OpenRouter Agent implementation. N)AnyCallableDictListOptional)MessageRoleToolc@s6eZdZdZ      d-ded ed ed ed ed edeeee ffddZ dededeee fde fddZ de ddfddZ d.deeee fdeeeee fdeee ffddZdeedeefddZd eddfd!d"Z  #d/d$ed%eed&edefd'd(Zdeeee ffd)d*Zd0d+d,ZdS)1OpenRouterAgentz:AI Agent using OpenRouter API with tool calling capabilityqwen/qwen3-coder-nexthttps://openrouter.ai/api/v1ffffff? Napi_keymodelbase_url temperature max_tokensmax_iterationstoolscCsT||_||_||_||_||_||_|r|ni|_g|_d|jdddd|_dS)a Initialize the OpenRouter agent. Args: api_key: OpenRouter API key model: Model identifier (default: "qwen/qwen3-coder-next") base_url: OpenRouter API base URL temperature: Sampling temperature max_tokens: Maximum tokens to generate max_iterations: Maximum tool-calling iterations tools: Optional dictionary of pre-registered tools zBearer zapplication/jsonzhttps://github.comzAI Agent) Authorizationz Content-Typez HTTP-RefererzX-TitleN) rrrrrrrconversation_historyheaders)selfrrrrrrrr0/var/www/apps/myagent/mysuperagent/core/agent.py__init__s  zOpenRouterAgent.__init__name description parametersreturncs dtdtffdd }|S)z%Decorator to register a tool functionfuncr!cst|d}|j<|S)N)rrr function)r r)r"toolrrr rrr decorator?s z0OpenRouterAgent.register_tool..decorator)r)rrrr r&rr%r register_tool8s zOpenRouterAgent.register_toolr$cCs||j|j<dS)zAdd a tool to the agentN)rr)rr$rrradd_toolJzOpenRouterAgent.add_toolmessagescCsL|j||j|jd}|r||d<tj|jd|j|dd}||S)zMake API call to OpenRouter)rr*rrrz/chat/completions)rjsontimeout) rrrrequestspostrrraise_for_statusr,)rr*rpayloadresponserrr_make_api_callNs zOpenRouterAgent._make_api_call tool_callsc Csg}|D]S}|dd}t|dd}||jvrC|j|j}z |d i|}WntyB}z dt|}WYd}~n d}~wwd|}|ttj t||ddq|S) z2Process tool calls and return tool result messagesr#r argumentszError: NzError: Unknown tool id)rolecontent tool_call_idr) r,loadsrr# ExceptionstrappendrrTOOL) rr4 tool_messages tool_call func_name func_argsr"resulterrr_process_tool_callsgs*    z#OpenRouterAgent._process_tool_callsmessagecCs|j|dS)z)Add a message to the conversation historyN)rr=)rrFrrr add_messager)zOpenRouterAgent.add_messageF user_input system_promptreset_conversationcCs`|rg|_|rtdd|jDs|ttj|d|ttj|dd}d}||jkr|d7}t d|d|jd d |jD}d}|j rUd d |j D}| ||}|d d} | d } d| vr| drttj | d| dd} || || d} | D]} || qq)ttj | ddd} || | dd} |durd}|S)aF Main agentic loop: Process user input and handle tool calls Args: user_input: User's message system_prompt: Optional system prompt reset_conversation: Whether to reset conversation history Returns: Final assistant response css|] }|jtjkVqdSN)r7rSYSTEM).0mrrr sz&OpenRouterAgent.run..)r7r8rNz Iteration /cSg|]}|qSrto_dictrMmsgrrr z'OpenRouterAgent.run..cSrRrrS)rMr$rrrrWrXchoicesrFr4r8)r7r8r4z2Maximum iterations reached without final response.)ranyrGrrrLUSERrloggerinforvaluesr3 ASSISTANTgetrE)rrHrIrJ iterationfinal_responser* tools_listr2choicerFassistant_messager?tool_msgrrrrunsL        zOpenRouterAgent.runcCsdd|jDS)z1Get the conversation history in a readable formatcSrRrrSrUrrrrWrXz.rrrrrget_conversation_historyr)z(OpenRouterAgent.get_conversation_historycCs g|_dS)zClear the conversation historyNrirjrrr clear_historys zOpenRouterAgent.clear_history)r r r rrNrK)NF)r!N)__name__ __module__ __qualname____doc__r<floatintrrr rrrr'r(rr3rrErGboolrhrkrlrrrrr sp &     Ir )rpr,loggingr.typingrrrrr core.modelsrrr getLoggerrmr]r rrrrs