fix: llm_generator.py JSONDecodeError (#8504)

2 years ago · eaa7e9b1f0
parent 6e37750fbd
commit eaa7e9b1f0
2 changed files with 5 additions and 3 deletions
--- a/api/core/llm_generator/llm_generator.py
+++ b/api/core/llm_generator/llm_generator.py
@ -47,6 +47,8 @@ class LLMGenerator:
            )
        answer = response.message.content
        cleaned_answer = re.sub(r"^.*(\{.*\}).*$", r"\1", answer, flags=re.DOTALL)
        if cleaned_answer is None:
            return ""
        result_dict = json.loads(cleaned_answer)
        answer = result_dict["Your Output"]
        name = answer.strip()