From 39d60a852cbcf9dc50b559171e65aa94ea07fd11 Mon Sep 17 00:00:00 2001
From: Dillon DuPont <ddupont@mit.edu>
Date: Fri, 29 Aug 2025 11:33:04 -0400
Subject: [PATCH] Updated license of cua-som and cua-agent[omni]

---
 README.md                        | 8 ++++++++
 libs/python/agent/pyproject.toml | 4 ----
 libs/python/som/pyproject.toml   | 4 ++--
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 88cced63..5183598c 100644
--- a/README.md
+++ b/README.md
@@ -187,6 +187,14 @@ The base image `kasmweb/core-ubuntu-jammy` is maintained by [Kasm Technologies](
 
 Microsoft's OmniParser, which is used in this project, is licensed under the Creative Commons Attribution 4.0 International License (CC-BY-4.0) - see the [OmniParser LICENSE](https://github.com/microsoft/OmniParser/blob/master/LICENSE) file for details.
 
+### Third-Party Licenses and Optional Components
+
+Some optional extras for this project depend on third-party packages that are licensed under terms different from the MIT License.
+
+- The optional "omni" extra (installed via `pip install "cua-agent[omni]"`) installs the `cua-som` module, which includes `ultralytics` and is licensed under the AGPL-3.0.
+
+When you choose to install and use such optional extras, your use, modification, and distribution of those third-party components are governed by their respective licenses (e.g., AGPL-3.0 for `ultralytics`).
+
 ## Contributing
 
 We welcome contributions to Cua! Please refer to our [Contributing Guidelines](CONTRIBUTING.md) for details.
diff --git a/libs/python/agent/pyproject.toml b/libs/python/agent/pyproject.toml
index 4be2f6b6..c92c4dfa 100644
--- a/libs/python/agent/pyproject.toml
+++ b/libs/python/agent/pyproject.toml
@@ -30,7 +30,6 @@ requires-python = ">=3.12"
 openai = []
 anthropic = []
 omni = [
-    "ultralytics>=8.0.0",
     "cua-som>=0.1.0,<0.2.0",
 ]
 uitars = []
@@ -58,9 +57,6 @@ hud = [
     "hud-python>=0.4.12,<0.5.0",
 ]
 all = [
-    # omni requirements
-    "ultralytics>=8.0.0",
-    "cua-som>=0.1.0,<0.2.0",
     # uitars requirements
     "mlx-vlm>=0.1.27; sys_platform == 'darwin'",
     "accelerate",
diff --git a/libs/python/som/pyproject.toml b/libs/python/som/pyproject.toml
index 0bae7ea2..10b29ff8 100644
--- a/libs/python/som/pyproject.toml
+++ b/libs/python/som/pyproject.toml
@@ -26,12 +26,12 @@ dependencies = [
 ]
 requires-python = ">=3.11"
 readme = "README.md"
-license = {text = "MIT"}
+license = {text = "AGPL-3.0-or-later"}
 keywords = ["computer-vision", "ocr", "ui-analysis", "icon-detection"]
 classifiers = [
     "Development Status :: 4 - Beta",
+    "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
     "Intended Audience :: Developers",
-    "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.11",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",