~h-ddlZddlmZmZddlmZddlmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZddlmZmZerddlZddl m Z dd lmZd gZeeZeGd d ZeGd dZeGddeZeGddeZeGddeZeGddeZ eGddeZ!Gdd Z"GddZ#y)N)asdict dataclass) getLogger) TYPE_CHECKINGAnyCallableType) ChatMessage MessageRole) AgentLoggerLogLevelTiming TokenUsage) AgentErrormake_json_serializable)r )r AgentMemoryc2eZdZUeed<eed<eed<dZy)ToolCallname argumentsidcb|jd|jt|jddS)Nfunction)rr)rtyper)rrrrselfs M/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/memory.pydictz ToolCall.dicts.'' 3DNNC  N)__name__ __module__ __qualname__str__annotations__rrrrrrs IN G rrc*eZdZdZddedeefdZy) MemoryStepct|SN)rrs rrzMemoryStep.dict+s d|r summary_modereturnctr))NotImplementedErrorrr*s r to_messageszMemoryStep.to_messages.s!!rNF)r r!r"rboollistr r/r%rrr'r')s""k9J"rr'c8eZdZUeed<eed<dZeedzed<dZ ee dzed<dZ e dzed<dZ edzed<dZeeeeefzdzed<dZedzed <dZedzed <dZed dzed <dZeed <dZedzed<dZeed<dZddedeefdZy) ActionStep step_numbertimingNmodel_input_messages tool_callserrormodel_output_message model_output code_action observationsPIL.Image.Imageobservations_images action_output token_usageFis_final_answerc|j|jj|j|jr(|jDcgc]}|jc}ng|j r|j jnd|j r|j jnd|j|j|j|jr(|jDcgc]}|jc}ndt|j|jrt|jnd|j d Scc}wcc}w)N) r5r6r7r8r9r:r;r<r=r?r@rArB)r5r6rr7r8r9r:r;r<r=r?tobytesrr@rArrB)rtcimages rrzActionStep.dictBs ++kk&&($($=$=CG??t?2779?XZ*.**TZZ__&$HLHaHaD$=$=$B$B$Dgk --++ --''BFAYAY#ZEMMO#Z3D4F4FG7;7G7G6$"2"23T#33  @ $[s E .Er*r+cg}|jH|sF|jttjd|jj dg|j `|jttjddt|j Dcgc]}|jc}zdg|jrE|jttj|jDcgc]}d|d c}|j;|jttjdd|jdg|jydt|jzd z}|j rd |j d jd nd }||z }|jttjd|dg|Scc}wcc}w)NtextrrHrolecontentzCalling tools: rFrrFz Observation: zError: z Now let's retry: take care not to repeat previous errors! If you have retried several times, try a completely different approach. z Call id: r )r;appendr r ASSISTANTstripr8 TOOL_CALLr#rr?USERr= TOOL_RESPONSEr9r)rr*messagesrErF error_messagemessage_contents rr/zActionStep.to_messagesVs    ( OO!6!6&Z^ZkZkZqZqZsAt@uv  ?? & OO$..%+$6RVRaRa=bBbggi=b9c$c   # # OO$)) &*%=%=  "%,%*      ( OO$22%+&4T5F5F4G$H  :: !djj/"ZZ  HL $//!*<*?*?)@C\^O } ,O OO!:!:f^mEnDop Y>cs G"< G'r0)r r!r"intr$rr7r2r r8rr9rr:r;r#rrr<r=r?r@rArrBr1r/r%rrr4r42s N59${+d29(,JX%,#E: #/3+,36:L#T#s(^,,t3:"Kt"#L#*#:>/047>M3%)Kd")!OT! (::k9J:rr4cheZdZUeeed<eed<eed<eed<dZe dzed<d de deefd Z y) PlanningStepr7r:planr6NrAr*r+c|rgSttjd|jj dgttj dddggS)NrHrIrJz$Now proceed and carry out this plan.)r r rQr\rRrTr.s rr/zPlanningStep.to_messagessX I [22fVZV_V_VeVeVg=h task_imagesr*r+cdd|jdg}|jr,|j|jDcgc]}d|d c}ttj |gScc}w)NrHz New task: rIrFrMrJ)r`raextendr r rT)rr*rLrFs rr/zTaskStep.to_messagessa"k$)),EFG    NN4K[K[\%Wu=\ ]!1!17CDD]s A(r0) r r!r"r#r$rar2r1r r/r%rrr_r_s7 I26K'(4/6EEk9JErr_c0eZdZUeed<ddedeefdZy)SystemPromptStep system_promptr*r+c^|rgSttjd|jdggS)NrHrIrJ)r r SYSTEMrfr.s rr/zSystemPromptStep.to_messagess/ I!3!3vW[WiWi>j=klmmrNr0) r r!r"r#r$r1r2r r/r%rrreres#nnk9JnrreceZdZUeed<y)FinalAnswerStepoutputN)r r!r"rr$r%rrrjrjs KrrjcdeZdZdZdefdZdZdeefdZ deefdZ d de d e fd Z defd Zy )raMemory for the agent, containing the system prompt and all steps taken by the agent. This class is used to store the agent's steps, including tasks, actions, and planning steps. It allows for resetting the memory, retrieving succinct or full step information, and replaying the agent's steps. Args: system_prompt (`str`): System prompt for the agent, which sets the context and instructions for the agent's behavior. **Attributes**: - **system_prompt** (`SystemPromptStep`) -- System prompt step for the agent. - **steps** (`list[TaskStep | ActionStep | PlanningStep]`) -- List of steps taken by the agent, which can include tasks, actions, and planning steps. rfc4t||_g|_y)N)rf)rerfsteps)rrfs r__init__zAgentMemory.__init__s/?m/\AC rcg|_y)zKReset the agent's memory, clearing all steps and keeping the system prompt.N)rnrs rresetzAgentMemory.resets  rr+c |jDcgc]:}|jjDcic]\}}|dk7s ||c}}<c}}}Scc}}wcc}}}w)zVReturn a succinct representation of the agent's steps, excluding model input messages.r7)rnritems)rstepkeyvalues rget_succinct_stepszAgentMemory.get_succinct_stepss[mqlvlv  dh$))+*;*;*= _JCH^A^S%Z _  _ s&A AA AAct|jdk(rgS|jDcgc]}|jc}Scc}w)zRReturn a full representation of the agent's steps, including model input messages.r)lenrnrrrts rget_full_stepszAgentMemory.get_full_stepss5 tzz?a I(, 3 333sAloggerdetailedc|jjd|jd|jjtj |j D]}t|tr-|j|jdtj At|tr|jd|jtj |r7|j+|j|jtj |j |jd|j tj t|t"s|jd tj |r7|j+|j|jtj |jd|j$tj y) aXPrints a pretty replay of the agent's steps. Args: logger (`AgentLogger`): The logger to print replay logs to. detailed (`bool`, default `False`): If True, also displays the memory at each step. Defaults to False. Careful: will increase log length exponentially. Use only for debugging. zReplaying the agent's steps:z System prompt)titlerLlevelrO)rzStep Nz Agent output:z Planning step)consolelog log_markdownrfr ERRORrn isinstancer_log_taskr`r4log_ruler5r7 log_messagesr;r[r\)rr|r}rts rreplayzAgentMemory.replaysf 9:/4;M;M;[;[ckcqcqrJJ dD$) 2X^^DD*-%(8(8'9 :(..Q 9 9 E''(A(A'X$$0''otGXGX`h`n`n'oD,/x~~F 9 9 E''(A(A'X##/499T\TbTb#c drc dj|jDcgc],}t|ts|j!|j.c}Scc}w)zQReturns all code actions from the agent's steps, concatenated as a single script.z )joinrnrr4r<rzs rreturn_full_codezAgentMemory.return_full_codesC{{*.** v$ 48TY]YiYiYuT   v  vsA AANr0)r r!r"__doc__r#rorqr2rrwr{r r1rrr%rrrrs\ DcD DJ 4T 4 d[dDd2 # rc2eZdZdZdZdeedefdZdZ y)CallbackRegistryzRegistry for callbacks that are called at each step of the agent's execution. Callbacks are registered by passing a step class and a callback function. ci|_yr)) _callbacksrs rrozCallbackRegistry.__init__ s BDrstep_clscallbackcz||jvrg|j|<|j|j|y)zRegister a callback for a step class. Args: step_cls (Type[MemoryStep]): Step class to register the callback for. callback (Callable): Callback function to register. N)rrP)rrrs rregisterzCallbackRegistry.registers5 4?? *(*DOOH % !((2rc |jjD]_}|jj|gD]>}t t j |jdk(r||n||fi|@ay)aCall callbacks registered for a step type. Args: memory_step (MemoryStep): Step to call the callbacks for. **kwargs: Additional arguments to pass to callbacks that accept them. Typically, includes the agent instance. Notes: For backwards compatibility, callbacks with a single parameter signature receive only the memory_step, while callbacks with multiple parameters receive both the memory_step and any additional kwargs. N) __class____mro__rgetryinspect signature parameters)r memory_stepkwargsclscbs rrzCallbackRegistry.callbacksw((00 mCoo))#r2 m#&w'8'8'<'G'G#HA#M;SUVaSlekSl m mrN) r r!r"rror r'rrrr%rrrrs+ E 3j!1 3X 3mrr)$r dataclassesrrloggingrtypingrrrr smolagents.modelsr r smolagents.monitoringr r rrsmolagents.utilsrr PIL.ImagePIL__all__r r|rr'r4r[r_rerjrrr%rrrs)556KK?-1 / 8         "" " ]] ]@  :   &  Ez E  E nzn n j ? ? D$m$mr