start add api extras

MadcowD · Nov 2, 2024 · 29dd707 · 29dd707
1 parent bc77671
commit 29dd707
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ include = [
 ]
 
 [tool.poetry.dependencies]
-python = ">=3.9"
+python = ">=3.9,<4.0"
 numpy = ">=1.26.0"
 dill = "^0.3.8"
 colorama = "^0.4.6"
@@ -46,6 +46,9 @@ sqlmodel = { version = ">=0.0.21, <0.1.0", optional = true }
 # Studio
 fastapi = { version = "^0.111.1", optional = true }
 uvicorn = { version = "^0.30.3", optional = true }
+# Studio + API Server (optional)
+aiomqtt = { version="^2.3.0", optional = true }
+
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.2"
@@ -60,6 +63,8 @@ groq = ["groq"]
 sqlite = [ 'sqlmodel' ]
 postgres = ['sqlmodel', 'psychopg2']
 studio = ['fastapi', 'uvicorn', 'sqlmodel']
+api_server = ["fastapi", "uvicorn"]
+mqtt = ["aiomqtt"]
 all = [
     "anthropic",
     "groq",