v1.1.0
Browse files
run.py
CHANGED
|
@@ -10,7 +10,7 @@ from aiflows.utils.general_helpers import read_yaml_file, quick_load_api_keys
|
|
| 10 |
from aiflows import logging
|
| 11 |
from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
|
| 12 |
|
| 13 |
-
from aiflows.utils import
|
| 14 |
from aiflows.workers import run_dispatch_worker_thread
|
| 15 |
from aiflows.messages import FlowMessage
|
| 16 |
from aiflows.interfaces import KeyInterface
|
|
@@ -52,7 +52,7 @@ if __name__ == "__main__":
|
|
| 52 |
|
| 53 |
|
| 54 |
#3. ~~~~ Serve The Flow ~~~~
|
| 55 |
-
|
| 56 |
cl = cl,
|
| 57 |
flow_class_name="flow_modules.aiflows.VisionFlowModule.VisionAtomicFlow",
|
| 58 |
flow_endpoint="VisionAtomicFlow",
|
|
@@ -62,7 +62,7 @@ if __name__ == "__main__":
|
|
| 62 |
run_dispatch_worker_thread(cl)
|
| 63 |
|
| 64 |
#5. ~~~~~Mount the flow and get an instance of it via a proxy~~~~~~
|
| 65 |
-
proxy_flow=
|
| 66 |
cl=cl,
|
| 67 |
flow_endpoint="VisionAtomicFlow",
|
| 68 |
user_id="local",
|
|
@@ -118,5 +118,5 @@ if __name__ == "__main__":
|
|
| 118 |
|
| 119 |
|
| 120 |
#9. ~~~~~Optional: Unserve Flow~~~~~~
|
| 121 |
-
#
|
| 122 |
|
|
|
|
| 10 |
from aiflows import logging
|
| 11 |
from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
|
| 12 |
|
| 13 |
+
from aiflows.utils import serving
|
| 14 |
from aiflows.workers import run_dispatch_worker_thread
|
| 15 |
from aiflows.messages import FlowMessage
|
| 16 |
from aiflows.interfaces import KeyInterface
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
#3. ~~~~ Serve The Flow ~~~~
|
| 55 |
+
serving.serve_flow(
|
| 56 |
cl = cl,
|
| 57 |
flow_class_name="flow_modules.aiflows.VisionFlowModule.VisionAtomicFlow",
|
| 58 |
flow_endpoint="VisionAtomicFlow",
|
|
|
|
| 62 |
run_dispatch_worker_thread(cl)
|
| 63 |
|
| 64 |
#5. ~~~~~Mount the flow and get an instance of it via a proxy~~~~~~
|
| 65 |
+
proxy_flow= serving.get_flow_instance(
|
| 66 |
cl=cl,
|
| 67 |
flow_endpoint="VisionAtomicFlow",
|
| 68 |
user_id="local",
|
|
|
|
| 118 |
|
| 119 |
|
| 120 |
#9. ~~~~~Optional: Unserve Flow~~~~~~
|
| 121 |
+
# serving.delete_served_flow(cl, "VisionFlowModule")
|
| 122 |
|