feat: read config from stdin to allow pipe (#758)

jina-ai · Jun 20, 2022 · 6902d2d · 6902d2d
1 parent 4a298d4
commit 6902d2d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/server/clip_server/__main__.py b/server/clip_server/__main__.py
@@ -10,8 +10,16 @@
 
     from jina import Flow
 
+    if len(sys.argv) > 1:
+        if sys.argv[1] == '-i':
+            _input = sys.stdin.read()
+        else:
+            _input = sys.argv[1]
+    else:
+        _input = 'torch-flow.yml'
+
     f = Flow.load_config(
-        'torch-flow.yml' if len(sys.argv) == 1 else sys.argv[1],
+        _input,
         extra_search_paths=[os.path.dirname(inspect.getfile(inspect.currentframe()))],
     )
     with f: