feat: serve any fastapi app

jina-ai · May 12, 2023 · c713c5d · c713c5d
1 parent 204d298
commit c713c5d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/examples/fastapi/endpoints.py b/examples/fastapi/endpoints.py
@@ -0,0 +1,15 @@
+from typing import Union
+
+from fastapi import FastAPI
+
+app = FastAPI()
+
+
+@app.get("/status")
+def read_root():
+    return {"Hello": "World"}
+
+
+@app.get("/items/{item_id}")
+def read_item(item_id: int, q: Union[str, None] = None):
+    return {"item_id": item_id, "q": q}
diff --git a/tests/unit/test_flow.py b/tests/unit/test_flow.py
@@ -89,7 +89,7 @@ def test_get_jcloud_config(
 
 def test_get_flow_dict_for_local():
     flow_dict = get_flow_dict(
-        module="dummy",
+        module_str="dummy",
         jcloud=False,
         is_websocket=False,
     )
@@ -141,14 +141,14 @@ def test_get_flow_dict_for_jcloud(is_websocket, has_config, instance, autoscale_
         mocked_open.return_value.__enter__.return_value = io.StringIO(file_content)
         with patch("builtins.open", mocked_open):
             flow_dict = get_flow_dict(
-                module="dummy",
+                module_str="dummy",
                 jcloud=True,
                 jcloud_config_path="dummy.yaml",
                 is_websocket=is_websocket,
             )
     else:
         flow_dict = get_flow_dict(
-            module="dummy",
+            module_str="dummy",
             jcloud=True,
             jcloud_config_path=None,
             is_websocket=is_websocket,