{h08 ddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZmZddlmZe ee fZGdd eZGd d Zed k(rpddlZdd lmZmZddlZGddeZGddeZGddeZGddeZedZ e jCdeddgdedde jCd ed!d"ed#ded$d%d&'ejDjGd(d)Z$ejDjKe$r4e jCded*ejjMe$g+e jOZ(e)d,e)e(eZ*e*jWe(e)d-e,e*jZD]\Z.Z/e)d.e.d/zd0e)d1e/j`e)d2e/jbe)d3e2e/jfj.e)d4e/jfjijkD]\Z6Z7e)d5e6d6e7e)d7e)d8e*jpe)d9e*jse)d:e*jZd;Z:e)e:jfjiyy)<N)Enum)Path)DictListOptionalType)PDFImageAudio) BaseModelField) BaseIOSchemac:eZdZUdZeed<eed<dZeeed<y)Messagea. Represents a message in the chat history. Attributes: role (str): The role of the message sender (e.g., 'user', 'system', 'tool'). content (BaseIOSchema): The content of the message. turn_id (Optional[str]): Unique identifier for the turn this message belongs to. rolecontentNturn_id) __name__ __module__ __qualname____doc__str__annotations__rrrc/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/atomic_agents/context/chat_history.pyrrs" I !GXc]!rrceZdZdZddeefdZddZdede ddfd Z dd Z de e fd Zdd Zdeefd ZdefdZdefdZdefdZdeddfdZededee fdZdZy) ChatHistorya1 Manages the chat history for an AI agent. Attributes: history (List[Message]): A list of messages representing the chat history. max_messages (Optional[int]): Maximum number of messages to keep in history. current_turn_id (Optional[str]): The ID of the current turn. N max_messagesc.g|_||_d|_y)a Initializes the ChatHistory with an empty history and optional constraints. Args: max_messages (Optional[int]): Maximum number of messages to keep in history. When exceeded, oldest messages are removed first. Nhistoryrcurrent_turn_id)selfrs r__init__zChatHistory.__init__)s') (.2rreturncHttj|_y)zH Initializes a new turn by generating a random turn ID. N)ruuiduuid4r#r$s rinitialize_turnzChatHistory.initialize_turn5s #4::<0rrrc|j|jt|||j}|jj ||j y)z Adds a message to the chat history and manages overflow. Args: role (str): The role of the message sender. content (BaseIOSchema): The content of the message. Nrrr)r#r+rr"append_manage_overflow)r$rrmessages r add_messagezChatHistory.add_message;sU    '  "((  G$ rc|jbt|j|jkDr?|jjdt|j|jkDr>yyy)zU Manages the chat history overflow based on max_messages constraint. Nr)rlenr"popr*s rr/zChatHistory._manage_overflowRsW    (dll#d&7&77   #dll#d&7&77 )rcFg}|jD]}|j}g}|jjj D]\}}t ||}t |trSd}|D]&} t | ts|j| d}(|rR|j|j|hut |tr|j||j|j|h|j|j|d|S)aX Retrieves the chat history, handling both regular and multimodal content. Returns: List[Dict]: The list of messages in the chat history as dictionaries. Each dictionary has 'role' and 'content' keys, where 'content' is a list that may contain strings (JSON) or multimodal objects. Note: This method does not support nested multimodal content. If your schema contains nested objects that themselves contain multimodal content, only the top-level multimodal content will be properly processed. FT)include)rr) r"r __class__ model_fieldsitemsgetattr isinstancelistINSTRUCTOR_MULTIMODAL_TYPESr.model_dump_jsonr) r$r"r0 input_contentprocessed_content field_namefield field_valuehas_multimodal_in_listitems r get_historyzChatHistory.get_historyZs|| QG#OOM " %2%<%<%I%I%O%O%Q f! E%mZ@ k40-2* +:%d,GH-44T:592:2)001N1NXbWc1N1de!+/JK)00=)001N1NXbWc1N1de! f$ NNGLL=NO P+ Q.rct|j}|j|j|j|_|S)z| Creates a copy of the chat history. Returns: ChatHistory: A copy of the chat history. r)rrloaddumpr#)r$ new_historys rcopyzChatHistory.copys<"t/@/@A %&*&:&: #rc|jS)z Returns the current turn ID. Returns: Optional[str]: The current turn ID, or None if not set. )r#r*s rget_current_turn_idzChatHistory.get_current_turn_ids###rrczt|j}|jDcgc]}|j|k7s|c}|_t|j|k(rtd|dt|jsd|_y||jk(r|jdj|_yycc}w)a@ Delete messages from the history by its turn ID. Args: turn_id (int): The turn ID of the message to delete. Returns: str: A success message with the deleted turn ID. Raises: ValueError: If the specified turn ID is not found in the history. zTurn ID z not found in history.N)r3r"r ValueErrorr#)r$rinitial_lengthmsgs rdelete_turn_idzChatHistory.delete_turn_idsT\\*'+||Ns{{g7MN t||  .xy0FGH H4<< #'D ,, ,#'<<#3#;#;D -Os B8B8c,t|jS)z Returns the number of messages in the chat history. Returns: int: The number of messages. )r3r"r*s rget_message_countzChatHistory.get_message_counts4<<  rcpg}|jD]w}|jj}|j|jd|j |jj d|jd}|j|y||j|jd}tj|S)z Serializes the entire ChatHistory instance to a JSON string. Returns: str: A JSON string representation of the ChatHistory. .) class_namedatar-r!) r"rr7rrrr>rr.rr#jsondumps)r$serialized_historyr0 content_classserialized_message history_datas rrJzChatHistory.dumps || :G#OO55M %2%=%=$>a @V@V?W"X#OO;;=#?? "   % %&8 9 :* --#33 zz,''rserialized_datac tj|}g|_|d|_|d|_|dD]o}|d}|j |d}|j |d}|j|t|d||d }|jj|qy #tjtttf$r}td |d }~wwxYw) a! Deserializes a JSON string and loads it into the ChatHistory instance. Args: serialized_data (str): A JSON string representation of the ChatHistory. Raises: ValueError: If the serialized data is invalid or cannot be deserialized. rr#r"rrYrZrrr-zInvalid serialized data: N)r[loadsr"rr#_get_class_from_stringmodel_validate_json_process_multimodal_pathsrr.JSONDecodeErrorKeyErrorAttributeError TypeErrorrQ) r$rar` message_data content_infor^content_instancer0es rrIzChatHistory.loads >::o6LDL ,^ 8<= = >sB'B**#C  CC  class_stringc`|jdd\}}t||g}t||S)a( Retrieves a class object from its string representation. Args: class_string (str): The fully qualified class name. Returns: Type[BaseIOSchema]: The class object. Raises: AttributeError: If the class cannot be found. rX)fromlist)rsplit __import__r:)ro module_namerYmodules rrdz"ChatHistory._get_class_from_strings6#/"5"5c1"= ZK:,?vz**rct|ttfrQt|jtr7|jj dst |j|_yyt|tr|D]}|j|yt|tr'|jD]}|j|yt|dr:|jD]*}t||s|jt||,yt|drKt|ts:|jj!D]\}}|dk7s |j|yyy)a: Process multimodal objects to convert string paths to Path objects. Note: this is necessary only for PDF and Image instructor types. The from_path behavior is slightly different for Audio as it keeps the source as a string. Args: obj: The object to process. )zhttp://zhttps://zdata:r8__dict____pydantic_fields_set__N)r;r r sourcer startswithrr<rfdictvalueshasattrr8r:rrxr9)r$objrEvaluerA attr_name attr_values rrfz%ChatHistory._process_multimodal_pathss2 cE3< (Z C-H::(()IJ!#**- K T " 5..t4 5 T " 6..u5 6 S. )!.. M 3 +2273 3KL MS* %jd.C),););)= ?% : 9922:> ?/D %r)N)r&N)r&r)rrrrrintr%r+rrr1r/rrrFrLrNrTrVrJrI staticmethodrrdrfrrrrrs 3Xc] 31       .$'T$Z'R $Xc]$C>D>:+S+T,5G++" ?rr__main__)rrcNeZdZUdZeddZeed<eddZe ed<y) NestedSchemazA nested schema for testing.zA nested field description nested_fieldzA nested integer nested_intN) rrrrr rrrrrrrrrr.s))!#3CD cD1CD CDrrceZdZUdZeddZeed<eddZe ed<eddZ e eed <edd Z e ed <y ) ComplexInputSchemazComplex Input Schema.z A text fieldr text_fieldzA number field number_fieldzA list of strings list_fieldzA nested schemarN)rrrrr rrrrfloatrTypeListrrrrrrr4sP"@ C@#C5EF eF$)#;N$O HSMO%*3rs --33%: %eS1 "i "F?F?R z9 E|E O\OjljY<Y#3  7 2%>bQ    6 $2CPRS$2CPRS  ggll=*=O ww~~o&$$  !?@ j#OO668>>@ )LE5 BugRw' ( ) ) !" N>667 89 n@@BC DE !"!))"-L ,   ) ) +,Er