From db823ac5612ae8f9b6cad8f22309835ca89ebfc8 Mon Sep 17 00:00:00 2001
From: Dillon DuPont <v-ddupont@microsoft.com>
Date: Sat, 10 May 2025 17:49:32 -0400
Subject: [PATCH] moved mlx into optional dep

---
 libs/agent/README.md      | 1 +
 libs/agent/pyproject.toml | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/libs/agent/README.md b/libs/agent/README.md
index 07f3d3fd..3a255c71 100644
--- a/libs/agent/README.md
+++ b/libs/agent/README.md
@@ -32,6 +32,7 @@ pip install "cua-agent[all]"
 pip install "cua-agent[openai]" # OpenAI Cua Loop
 pip install "cua-agent[anthropic]" # Anthropic Cua Loop
 pip install "cua-agent[uitars]"    # UI-Tars support
+pip install "cua-agent[uitars-mlx]"    # local UI-Tars support with MLXVLM
 pip install "cua-agent[omni]" # Cua Loop based on OmniParser (includes Ollama for local models)
 pip install "cua-agent[ui]" # Gradio UI for the agent
 ```
diff --git a/libs/agent/pyproject.toml b/libs/agent/pyproject.toml
index 5f23c2a1..89d14883 100644
--- a/libs/agent/pyproject.toml
+++ b/libs/agent/pyproject.toml
@@ -36,6 +36,8 @@ openai = [
 ]
 uitars = [
     "httpx>=0.27.0,<0.29.0",
+]
+uitars-mlx = [
     "mlx-vlm @ git+https://github.com/ddupont808/mlx-vlm.git@stable/fix/qwen2-position-id"
 ]
 ui = [