new version compatible

ChromaDBFlow.yaml

@@ -1,5 +1,6 @@
 name: chroma_db
 description: ChromaDB is a document store that uses vector embeddings to store and retrieve documents
+_target_: flow_modules.aiflows.VectorStoreFlowModule.ChromaDBFlow.instantiate_from_default_config
 
 backend:
   _target_: aiflows.backends.llm_lite.LiteLLMBackend

VectorStoreFlow.yaml

@@ -1,5 +1,6 @@
 name: "VectorStoreFlow"
 description: "VectorStoreFlow"
+_target_: flow_modules.aiflows.VectorStoreFlowModule.VectorStoreFlow.instantiate_from_default_config
 
 backend:
   _target_: aiflows.backends.llm_lite.LiteLLMBackend

run.py

@@ -33,7 +33,6 @@ flow_verse.sync_dependencies(dependencies)
 if __name__ == "__main__":
     
     #1. ~~~~~ Set up a colink server ~~~~
-    FLOW_MODULES_PATH = "./"
     
     cl = start_colink_server()
 
@@ -58,48 +57,42 @@ if __name__ == "__main__":
 
     
     #3. ~~~~ Serve The Flow ~~~~
-    serve_utils.recursive_serve_flow(
+    serve_utils.serve_flow(
         cl = cl,
-        flow_type="ChromaDBFlowModule",
-        default_config=cfg["chroma_demo_flow"],
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.VectorStoreFlowModule.ChromaDBFlow",
+        flow_endpoint="ChromaDBFlow",
     )
     
     #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_cdb = serve_utils.recursive_mount(
+    proxy_flow_cdb= serve_utils.get_flow_instance(
         cl=cl,
-        client_id="local",
-        flow_type="ChromaDBFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    ) 
+        flow_endpoint="ChromaDBFlow",
+        user_id="local",
+        config_overrides = cfg["chroma_demo_flow"]
+    )
     
     #3.(2) ~~~~ Serve The Flow ~~~~
-    serve_utils.recursive_serve_flow(
+    serve_utils.serve_flow(
         cl = cl,
-        flow_type="VectoreStoreFlowModule",
-        default_config=cfg["vector_store_demo_flow"],
-        default_state=None,
-        default_dispatch_point="coflows_dispatch"
+        flow_class_name="flow_modules.aiflows.VectorStoreFlowModule.VectorStoreFlow",
+        flow_endpoint="VectorStoreFlow",
     )
+ 
     
     #4.(2) ~~~~~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.(2) ~~~~~Mount the flow and get its proxy~~~~~~
-    proxy_flow_vs = serve_utils.recursive_mount(
+    proxy_flow_vs= serve_utils.get_flow_instance(
         cl=cl,
-        client_id="local",
-        flow_type="VectoreStoreFlowModule",
-        config_overrides=None,
-        initial_state=None,
-        dispatch_point_override=None,
-    )   
+        flow_endpoint="VectorStoreFlow",
+        user_id="local",
+        config_overrides = cfg["vector_store_demo_flow"]
+    )
     
     #6. ~~~ Get the data ~~~
     data_write = {"id": 0, "operation": "write", "content": "The capital of Switzerland is Bern"}  # Add your data here