{hD,ddlZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z mZddlmZddlmZddlmZd Zd Zeee_eee_Gd d eZGd deZGddeZdZedk(rddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddlZddl,Z,ddl-m.Z.ddl/Z/de0deede$fdZ1dede$fdZ2defd Z3d&d!e4fd"Z5e Z6e6jod#e,jpe5d$%yy)'N) BaseModelField)OptionalType GeneratorAsyncGeneratorget_args) ChatHistory)BaseDynamicContextProviderSystemPromptGenerator BaseIOSchema) PartialBase) from_jsonc+Kd}|j}|D]?}||z }t|xsdjd}|j|fddi|}|AywNz{}ztrailing-strings) partial_modestrictget_partial_modelrencodemodel_validatecls json_chunkskwargspotential_object partial_modelchunkobjs b/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/atomic_agents/agents/atomic_agent.pymodel_from_chunks_patchedr#sq))+ME!)1T99;J\]*m**3FtFvF sAAc Kd}|j}|23d{}||z }t|xsdjd}|j|fddi|}|H7C6ywrrrs r"model_from_chunks_async_patchedr%sy))+M"eE!)1T99;J\]*m**3FtFvF {s&A#A!AA!AA#A!!A#c0eZdZUdZeddZeed<y)BasicChatInputSchemaz?This schema represents the input from the user to the AI agent..z3The chat message sent by the user to the assistant. description chat_messageN__name__ __module__ __qualname____doc__rr*str__annotations__r"r'r''sI IL#r3r'c0eZdZUdZeddZeed<y)BasicChatOutputSchemaz@This schema represents the response generated by the chat agent..zThe chat message exchanged between the user and the chat agent. This contains the markdown-enabled response generated by the chat agent.r(r*Nr+r2r3r"r5r50sJ  WL#r3r5c eZdZUeddZejj ed<eddZe ed<ed d Z e e ed <ed d Z e eed <eddZe e ed<ddiZed dZe eed<y ) AgentConfig.z/Client for interacting with the language model.r(client gpt-4o-miniz*The model to use for generating responses.)defaultr)modelNz+History component for storing chat history.historyz(Component for generating system prompts.system_prompt_generatorsystemzHThe role of the system in the conversation. None means no system prompt. system_rolearbitrary_types_allowedTz1Additional parameters passed to the API provider.model_api_parameters)r,r-r.rr8 instructor Instructorr1r;r0r<rr r=r r? model_configrAdictr2r3r"r7r7<s+0Bs+tFJ   ( (t}:fgE3g%*4Er%sGXk "s?D"L@X&;<"'&p"K#.t4L+0Cv+w(4.wr3r7cHddxddxfGfdd }|S) InputSchemactSNr r2r3r"rGrGJs|+r3 OutputSchemactSrIr r2r3r"rJrJJs<-Gr3c"eZdZWZdZdefdZdZede e fdZ ede e fdZ dZ dd eWdWfd Zdd eWdeWd Wffd Zdd eWdWfd Zdd eWdeWd ffdZdede efdZdedefdZdefdZy ) AtomicAgenta Base class for chat agents. This class provides the core functionality for handling chat interactions, including managing history, generating system prompts, and obtaining responses from a language model. Type Parameters: InputSchema: Schema for the user input, must be a subclass of BaseIOSchema. OutputSchema: Schema for the agent's output, must be a subclass of BaseIOSchema. Attributes: client: Client for interacting with the language model. model (str): The model to use for generating responses. history (ChatHistory): History component for storing chat history. system_prompt_generator (SystemPromptGenerator): Component for generating system prompts. system_role (Optional[str]): The role of the system in the conversation. None means no system prompt. initial_history (ChatHistory): Initial state of the history. current_user_input (Optional[InputSchema]): The current user input being processed. model_api_parameters (dict): Additional parameters passed to the API provider. - Use this for parameters like 'temperature', 'max_tokens', etc. configcT|j|_|j|_|jxs t|_|jxs t |_|j |_|jj|_d|_ |jxsi|_ y)z Initializes the AtomicAgent. Args: config (AgentConfig): Configuration for the chat agent. N) r8r;r<r r=r r?copyinitial_historycurrent_user_inputrA)selfrNs r"__init__zAtomicAgent.__init__asmm \\ ~~6 '-'E'E'`I^I`$!--#||002"&$*$?$?$E2!r3cB|jj|_y)z: Resets the history to its initial state. N)rQrPr<rSs r" reset_historyzAtomicAgent.reset_historyqs++002 r3returnc^t|drt|j\}}|St}|SN__orig_class__)hasattrr r[r')rSTI_s r" input_schemazAtomicAgent.input_schemaws4 4) *T001EB &B r3c^t|drt|j\}}|St}|SrZ)r\r r[r5)rSr^TOs r" output_schemazAtomicAgent.output_schemas4 4) *T001EAr 'B r3c|jg|_n-|j|jjdg|_|xj|jj z c_y)N)rolecontent)r?messagesr=generate_promptr< get_historyrVs r"_prepare_messageszAtomicAgent._prepare_messagess]    #DM!,,#;;KKMDM 1133 r3N user_inputct|jtjjrJd|r=|jj ||_|jjd||j|jjjjd|j|j|jd|j}|jjd||S)a Runs the chat agent with the given user input synchronously. Args: user_input (Optional[InputSchema]): The input from the user. If not provided, skips adding to history. Returns: OutputSchema: The response from the chat agent. zIThe run method is not supported for async clients. Use run_async instead.user)rfr;response_model assistantr2) isinstancer8rBAsyncInstructorr<initialize_turnrR add_messagerichat completionscreaterfr;rbrArSrjresponses r"runzAtomicAgent.runs KK**::  W V W   LL ( ( *&0D # LL $ $VZ 8  64;;##//66 ]]**-- ''     h7r3c3dKt|jtjjrJd|r=|jj ||_|jjd||j|jjjjd|j|j|jd|jddi}|D]}||jdij!}|jjd||Sw)a Runs the chat agent with the given user input, supporting streaming output. Args: user_input (Optional[InputSchema]): The input from the user. If not provided, skips adding to history. Yields: OutputSchema: Partial responses from the chat agent. Returns: OutputSchema: The final response from the chat agent. zPThe run_stream method is not supported for async clients. Use run_async instead.rlr;rfrmstreamTrnr2ror8rBrpr<rqrRrrrirsrtcreate_partialr;rfrbrA model_dump)rSrjresponse_streampartial_responsefull_response_contents r" run_streamzAtomicAgent.run_streams# KK**::  ^ ] ^   LL ( ( *&0D # LL $ $VZ 8  E$++**66EE **]]-- ''    !0 # " " #!3 2 2 S5E5P5P5R S   .CD$$sD.D0cKt|jtjjsJd|r=|jj ||_|jjd||j|jjjjd|j|j|jd|jd{}|jjd||S7"w)a Runs the chat agent asynchronously with the given user input. Args: user_input (Optional[InputSchema]): The input from the user. If not provided, skips adding to history. Returns: OutputSchema: The response from the chat agent. Raises: NotAsyncIterableError: If used as an async generator (in an async for loop). Use run_async_stream() method instead for streaming responses. *The run_async method is for async clients.rlrzNrnr2)ror8rBrpr<rqrRrrrirsrtrur;rfrbrArvs r" run_asynczAtomicAgent.run_asyncs$++z'8'8'H'HIwKwwI  LL ( ( *&0D # LL $ $VZ 8  <))55<< **t}}TEWEW [_[t[t     h7  sC$D &D '#D cKt|jtjjsJd|r=|jj ||_|jjd||j|jjjjd|j|j|jd|jddi}d}|23d{}|}|7 6|r=|jdi|j!}|jjd|yyw) a@ Runs the chat agent asynchronously with the given user input, supporting streaming output. Args: user_input (Optional[InputSchema]): The input from the user. If not provided, skips adding to history. Yields: OutputSchema: Partial responses from the chat agent. rrlrzr{TNrnr2r|)rSrjr last_responserrs r"run_async_streamzAtomicAgent.run_async_streams)$++z'8'8'H'HIwKwwI  LL ( ( *&0D # LL $ $VZ 8  E$++**66EE **]]-- ''     &5 # #",M" " #o $6D$6$6$T9Q9Q9S$T ! LL $ $[2G H s+C)D?+C>/C<0C>3 D?<C>>AD? provider_namec||jjvrtd|d|jj|S)a/ Retrieves a context provider by name. Args: provider_name (str): The name of the context provider. Returns: BaseDynamicContextProvider: The context provider if found. Raises: KeyError: If the context provider is not found. Context provider ' ' not found.r=context_providersKeyErrorrSrs r"get_context_providerz AtomicAgent.get_context_providersD  < < N N N/ lKL L++==mLLr3providerc6||jj|<y)z Registers a new context provider. Args: provider_name (str): The name of the context provider. provider (BaseDynamicContextProvider): The context provider instance. N)r=r)rSrrs r"register_context_providerz%AtomicAgent.register_context_provider+sIQ$$66}Er3c||jjvr|jj|=ytd|d)z Unregisters an existing context provider. Args: provider_name (str): The name of the context provider to remove. rrNrrs r"unregister_context_providerz'AtomicAgent.unregister_context_provider5sA D88JJ J,,>>}M/ lKL Lr3rI)r,r-r.__type_params__r/r7rTrWpropertyrrr_rbrirrxrrrrrr0r rrr) .type_paramsrGrJsr"rMrMJs",F{F 3 d<0tL1 4h{3|:%%Xk%:%%iP\^bdpPpFq%%N(;*?<:!I+1F!IR`amosasRt!IFM#M$?Y:ZM"QsQF`Q M Mr3rMr2) .generic_baserrGrJs @@@r"#rJs!+-G+uMuMuMr3__main__)Console)Panel)Table)Syntax)box)OpenAI AsyncOpenAI)LivetitleschemarXcZt|tj}|jdd|jdd|jdd|jj D]9\}}|j |t|j|jxsd ;|S) zCreate a table displaying schema information. Args: title (str): Title of the table schema (Type[BaseModel]): Schema to display Returns: Table: Rich table containing schema information rrrcyanstylermagenta Descriptiongreenr) rrROUNDED add_column model_fieldsitemsadd_rowr0 annotationr))rr schema_table field_namefields r"_create_schema_tablerNs5ckk: v6i8 W=!'!4!4!:!:!< ] J  S1A1A-BEDUDUD[Y[ \ ]r3agentctdtj}|jdd|jdd|j d|j |j d t t|jj|j d t t|jj|S) zCreate a table displaying agent configuration. Args: agent (AtomicAgent): Agent instance Returns: Table: Rich table containing configuration information zAgent ConfigurationrPropertyrrValueyellowModelHistoryzSystem Prompt Generator) rrrrrr;r0typer<r,r=)r info_tables r"_create_config_tablerbs!6CKKH j7gX67EKK09c$u}}*=*F*F&GH4c$u?\?\:]:f:f6ghr3c t}|jtjdddt d|j }|j|t d|j }|j|t|}|j||jj}|jtt|ddd d d d y)z@Display information about the agent's configuration and schemas.z([bold blue]Agent Information[/bold blue]bluer border_stylepaddingz Input Schemaz Output SchemamarkdownmonokaiT)theme line_numberszSample System PromptrF)rrexpandN) rprintrfitrr_rbrr=rgr)rconsoleinput_schema_tableoutput_schema_tabler system_prompts r"display_agent_inforus) II:#  2.%BTBTU ()3?EDWDWX )**%0  j!55EEG  }j PTU,$   r3 streamingcK|rCtjt}t|d}t t t f|}nBtjt}t|d}t t t f|}t|t}|jtjd|ddd |jd}|jd k(r|jd y|j| }|jd |rat!|d 5}|j#|23d{}|j%} t'j(| d} |j+| B|j-|} | j%} t'j(| d} |j| 76 ddd#1swYxYww)zInteractive chat loop with the AI agent. Args: streaming (bool): Whether to use streaming mode for responses r9)r8r;z>[bold blue]Interactive Chat Mode[/bold blue] [cyan]Streaming: z[/cyan] Type 'exit' to quitrrrz [bold green]You:[/bold green] exitz[yellow]Goodbye![/yellow])r*z![bold blue]Assistant:[/bold blue])rrefresh_per_secondN)indent)rB from_openairr7rMr'r5rrrrrrinputlowerr_rrr~jsondumpsupdaterx) rr8rNrr user_messagerjliver response_jsonjson_strrws r" chat_looprs ++KM:F mDF 46K KLVTE++FH5F mDF 46K KLVTE 5!) II$$-;/&&$   "==)KLL!!#v- 9:+++FJ MM= >'a@.D272H2H2T...(8(C(C(E #'::mA#F H- 99Z0 ( 3 3 5 ::mA> h'+.2T...sID1G+3GG G G :GA G+GG G+G($G+z# [bold]Starting chat loop...[/bold]T)r)F)9rBpydanticrrtypingrrrrr "atomic_agents.context.chat_historyr -atomic_agents.context.system_prompt_generatorr r !atomic_agents.base.base_io_schemarinstructor.dsl.partialrjiterrr#r% classmethodmodel_from_chunksmodel_from_chunks_asyncr'r5r7rMr, rich.consoler rich.panelr rich.tabler rich.syntaxrrichropenairrasyncio rich.liverrr0rrrboolrrrrxr2r3r"rs%FF:;.!,,E F &12Q&R #< L  x) xuMp z$  "*CiU(KE&  +  D2(42(hiG MM89GKK D)*Wr3