Clean Versions

run.py

@@ -45,26 +45,22 @@ if __name__ == "__main__":
     
     
     #3. ~~~~ Serve The Flow ~~~~
-    serve_utils.recursive_serve_flow(
+    serve_utils.serve_flow(
         cl = cl,
-        flow_type="FixedReplyFlowModule",
-        default_config=cfg,
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.FixedReplyFlowModule.FixedReplyFlow",
+        flow_endpoint="FixedReplyFlow",
     )
     
     #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="FixedReplyFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="FixedReplyFlow",
+        user_id="local",
+        config_overrides = cfg
+    )
     
     #6. ~~~ Get the data ~~~
     data = {"id": 0}