Function run_workflow_as_server

src/WorkFlow.py:264–299

(llm)

def run_workflow_as_server(llm):
    # Load subgraph data
    with open("workflow.json", 'r') as file:
        graphs = json.load(file)
    
    # Process each subgraph
    for graph in graphs:
        subgraph_name = graph.get("name")        
        node_map = parse_nodes_from_json(graph)
        
        # Register the tool functions dynamically if has tool node, must before build graph
        for tool_node in find_nodes_by_type(node_map, "TOOL"):
            tool_code = f"{tool_node.description}"
            exec(tool_code, globals())

        
        subgraph = build_subgraph(node_map, llm)
        subgraph_registry[subgraph_name] = subgraph

    
    # Main Graph
    main_graph = StateGraph(MainGraphState)
    main_graph.add_node("subgraph", invoke_root)
    main_graph.set_entry_point("subgraph")
    main_graph = main_graph.compile()


    # ==========================
    # Run
    # ==========================
    for state in main_graph.stream(
        {
            "input": None,
        }
    ):
        logger(state)