diff --git a/vision_agent/agent/vision_agent_v3.py b/vision_agent/agent/vision_agent_v3.py index 6d194d9c..27f0cdd6 100644 --- a/vision_agent/agent/vision_agent_v3.py +++ b/vision_agent/agent/vision_agent_v3.py @@ -186,6 +186,18 @@ def write_and_test_code( "result": fixed_code_and_test["reflections"], } ) + log_progress( + { + "log": "Debug code:", + "code": code, + } + ) + log_progress( + { + "log": "Debug test:", + "code": test, + } + ) _LOGGER.info( f"Debug attempt {count + 1}, reflection: {fixed_code_and_test['reflections']}" ) @@ -208,6 +220,32 @@ def write_and_test_code( _CONSOLE.print( Syntax(f"{code}\n{test}", "python", theme="gruvbox-dark", line_numbers=True) ) + + if count == max_retries: + log_progress( + { + "log": f"{max_retries} max retries reached.", + } + ) + + log_progress( + { + "log": "Final code:", + "code": code, + } + ) + log_progress( + { + "log": "Final test:", + "code": test, + } + ) + log_progress( + { + "log": "Final result:", + "result": result, + } + ) _LOGGER.info(f"Final Result: {result}") return { @@ -367,7 +405,7 @@ def chat_with_workflow( self.log_progress( { - "log": f"The Vision Agent V3 has concluded this chat.\nSuccess: {success}", + "log": f"The Vision Agent V3 has concluded this chat.", "finished": True, } )