SgddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z eGddZ Gdd e ZGd d eZy ) ) dataclass)Type) AgentOutput) ActionModel) BaseModel ConfigDictField create_modelcTeZdZUeed<eed<eed<eed<eed<eed<eed<y) CustomAgentStepInfo step_number max_stepstask add_infosmemory task_progress future_plansN)__name__ __module__ __qualname__int__annotations__strF/Users/kosar/src/browser-use-projects/web-ui/src/agent/custom_views.pyr r s(N IN Krr cNeZdZUdZeed<eed<eed<eed<eed<eed<y) CustomAgentBrainzCurrent state of the agentprev_action_evaluationimportant_contentsrrthoughtsummaryN)rrr__doc__rrrrrrrs'$ L LrrcdeZdZUdZedZeed<ee ed<e de e de dfdZ y ) CustomAgentOutputzOutput model for agent @dev note: this model is extended with custom actions in AgentService. You can also use some fields that are not in this model as provided by the linter, as long as they are registered in the DynamicActions model. T)arbitrary_types_allowed current_stateactioncustom_actionsreturnchtdtt|tdftjS)z"Extend actions with custom actionsr%.)__base__r(r)r r%listr r)r)s rtype_with_custom_actionsz*CustomAgentOutput.type_with_custom_actions*s6  &^$c )33  rN) rrrr#r model_configrrr-r staticmethodrr.rrrr%r%sQ d;L##    [)  ! "    rr%N) dataclassesrtypingrbrowser_use.agent.viewsr%browser_use.controller.registry.viewsrpydanticrrr r r rr%rrrr6sH!/=??  y  r