Update code_assistant_runnable.py

code_assistant_runnable.py

@@ -612,22 +612,24 @@ def get_runnable():
 
     # Define the nodes
     workflow.add_node("generate", generate)  # generation solution
-    workflow.add_node("check_code", code_check)  # check code
-    workflow.add_node("reflect", reflect)  # reflect
+    # workflow.add_node("check_code", code_check)  # check code
+    # workflow.add_node("reflect", reflect)  # reflect
 
     # Build graph
     workflow.add_edge(START, "generate")
-    workflow.add_edge("generate", "check_code")
-    workflow.add_conditional_edges(
-        "check_code",
-        decide_to_finish,
-        {
-            "end": END,
-            "reflect": "reflect",
-            "generate": "generate",
-        },
-    )
-    workflow.add_edge("reflect", "generate")
+    workflow.add_edge("generate", END)
+    # workflow.add_edge("generate", "check_code")
+    # workflow.add_conditional_edges(
+    #     "check_code",
+    #     decide_to_finish,
+    #     {
+    #         "end": END,
+    #         "reflect": "reflect",
+    #         "generate": "generate",
+    #     },
+    # )
+    # workflow.add_edge("reflect", "generate")
+    
 
     # Remove the checkpointer for now since it's causing issues
     code_assistant_app = workflow.compile()