Clean Versions

run.py

@@ -55,25 +55,21 @@ if __name__ == "__main__":
     
     #3. ~~~~ Serve The Flow ~~~~
     serve_utils.serve_flow(
-        cl = cl,
-        flow_type="ChatFlowModule",
-        default_config=cfg,
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        cl=cl,
+        flow_class_name="flow_modules.aiflows.ChatFlowModule.ChatAtomicFlow",
+        flow_endpoint="ChatAtomicFlow",
     )
-    
+
     #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="ChatFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="ChatAtomicFlow",
+        user_id="local",
+        config_overrides= cfg
+    )
     
 
     #6. ~~~ Get the data ~~~