nbaldwin commited on
Commit
da88ab1
·
1 Parent(s): 2105a52
Files changed (1) hide show
  1. run.py +4 -4
run.py CHANGED
@@ -11,7 +11,7 @@ from aiflows.utils.general_helpers import read_yaml_file, quick_load_api_keys
11
  from aiflows import logging
12
  from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
13
 
14
- from aiflows.utils import serve_utils
15
  from aiflows.workers import run_dispatch_worker_thread
16
  from aiflows.messages import FlowMessage
17
  from aiflows.interfaces import KeyInterface
@@ -45,7 +45,7 @@ if __name__ == "__main__":
45
 
46
 
47
  #3. ~~~~ Serve The Flow ~~~~
48
- serve_utils.serve_flow(
49
  cl=cl,
50
  flow_class_name="flow_modules.aiflows.InterpreterFlowModule.InterpreterAtomicFlow",
51
  flow_endpoint="InterpreterAtomicFlow",
@@ -55,7 +55,7 @@ if __name__ == "__main__":
55
  run_dispatch_worker_thread(cl)
56
 
57
  #5. ~~~~~Mount the flow and get its proxy~~~~~~
58
- proxy_flow= serve_utils.get_flow_instance(
59
  cl=cl,
60
  flow_endpoint="InterpreterAtomicFlow",
61
  user_id="local",
@@ -93,6 +93,6 @@ print("hi")
93
 
94
 
95
  #9. ~~~~~Optional: Unserve Flow~~~~~~
96
- # serve_utils.delete_served_flow(cl, "FlowModule")
97
 
98
 
 
11
  from aiflows import logging
12
  from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
13
 
14
+ from aiflows.utils import serving
15
  from aiflows.workers import run_dispatch_worker_thread
16
  from aiflows.messages import FlowMessage
17
  from aiflows.interfaces import KeyInterface
 
45
 
46
 
47
  #3. ~~~~ Serve The Flow ~~~~
48
+ serving.serve_flow(
49
  cl=cl,
50
  flow_class_name="flow_modules.aiflows.InterpreterFlowModule.InterpreterAtomicFlow",
51
  flow_endpoint="InterpreterAtomicFlow",
 
55
  run_dispatch_worker_thread(cl)
56
 
57
  #5. ~~~~~Mount the flow and get its proxy~~~~~~
58
+ proxy_flow= serving.get_flow_instance(
59
  cl=cl,
60
  flow_endpoint="InterpreterAtomicFlow",
61
  user_id="local",
 
93
 
94
 
95
  #9. ~~~~~Optional: Unserve Flow~~~~~~
96
+ # serving.delete_served_flow(cl, "FlowModule")
97
 
98