added the final return

agent.py

@@ -173,6 +173,7 @@ builder.add_conditional_edges("assistant", tools_condition)
 builder.add_edge("tools", "assistant")
 
 graph = builder.compile()
+return graph
 
 ## ----- TESTING (Optional) ----- ##