Clean Versions

LCToolFlow.py

@@ -85,10 +85,8 @@ class LCToolFlow(AtomicFlow):
     def run(self, input_message: FlowMessage):
         """ This method runs the flow. It runs the backend on the input data.
         
-        :param input_data: The input data of the flow.
-        :type input_data: Dict[str, Any]
-        :return: The output data of the flow.
-        :rtype: Dict[str, Any]
+        :param input_message: The input message of the flow.
+        :type input_message: FlowMessage
         """
         input_data = input_message.data
         observation = self.backend.run(tool_input=input_data)

run.py

@@ -45,24 +45,20 @@ if __name__ == "__main__":
     #3. ~~~~ Serve The Flow ~~~~
     serve_utils.serve_flow(
         cl = cl,
-        flow_type="LCToolFlowModule",
-        default_config=cfg,
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.LCToolFlowModule.LCToolFlow",
+        flow_endpoint="LCToolFlow",
     )
     
     #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="LCToolFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="LCToolFlow",
+        user_id="local",
+        config_overrides = cfg
+    ) 
     
 
     #6. ~~~ Get the data ~~~
@@ -97,4 +93,4 @@ if __name__ == "__main__":
     
     
     #9. ~~~~~Optional: Unserve Flow~~~~~~
-    # serve_utils.delete_served_flow(cl, "ReverseNumberAtomicFlow_served")
+    # serve_utils.delete_served_flow(cl, "LCToolFlow")