(state, agent, to_generate_code)
| 5 | from integuru.util.print import print_dag, visualize_dag, print_dag_in_reverse |
| 6 | |
| 7 | def check_end_condition(state, agent, to_generate_code): |
| 8 | agent.dag_manager.detect_cycles() |
| 9 | |
| 10 | if len(state.get("to_be_processed_nodes", [])) == 0: |
| 11 | print("------------------------Successfully analyzed!!!-------------------------------", flush=True) |
| 12 | print_dag(agent.dag_manager.graph, agent.global_master_node_id) |
| 13 | visualize_dag(agent.dag_manager.graph) |
| 14 | print_dag_in_reverse(agent.dag_manager.graph, to_generate_code=to_generate_code) |
| 15 | return "end" |
| 16 | else: |
| 17 | print("Continuing execution", flush=True) |
| 18 | print(f"Generated graph at current step: {print_dag(agent.dag_manager.graph, agent.global_master_node_id)}", flush=True) |
| 19 | return "continue" |
| 20 | |
| 21 | |
| 22 | def build_graph(prompt, har_file_path="network_requests.har", cookie_path="cookies.json", to_generate_code=False): |
nothing calls this directly
no test coverage detected