This commit is contained in:
BillSchumacher
2023-04-18 19:40:14 -05:00
parent 239d64a602
commit 6e94409594
4 changed files with 13 additions and 7 deletions

View File

@@ -4,16 +4,16 @@ from __future__ import annotations
import contextlib import contextlib
import json import json
from typing import Dict, Any from typing import Any, Dict
from colorama import Fore from colorama import Fore
from regex import regex from regex import regex
from autogpt.config import Config
from autogpt.json_utils.json_fix_general import correct_json from autogpt.json_utils.json_fix_general import correct_json
from autogpt.llm_utils import call_ai_function from autogpt.llm_utils import call_ai_function
from autogpt.logs import logger from autogpt.logs import logger
from autogpt.speech import say_text from autogpt.speech import say_text
from autogpt.config import Config
JSON_SCHEMA = """ JSON_SCHEMA = """
{ {
@@ -102,7 +102,10 @@ def fix_json_using_multiple_techniques(assistant_reply: str) -> Dict[Any, Any]:
if assistant_reply_json != {}: if assistant_reply_json != {}:
return assistant_reply_json return assistant_reply_json
logger.error("Error: The following AI output couldn't be converted to a JSON:\n", assistant_reply) logger.error(
"Error: The following AI output couldn't be converted to a JSON:\n",
assistant_reply,
)
if CFG.speak_mode: if CFG.speak_mode:
say_text("I have received an invalid JSON response from the OpenAI API.") say_text("I have received an invalid JSON response from the OpenAI API.")

View File

@@ -4,8 +4,8 @@ import re
from jsonschema import Draft7Validator from jsonschema import Draft7Validator
from autogpt.logs import logger
from autogpt.config import Config from autogpt.config import Config
from autogpt.logs import logger
CFG = Config() CFG = Config()

View File

@@ -204,7 +204,10 @@ logger = Logger()
def print_assistant_thoughts(ai_name, assistant_reply): def print_assistant_thoughts(ai_name, assistant_reply):
"""Prints the assistant's thoughts to the console""" """Prints the assistant's thoughts to the console"""
from autogpt.json_utils.json_fix_llm import fix_and_parse_json, attempt_to_fix_json_by_finding_outermost_brackets from autogpt.json_utils.json_fix_llm import (
attempt_to_fix_json_by_finding_outermost_brackets,
fix_and_parse_json,
)
try: try:
try: try: