a dg'@sddlZddlmZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZGd d d eZdS) N)datetime)AnyDictListOptionalTuple)LLMChain)BaseLanguageModel)PromptTemplate) BaseModel ConfigDictField)GenerativeAgentMemoryc@seZdZUdZeed<dZeeed<dZ eed<eed<e ed<e ed <d Z e ed <d Zeed <dZeed<eejdZeed<eedZeeed<eddZeeeedddZeedddZeedddZeeeddd Zeedd!d"Z d5eeeeed#d$d%Z!eedd&d'Z"d6eeee#e efd(d)d*Z$d7eeee#e efd(d+d,Z%ed-d.d/Z&d8e eeed0d1d2Z'd9e eeed0d3d4Z(dS):GenerativeAgentz/z/GenerativeAgent._parse_list..)r splitr")rlinesr%r%r& _parse_list+szGenerativeAgent._parse_list)promptrcCst|j||j|jdS)z3Create a chain with the same settings as the agent.)rr,rr)rrrrselfr,r%r%r&chain1szGenerativeAgent.chain) observationrcCs td}||j|dS)NzOWhat is the observed entity in the following observation? {observation} Entity=)r0r from_templater/runr")r.r0r,r%r%r&_get_entity_from_observation8sz,GenerativeAgent._get_entity_from_observation)r0 entity_namercCs"td}||j||dS)NzVWhat is the {entity} doing in the following observation? {observation} The {entity} is)entityr0r1)r.r0r5r,r%r%r&_get_entity_action?s z"GenerativeAgent._get_entity_actioncCs^td}||}|||}d|jd|}|d|}|j|dj|||gdS)zN%B %d, %Y, %I:%M %p)agent_summary_description current_timeZrelevant_memoriesZ agent_namer0Z agent_statusZmost_recent_memoriesrr8)r r2 get_summaryr<rr>strftimedictrrrZget_num_tokensformatrZmost_recent_memories_token_keyr/r3r") r.r0r=r>r,rAZrelevant_memories_strcurrent_time_strkwargsZconsumed_tokensr%r%r&_generate_reactionXs0   z"GenerativeAgent._generate_reactioncCs td|jdd|S)N^ r)r r!rr")r.rr%r%r&_clean_response|szGenerativeAgent._clean_response)r0r>rc Csd}|j|||d}|dd}|ji|jj|jd|d||jj|id|vr||dd}d |jd |fSd |vr||d d}d |jd |fSd |fSdS)React to a given observation.a1Should {agent_name} react to the observation, and if so, what would be an appropriate reaction? Respond in one line. If the action is to engage in dialogue, write: SAY: "what to say" otherwise, write: REACT: {agent_name}'s reaction (if anything). Either do nothing, react, or say something but not both. r? r observed z and reacted by zREACT:FrKSAY:T said N) rIr"r)r save_contextadd_memory_keyrnow_keyrL)r.r0r>call_to_action_template full_resultresultZreactionZ said_valuer%r%r&generate_reactions.z!GenerativeAgent.generate_reactionc Csd}|j|||d}|dd}d|vr||dd}|ji|jj|jd|d||jj|id |jd |fSd |vr||d d}|ji|jj|jd|d||jj|id |jd |fSd |fSd S)rMzWhat would {agent_name} say? To end the conversation, write: GOODBYE: "what to say". Otherwise to continue the conversation, write: SAY: "what to say next" r?rNrzGOODBYE:rPrOz and said FrRrQTN) rIr"r)rLrrSrTrrU)r.r0r>rVrWrXZfarewellZ response_textr%r%r&generate_dialogue_responsesBz*GenerativeAgent.generate_dialogue_response)rcCs.td}||j|j|jdgdS)rzHow would you summarize {name}'s core characteristics given the following statements: {relevant_memories}Do not embellish. Summary: z's core characteristics)rr:)r r2r/r3rr"r-r%r%r&_compute_agent_summarys z&GenerativeAgent._compute_agent_summary) force_refreshr>rcCs|durtn|}||jj}|jr4||jks4|rD||_||_|jdurT|jnd}d|jd|dd|j d|jS)z*Return a descriptive summary of the agent.NrzName: z (age: )z Innate traits: rN) rr>rsecondsrrr[rrr)r.r\r>rBZ since_refreshrr%r%r&rCs"    zGenerativeAgent.get_summarycCsJ|durtn|}|j||d}|d}|d|d|jd|jS)zFReturn a full header of the agent's status, summary, and current time.N)r\r>r@z It is z. z 's status: )rr>rCrDrr)r.r\r>rrGr%r%r&get_full_headers  zGenerativeAgent.get_full_header)N)N)N)FN)FN))__name__ __module__ __qualname____doc__str__annotations__rrintrrr rboolrrr rr>rlistrrr Z model_config staticmethodr+r rr/r4r7r<rIrLrrYrZr[rCr_r%r%r%r&r s`        $ " , r)r rtypingrrrrrZlangchain.chainsrZlangchain_core.language_modelsr Zlangchain_core.promptsr Zpydanticr r r Z/langchain_experimental.generative_agents.memoryrrr%r%r%r&s