fix: json typo

Signed-off-by: hang lv <xlv20@fudan.edu.cn>
mosecorg · Jun 12, 2023 · 0c288e3 · 0c288e3
1 parent 1672cf8
commit 0c288e3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mosec/server.py b/mosec/server.py
@@ -239,7 +239,9 @@ def make_body(description, mime, schema):
                 request_worker_cls.resp_mime_type,
                 input_schema,
             ),
-            "responses": {
+            "responses": None
+            if not return_schema
+            else {
                 200: make_body(
                     "Mosec Inference Result",
                     response_worker_cls.resp_mime_type,

diff --git a/src/main.rs b/src/main.rs
@@ -268,7 +268,7 @@ async fn run(opts: &Opts) {
     let doc = MosecApiDoc {
         api: RustApiDoc::openapi(),
     }
-    .merge("/inference", python_api.parse().unwrap())
+    .merge("/inference", python_api.parse().unwrap_or_default())
     .move_path("/inference", &opts.endpoint);
 
     let state = AppState {