Clean Versions

run.py

@@ -17,12 +17,9 @@ from aiflows.interfaces import KeyInterface
 from aiflows.utils.colink_utils import start_colink_server
 from aiflows.workers import run_dispatch_worker_thread
 
-CACHING_PARAMETERS.do_caching = False  # Set to True in order to disable caching
-# clear_cache() # Uncomment this line to clear the cache
 
 # logging.set_verbosity_debug()
 
-
 dependencies = [
     {"url": "aiflows/VisionFlowModule", "revision": os.getcwd()}
 ]
@@ -31,9 +28,7 @@ from aiflows import flow_verse
 flow_verse.sync_dependencies(dependencies)
 if __name__ == "__main__":
     
-    #1. ~~~~~ Set up a colink server ~~~~
-    FLOW_MODULES_PATH = "./"
-    
+    #1. ~~~~~ Set up a colink server ~~~~    
     cl = start_colink_server()
 
 
@@ -57,26 +52,22 @@ if __name__ == "__main__":
 
     
     #3. ~~~~ Serve The Flow ~~~~
-    serve_utils.recursive_serve_flow(
+    serve_utils.serve_flow(
         cl = cl,
-        flow_type="VisionFlowModule",
-        default_config=cfg,
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.VisionFlowModule.VisionAtomicFlow",
+        flow_endpoint="VisionAtomicFlow",
     )
     
     #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(
+    #5. ~~~~~Mount the flow and get an instance of it via a proxy~~~~~~
+    proxy_flow= serve_utils.get_flow_instance(
         cl=cl,
-        client_id="local",
-        flow_type="VisionFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="VisionAtomicFlow",
+        user_id="local",
+        config_overrides = cfg
+    )
     
     #6. ~~~ Get the data ~~~
     url_image = {"type": "url",
@@ -127,5 +118,5 @@ if __name__ == "__main__":
     
     
     #9. ~~~~~Optional: Unserve Flow~~~~~~
-    # serve_utils.delete_served_flow(cl, "VisionFlowModule")    o_caching = False  # Set to True to enable caching
+    # serve_utils.delete_served_flow(cl, "VisionFlowModule")