Skip to content

Commit

Permalink
Catch JSON error in summary_memory.py (Significant-Gravitas#3996)
Browse files Browse the repository at this point in the history
Co-authored-by: k-boikov <[email protected]>
  • Loading branch information
kristianjackson and k-boikov authored May 12, 2023
1 parent b06ea61 commit a60512c
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions autogpt/memory_management/summary_memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from autogpt.config import Config
from autogpt.llm.llm_utils import create_chat_completion
from autogpt.log_cycle.log_cycle import PROMPT_SUMMARY_FILE_NAME, SUMMARY_FILE_NAME
from autogpt.logs import logger

cfg = Config()

Expand Down Expand Up @@ -75,10 +76,14 @@ def update_running_summary(
event["role"] = "you"

# Remove "thoughts" dictionary from "content"
content_dict = json.loads(event["content"])
if "thoughts" in content_dict:
del content_dict["thoughts"]
event["content"] = json.dumps(content_dict)
try:
content_dict = json.loads(event["content"])
if "thoughts" in content_dict:
del content_dict["thoughts"]
event["content"] = json.dumps(content_dict)
except json.decoder.JSONDecodeError:
if cfg.debug_mode:
logger.error(f"Error: Invalid JSON: {event['content']}\n")

elif event["role"].lower() == "system":
event["role"] = "your computer"
Expand Down

0 comments on commit a60512c

Please sign in to comment.