Update run.py (#1)

- Update run.py (f9c6994c1e2b458f40eee28be6dd5dbd8ca943ea)


Co-authored-by: Neel Shah <Neel-Shah-29@users.noreply.huggingface.co>

run.py

@@ -9,7 +9,7 @@ from aiflows.utils.general_helpers import read_yaml_file, quick_load_api_keys
 from aiflows import logging
 from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
 
-from aiflows.utils import serve_utils
+from aiflows.utils import serving
 from aiflows.workers import run_dispatch_worker_thread
 from aiflows.messages import FlowMessage
 from aiflows.interfaces import KeyInterface
@@ -54,7 +54,7 @@ if __name__ == "__main__":
 
     
     #3. ~~~~ Serve The Flow ~~~~
-    serve_utils.serve_flow(
+    serving.serve_flow(
         cl=cl,
         flow_class_name="flow_modules.aiflows.ChatFlowModule.ChatAtomicFlow",
         flow_endpoint="ChatAtomicFlow",
@@ -64,7 +64,7 @@ if __name__ == "__main__":
     run_dispatch_worker_thread(cl)
 
     #5. ~~~~~Mount the flow and get its proxy~~~~~~
-    proxy_flow= serve_utils.get_flow_instance(
+    proxy_flow= serving.get_flow_instance(
         cl=cl,
         flow_endpoint="ChatAtomicFlow",
         user_id="local",
@@ -100,6 +100,6 @@ if __name__ == "__main__":
     
     
     #9. ~~~~~Optional: Unserve Flow~~~~~~
-    # serve_utils.delete_served_flow(cl, "ReverseNumberAtomicFlow_served")
+    # serving.delete_served_flow(cl, "ReverseNumberAtomicFlow_served")