Fd#ddlmZddlZddlZddlmZmZddlmZerddl m Z ddl m Z ddl mZddlmZmZmZmZdd lmZdd lmZmZdd lmZmZdd lmZeGd dZdS)) annotationsN) dataclassfield) TYPE_CHECKING)Agent)Config)extract_json_from_response) ChatSequenceMessage MessageRole MessageType)OPEN_AI_CHAT_MODELS)count_string_tokenscreate_chat_completion)PROMPT_SUMMARY_FILE_NAMESUMMARY_FILE_NAME)loggerceZdZUded<eeZded<dZded<d Zd ed <d&d Z dZ dZ d'd(dZ d)dZ d*dZd'd+d!Zd,d"Zd-d$Zd%ZdS).MessageHistoryragent)default_factory list[Message]messagesz I was createdstrsummaryrintlast_trimmed_indexic|j|SN)r)selfrs RC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\memory\message_history.py __getitem__zMessageHistory.__getitem__s}Qc*t|jSr )iterrr!s r"__iter__zMessageHistory.__iter__ sDM"""r$c*t|jSr )lenrr's r"__len__zMessageHistory.__len__#s4=!!!r$Nroler contenttypeMessageType | NonecJ|t|||Sr )appendr )r!r,r-r.s r"addzMessageHistory.add&s" {{74$77888r$messager c6|j|Sr )rr1)r!r3s r"r1zMessageHistory.append.s}##G,,,r$current_message_chainconfigrreturntuple[Message, list[Message]]c fdtD}fd|D}|sgfS||}|d}j|_||fS)a6 Returns a list of trimmed messages: messages which are in the message history but not in current_message_chain. Args: current_message_chain (list[Message]): The messages currently in the context. config (Config): The config to use. Returns: Message: A message with the new running summary after adding the trimmed messages. list[Message]: A list of messages that are in full_message_history with an index higher than last_trimmed_index and absent from current_message_chain. c0g|]\}}|jk|S)r).0rmsgr!s r" z0MessageHistory.trim_messages..As1   AsT5L1L1LC1L1L1Lr$cg|]}|v| Sr;r;)r<r=r5s r"r>z0MessageHistory.trim_messages..Fs+% % % 36K+K+KC+K+K+Kr$) new_eventsr6) enumeratesummary_messageupdate_running_summaryrindexr)r!r5r6 new_messagesnew_messages_not_in_chainnew_summary_message last_messages`` r" trim_messageszMessageHistory.trim_messages1s     'oo   % % % % '% % % !) .''))2- -"990:   14 "&-"5"5l"C"C"$===r$list[Message] | Nonec #K|p|j}tdt|dz D]}||}|jdkr|dkr||dz jdkr ||dz nd}||dz} t |jiks Jd|jdksJ|||fV#t$r2}tj d|d ||dz |d zYd}~d}~wwxYwdS) z Yields: Message: a message containing user input Message: a message from the AI containing a proposed action Message: the message containing the result of the AI's proposed action r ai_responseuserNz&AI response is not a valid JSON object action_resultz!Invalid item in message history: z ; Messages: ) rranger*r.r,r r-AssertionErrorrdebug)r!r6rr ai_message user_messageresult_messageerrs r" per_cyclezMessageHistory.per_cycleWsj,t}q#h--!+,,  A!!J-//#$q55Xa!e_-AV-K-KQQU &a!e_N .z/ABBbHHH;IHH%*o===="J>>>>>!    \\\RSTURUVWXYVYRYIZ\\   s56B,, C(6(C##C(c2tdd|jS)Nsystemz2This reminds you of these events from your past: )r rr's r"rCzMessageHistory.summary_messagess$  P$, P P   r$r@cZ|s|Stj|}|D]}|jdkrd|_ t |j}d|vr|d=tj||_\#tj $rB}tj d||j rtj |jYd}~d}~wwxYw|jdkrd|_|jdkr| |d }tj|jj}t%t'|j|j}g} d } |D]}t%t'||j} | | z||z |z krC|| |t%t'|j|j}|g} | } v| || | z } | r|| ||S) ae This function takes a list of dictionaries representing new events and combines them with the current summary, focusing on key and potentially important information to remember. The updated summary is returned in a message formatted in the 1st person past tense. Args: new_events (List[Dict]): A list of dictionaries containing the latest events to be added to the summary. Returns: str: A message containing the updated summary of actions, formatted in the 1st person past tense. Example: new_events = [{"event": "entered the kitchen."}, {"event": "found a scrawled note with the number 7"}] update_running_summary(new_events) # Returns: "This reminds you of these events from your past: I entered the kitchen and found a scrawled note saying 7." assistantyouthoughtszError: Invalid JSON: Nr[z your computerrOdr)rCcopydeepcopyr,lowerr r-jsondumpsJSONDecodeErrorrerror debug_moderemovergetfast_llm_model max_tokensrrrsummarize_batchr1) r!r@r6event content_dicteprompt_template_lengthrlsummary_tlengthbatch batch_tlength event_tlengths r"rDz%MessageHistory.update_running_summaryysj& *'')) )]:..   ) )Ez!![00" 9#=em#L#LL!\11(4$(J|$<$$>$>$>L8    Q&Q&Q&Q&f# # # # # r$r) __future__rrard dataclassesrrtypingr autogpt.agentrautogpt.configrautogpt.json_utils.utilitiesr autogpt.llm.baser r r r autogpt.llm.providers.openairautogpt.llm.utilsrrautogpt.log_cycle.log_cyclerr autogpt.logsrrr;r$r"rsQ"""""" (((((((( $######!!!!!!CCCCCCLLLLLLLLLLLL<<<<<<IIIIIIIISSSSSSSS Z Z Z Z Z Z Z  Z Z Z r$