Update CodeGeneratorAtomicFlow.py

CodeGeneratorAtomicFlow.py

@@ -68,7 +68,7 @@ class CodeGeneratorAtomicFlow(ChatAtomicFlow):
             response = json.loads(api_output)
             return response
         except json.decoder.JSONDecodeError:
-            new_goal = f"Here is your previous response {api_output}, it cannot be parsed with json.loads, it could be the backslashes usesd for escaping single quotes in the string arguments of the code are not properly escaped themselves within the JSON context. Please fix this issue."
+            new_goal = f"Here is your previous response {api_output}, it cannot be parsed with json.loads, it could be the backslashes usesd for escaping single quotes in the string arguments of the code are not properly escaped themselves within the JSON context. Please fix this issue. Provide your response only in JSON format."
             new_input_data = input_data.copy()
             new_input_data['goal'] = new_goal
             new_api_output = super().run(new_input_data)["api_output"].strip()