uPj.UdZddlZddlZddlmZddlmZmZddlm Z Gdde eZ de e effd Z de e effd Ze Zeed <Gd d Zdee dzdefdZdee dzdee fdZeZeed<y)a Automatic Code Execution Handler for LiteLLM Skills When `litellm_code_execution` tool is present, this handler automatically: 1. Makes the LLM call 2. Executes any code the model generates 3. Continues the conversation with results 4. Returns final response with generated files inline (base64) This mimics Anthropic's behavior where code execution happens automatically. Generated files are returned directly in the response - no separate storage needed. N)Enum)AnyFinal)verbose_loggerceZdZdZdZy)LiteLLMInternalToolsz Enum for internal LiteLLM tools that are injected into requests. These tools are handled automatically by LiteLLM hooks and are not passed to the underlying LLM provider directly. litellm_code_executionN)__name__ __module__ __qualname____doc__CODE_EXECUTIONu/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/litellm_proxy/skills/code_execution.pyrrs.NrrreturncXdtjjddddddidgdd d S) z Returns the litellm_code_execution tool definition in OpenAI format. This tool enables automatic code execution in a sandboxed environment when skills include executable Python code. functionExecute Python code in a sandboxed environment. Use this to run code that generates files, processes data, or performs computations. Generated files will be returned directly.objectcodestringPython code to executetype descriptionr propertiesrequired)namer parameters)rrrrvaluerrrget_litellm_code_execution_toolr$!sF(77==M %Ia'bc#H   rcRtjjddddddidgddS) z Returns the litellm_code_execution tool definition in Anthropic/messages API format. This tool enables automatic code execution in a sandboxed environment when skills include executable Python code. rrrrrrr)r r input_schemar"rrr)get_litellm_code_execution_tool_anthropicr'6s>%3399I!HE]#^_  rLITELLM_CODE_EXECUTION_TOOLc|eZdZdZ d dedzdedzfdZ ddedeedeed eee fd edzd eee ff d Z y)CodeExecutionHandlerz Handles automatic code execution for LiteLLM skills. When enabled, this handler intercepts LLM responses with code execution tool calls, executes them in a sandbox, and continues the conversation automatically until completion. Nmax_iterationssandbox_timeoutc@ddlm}m}|xs||_|xs||_y)Nr)DEFAULT_MAX_ITERATIONSDEFAULT_SANDBOX_TIMEOUT)+litellm.llms.litellm_proxy.skills.constantsr.r/r+r,)selfr+r,r.r/s r__init__zCodeExecutionHandler.__init__Us%  -F0F.I2Irmodelmessagestools skill_filesskill_idrc Kddl}ddlm}t|} g} g} ||j} d} t |j D]}tjd|dz|j |jd.|| |d|d{} | jdj}| jdj}d|jd }|jrV|jDcgc]=}|jd |j j"|j j$d d ?c}|d <| j'||d k7s |js"tjd|dz| | | | dcS|jD]}|j j"}|t(j*j,k(r t/j0|j j$}|j3dd}tjdt5|| j7||}tjd|| j'||d|d|d|dDcgc]}|d c}d|dxsd}|dr|dz }|dD]}}t9j:|d}| j'|d|d|dt5|d|d|dd t5|d!z }tjd"|dt5||dr |d#|dz }| j'd'|j|d(| j'd'|jd)|d*d(tj@d+|j | | | | d,d-S7cc}wcc}w#t<$r-}d$|}| j'|d%t?|d&Yd}~d}~wwxYww)/a Execute an LLM call with automatic code execution handling. This method: 1. Makes the initial LLM call 2. If model calls litellm_code_execution, executes the code 3. Continues conversation with results 4. Repeats until model stops calling tools 5. Returns final response with generated files inline Args: model: Model to use messages: Initial messages tools: Tools including litellm_code_execution skill_files: Dict of skill files for execution skill_id: Optional skill ID for tracking **kwargs: Additional args for litellm.acompletion Returns: Dict with: - response: Final LLM response - files: List of generated files with content (base64) - execution_results: List of code execution results rN)SkillsSandboxExecutor)timeoutz%CodeExecutionHandler: Iteration %s/%s)r3r4r5 assistant)rolecontentr)r arguments)idrr tool_callsz3CodeExecutionHandler: Completed after %s iterations)responsefilesexecution_resultsr4rz/CodeExecutionHandler: Executing code (%s chars))rr6z*CodeExecutionHandler: Execution result: %ssuccessoutputerrorrCr ) iterationrFrGrHrCz Generated files:content_base64 mime_type)r rKrJsizez - z (z bytes)z2CodeExecutionHandler: Generated file %s (%s bytes)z Error: zCode execution failed: F)rIrFrHtool)r= tool_call_idr>zTool 'z'' not handled by code execution handlerz1CodeExecutionHandler: Max iterations (%s) reachedT)rBrCrDr4max_iterations_reachedr)!litellm2litellm.llms.litellm_proxy.skills.sandbox_executorr9listr,ranger+rdebug acompletionchoicesmessage finish_reasonr>rAr@rr r?appendrrr#jsonloadsgetlenexecutebase64 b64decode Exceptionstrwarning)r1r3r4r5r6r7kwargsrPr9current_messagesgenerated_filesrDexecutorrBrIassistant_message stop_reasonassistant_msg_dicttc tool_call tool_nameargsr exec_resultf tool_result file_contentes rexecute_with_code_executionz0CodeExecutionHandler.execute_with_code_executionbs\B  #'x.79/1/8L8LMt223} I  !H)VW-Y]YlYl m1W00) H!) 0 0 3 ; ; "**1-;;K$,442 !++0:: 4!ee *$&KK$4$4)+)>)>% 4"<0  # #$6 7l*2C2N2N$$%Z\ehi\ij (,): 0 /99Q  %..33  4 C C I II9#zz)*<*<*F*FG#xx3&,,-^`cdh`ij&.&6&6!%(3'7' ',,-Y[fg)00-6+6y+A*5h*?)4W)==H=Q)R!F))R '2(&;&Ar 'w/'+AAK%0%9"/5/?/?BR@S/T / 6 601& 56{^:;rs  +.3.c3h*4S>&&E%FUFJ J Z4:#44:#4d!5 66r