Make osam serve optional

wkentaro · Jul 30, 2024 · df960ea · df960ea
1 parent 497345b
commit df960ea
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/osam/__main__.py b/osam/__main__.py
@@ -8,7 +8,6 @@
 import imgviz
 import numpy as np
 import PIL.Image
-import uvicorn
 from loguru import logger
 
 from . import __version__
@@ -97,8 +96,15 @@ def rm(model_name):
 @cli.command(help="Start server")
 @click.option("--reload", is_flag=True, help="reload server on file changes")
 def serve(reload):
-    logger.info("Starting server...")
-    uvicorn.run("osam._server:app", host="127.0.0.1", port=11368, reload=reload)
+    try:
+        import uvicorn
+        import osam._server
+
+        logger.info("Starting server...")
+        uvicorn.run("osam._server:app", host="127.0.0.1", port=11368, reload=reload)
+    except ImportError:
+        logger.error("Run `pip install osam[serve]` to use `osam serve`")
+        sys.exit(1)
 
 
 @cli.command(help="Run a model")

diff --git a/pyproject.toml b/pyproject.toml
@@ -32,17 +32,19 @@ classifiers = [
 dependencies = [
   "click",
   "gdown",
-  "fastapi",
   "imgviz",
   "loguru",
   "onnxruntime",
   "Pillow",
   "pydantic",
-  "uvicorn",
 ]
 dynamic = ["readme", "version"]
 
 [project.optional-dependencies]
+serve = [
+  "fastapi",
+  "uvicorn",
+]
 test = [
   "build",
   "mypy",