github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-01-10 07:38:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Remove global pre_index from summary_memory.

Browse Source
This commit is contained in:
Toran Bruce Richards
2023-04-30 14:25:06 +12:00
parent a3522223d9
commit 88b2d5fb2d
3 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
autogpt/agent/agent.py
View File

@@ -56,6 +56,7 @@ class Agent:
cfg = Config()
self.ai_name = ai_name
self.memory = memory
self.last_memory_index = 0
self.full_message_history = full_message_history
self.next_action_count = next_action_count
self.command_registry = command_registry

1
autogpt/llm/chat.py
View File

@@ -157,6 +157,7 @@ def chat_with_ai(
newly_trimmed_messages = get_newly_trimmed_messages(
full_message_history=full_message_history,
current_context=current_context,
prev_index=agent.last_memory_index
)
memory = update_running_summary(newly_trimmed_messages)
current_context.insert(insertion_index, memory)

20
autogpt/memory_management/summary_memory.py
View File

@@ -10,25 +10,22 @@ 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]:
full_message_history: List[Dict[str, str]], current_context: List[Dict[str, str]], last_memory_index: int) -> 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.
with an index higher than last_memory_index that are absent from current_context, and the new index.
Args:
full_message_history (list): A list of dictionaries representing the full message history.
current_context (list): A list of dictionaries representing the current context.
prev_index (int): An integer representing the previous index.
last_memory_index (int): An integer representing the previous index.
Returns:
list: A list of dictionaries that are in full_message_history with an index higher than prev_index and absent from current_context.
list: A list of dictionaries that are in full_message_history with an index higher than last_memory_index and absent from current_context.
int: The new index value for use in the next loop.
"""
global prev_index
# Select messages in full_message_history with an index higher than prev_index
new_messages = [msg for i, msg in enumerate(full_message_history) if i > prev_index]
# Select messages in full_message_history with an index higher than last_memory_index
new_messages = [msg for i, msg in enumerate(full_message_history) if i > last_memory_index]
# Remove messages that are already present in current_context
new_messages_not_in_context = [
@@ -36,14 +33,13 @@ def get_newly_trimmed_messages(
]
# Find the index of the last message processed
new_index = prev_index
new_index = last_memory_index
if new_messages_not_in_context:
last_message = new_messages_not_in_context[-1]
new_index = full_message_history.index(last_message)
prev_index = new_index
return new_messages_not_in_context
return new_messages_not_in_context, new_index
def update_running_summary(new_events: List[Dict]) -> str: