github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Black formatting

Browse Source
This commit is contained in:
Toran Bruce Richards
2023-04-29 23:22:08 +12:00
parent 991bc77e0b
commit 317361da8c
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
autogpt/memory_management/summary_memory.py
View File

@@ -7,7 +7,10 @@ cfg = Config()
prev_index = 0
current_memory = ""
def get_newly_trimmed_messages(full_message_history: List[Dict[str, str]], current_context: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], int]:
def get_newly_trimmed_messages(
full_message_history: List[Dict[str, str]], current_context: List[Dict[str, str]]
) -> Tuple[List[Dict[str, str]], int]:
"""
This function returns a list of dictionaries contained in full_message_history
with an index higher than prev_index that are absent from current_context, and the new index.
@@ -27,7 +30,9 @@ def get_newly_trimmed_messages(full_message_history: List[Dict[str, str]], curre
new_messages = [msg for i, msg in enumerate(full_message_history) if i > prev_index]
# Remove messages that are already present in current_context
new_messages_not_in_context = [msg for msg in new_messages if msg not in current_context]
new_messages_not_in_context = [
msg for msg in new_messages if msg not in current_context
]
# Find the index of the last message processed
new_index = prev_index
@@ -59,7 +64,7 @@ def update_running_summary(new_events: List[Dict]) -> str:
"""
global current_memory
prompt = f'''Your task is to create a concise running summary of actions in the provided text, focusing on key and potentially important information to remember.
You will receive the current summary and the latest development. Combine them, adding relevant key information from the latest development in 1st person past tense and keeping the summary concise.
@@ -85,8 +90,8 @@ Latest Development:
current_memory = create_chat_completion(messages, cfg.fast_llm_model)
message_to_return = {
"role": "system",
"content": f"This reminds you of these events from your past: \n{current_memory}",
}
"role": "system",
"content": f"This reminds you of these events from your past: \n{current_memory}",
}
return message_to_return
return message_to_return