diff --git a/poetry.lock b/poetry.lock
index 29315d91..d77ffd90 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,5 +1,20 @@
 # This file is automatically @generated by Poetry 1.8.4 and should not be changed by hand.
 
+[[package]]
+name = "aiomqtt"
+version = "2.3.0"
+description = "The idiomatic asyncio MQTT client, wrapped around paho-mqtt"
+optional = true
+python-versions = "<4.0,>=3.8"
+files = [
+    {file = "aiomqtt-2.3.0-py3-none-any.whl", hash = "sha256:127926717bd6b012d1630f9087f24552eb9c4af58205bc2964f09d6e304f7e63"},
+    {file = "aiomqtt-2.3.0.tar.gz", hash = "sha256:312feebe20bc76dc7c20916663011f3bd37aa6f42f9f687a19a1c58308d80d47"},
+]
+
+[package.dependencies]
+paho-mqtt = ">=2.1.0,<3.0.0"
+typing-extensions = {version = ">=4.4.0,<5.0.0", markers = "python_version < \"3.10\""}
+
 [[package]]
 name = "alabaster"
 version = "0.7.16"
@@ -1121,6 +1136,20 @@ files = [
     {file = "packaging-24.1.tar.gz", hash = "sha256:026ed72c8ed3fcce5bf8950572258698927fd1dbda10a5e981cdf0ac37f4f002"},
 ]
 
+[[package]]
+name = "paho-mqtt"
+version = "2.1.0"
+description = "MQTT version 5.0/3.1.1 client class"
+optional = true
+python-versions = ">=3.7"
+files = [
+    {file = "paho_mqtt-2.1.0-py3-none-any.whl", hash = "sha256:6db9ba9b34ed5bc6b6e3812718c7e06e2fd7444540df2455d2c51bd58808feee"},
+    {file = "paho_mqtt-2.1.0.tar.gz", hash = "sha256:12d6e7511d4137555a3f6ea167ae846af2c7357b10bc6fa4f7c3968fc1723834"},
+]
+
+[package.extras]
+proxy = ["pysocks"]
+
 [[package]]
 name = "pathspec"
 version = "0.12.1"
@@ -1576,13 +1605,13 @@ use-chardet-on-py3 = ["chardet (>=3.0.2,<6)"]
 
 [[package]]
 name = "rich"
-version = "13.9.3"
+version = "13.9.4"
 description = "Render rich text, tables, progress bars, syntax highlighting, markdown and more to the terminal"
 optional = true
 python-versions = ">=3.8.0"
 files = [
-    {file = "rich-13.9.3-py3-none-any.whl", hash = "sha256:9836f5096eb2172c9e77df411c1b009bace4193d6a481d534fea75ebba758283"},
-    {file = "rich-13.9.3.tar.gz", hash = "sha256:bc1e01b899537598cf02579d2b9f4a415104d3fc439313a7a2c165d76557a08e"},
+    {file = "rich-13.9.4-py3-none-any.whl", hash = "sha256:6049d5e6ec054bf2779ab3358186963bac2ea89175919d699e378b99738c2a90"},
+    {file = "rich-13.9.4.tar.gz", hash = "sha256:439594978a49a09530cff7ebc4b5c7103ef57baf48d5ea3184f21d9a2befa098"},
 ]
 
 [package.dependencies]
@@ -2399,6 +2428,7 @@ type = ["pytest-mypy"]
 [extras]
 all = ["anthropic", "fastapi", "groq", "sqlmodel", "uvicorn"]
 anthropic = ["anthropic"]
+api-server = ["aiomqtt", "fastapi", "uvicorn"]
 groq = ["groq"]
 postgres = ["sqlmodel"]
 sqlite = ["sqlmodel"]
@@ -2406,5 +2436,5 @@ studio = ["fastapi", "sqlmodel", "uvicorn"]
 
 [metadata]
 lock-version = "2.0"
-python-versions = ">=3.9"
-content-hash = "a8e393e75d5f060226d0b25c4306a1d504ac68e92650a7aa34577e396620490d"
+python-versions = ">=3.9,<4.0"
+content-hash = "49a283549a81a5b13eade523c0cae915a003660185f199a60b21fee249a9e184"
diff --git a/pyproject.toml b/pyproject.toml
index 1a9991c0..dba1256a 100644
--- 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",