From 91aa40e0df66c719d50e8fa31a8e7d143a87bd45 Mon Sep 17 00:00:00 2001 From: James Collins Date: Sun, 23 Apr 2023 16:59:49 -0700 Subject: [PATCH] Remove another global memory access --- autogpt/processing/text.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/autogpt/processing/text.py b/autogpt/processing/text.py index 2122f0f0..9946951a 100644 --- a/autogpt/processing/text.py +++ b/autogpt/processing/text.py @@ -10,7 +10,6 @@ from autogpt.llm_utils import create_chat_completion from autogpt.memory import get_memory CFG = Config() -MEMORY = get_memory(CFG) def split_text( @@ -109,7 +108,8 @@ def summarize_text( memory_to_add = f"Source: {url}\n" f"Raw content part#{i + 1}: {chunk}" - MEMORY.add(memory_to_add) + memory = get_memory(CFG) + memory.add(memory_to_add) messages = [create_message(chunk, question)] tokens_for_chunk = token_counter.count_message_tokens(messages, model) @@ -128,7 +128,7 @@ def summarize_text( memory_to_add = f"Source: {url}\n" f"Content summary part#{i + 1}: {summary}" - MEMORY.add(memory_to_add) + memory.add(memory_to_add) print(f"Summarized {len(chunks)} chunks.")