new version compatible

ControllerAtomicFlow.yaml

@@ -1,5 +1,6 @@
 name: "ControllerFlow"
 description: "Proposes the next action to take towards achieving the goal, and prepares the input for the executor."
+_target_: flow_modules.aiflows.ControllerExecutorFlowModule.ControllerAtomicFlow.instantiate_from_default_config
 enable_cache: True
 
 #######################################################

ControllerExecutorFlow.yaml

@@ -1,7 +1,7 @@
 name: "CtrlEx"
 description: "ControllerExecutor (i.e., MRKL, ReAct) interaction implementation with Flows that approaches the problem solving in two phases: one Flow chooses the next step and another Flow executes it. This is repeated until the controller Flow concludes on an answer."
 max_rounds: 30
-
+_target_: flow_modules.aiflows.ControllerExecutorFlowModule.ControllerExecutorFlow.instantiate_from_default_config
 ### Information used by the default interface specification implementation
 input_interface:
   - "goal"
@@ -14,7 +14,7 @@ subflows_config:
   Controller:
     name: "ControllerAtomicFlow"
     description: "A flow that calls other flows to solve a problem."
-    _target_: flow_modules.aiflows.ControllerAtomicFlow.instantiate_from_default_config
+    _target_: flow_modules.aiflows.ControllerExecutorFlowModule.ControllerAtomicFlow.instantiate_from_default_config
     commands:
       finish:
         description: "Signal that the objective has been satisfied, and returns the answer to the user."

WikiSearchAtomicFlow.yaml

@@ -1,6 +1,6 @@
 name: WikiSearchAtomicFlow
 description: "A Flow that queries the wikipedia API for a page content."
-
+_target_: flow_modules.aiflows.ControllerExecutorFlowModule.WikiSearchAtomicFlow.instantiate_from_default_config
 keep_raw_response: false
 
 lang: en

demo.yaml

@@ -4,7 +4,11 @@ max_rounds: 30
 ### Subflows specification
 subflows_config:
   Controller:
-    _target_: flow_modules.aiflows.ControllerExecutorFlowModule.ControllerAtomicFlow.instantiate_from_default_config
+    _target_: aiflows.base_flows.AtomicFlow.instantiate_from_default_config
+    name: "Proxy Controller"
+    description: "A flow that acts as a proxy for the controller."
+    flow_endpoint: ControllerAtomicFlow
+    user_id: local
     commands:
       wiki_search:
         description: "Performs a search on Wikipedia."
@@ -22,6 +26,8 @@ subflows_config:
 
 
   wiki_search:
-    _target_: flow_modules.aiflows.ControllerExecutorFlowModule.WikiSearchAtomicFlow.instantiate_from_default_config
+    _target_: aiflows.base_flows.AtomicFlow.instantiate_from_default_config
     name: "WikiSearchAtomicFlow"
     description: "A flow that searches Wikipedia for information."
+    flow_endpoint: WikiSearchAtomicFlow
+    user_id: local

run.py

@@ -58,26 +58,39 @@ if __name__ == "__main__":
 
     
     #3. ~~~~ Serve The Flow ~~~~
-    serve_utils.recursive_serve_flow(
+    
+    
+    serve_utils.serve_flow(
+        cl = cl,
+        flow_class_name="flow_modules.aiflows.ControllerExecutorFlowModule.WikiSearchAtomicFlow",
+        flow_endpoint="WikiSearchAtomicFlow",
+    )
+    
+    serve_utils.serve_flow(
         cl = cl,
-        flow_type="ControllerExecutorFlowModule",
-        default_config=cfg,
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.ControllerExecutorFlowModule.ControllerAtomicFlow",
+        flow_endpoint="ControllerAtomicFlow",
     )
     
+    serve_utils.serve_flow(
+        cl = cl,
+        flow_class_name="flow_modules.aiflows.ControllerExecutorFlowModule.ControllerExecutorFlow",
+        flow_endpoint="ControllerExecutorFlow",
+    )
+
     #4. ~~~~~Start A Worker Thread~~~~~
-    run_dispatch_worker_thread(cl, dispatch_point="coflows_dispatch", flow_modules_base_path=FLOW_MODULES_PATH)
+    run_dispatch_worker_thread(cl)
 
     #5. ~~~~~Mount the flow and get its proxy~~~~~~
-    proxy_flow = serve_utils.recursive_mount(
+    proxy_flow= serve_utils.get_flow_instance(
         cl=cl,
-        client_id="local",
-        flow_type="ControllerExecutorFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="ControllerExecutorFlow",
+        user_id="local",
+        config_overrides = cfg
+    )
+    
+ 
+     
     
     #6. ~~~ Get the data ~~~
     data = {