REVERT files (#7)

.github/workflows/pre-commit.yml

@@ -31,4 +31,4 @@ jobs:
 
       - name: Run pre-commit
         run: |
-          uv run pre-commit run --all-files --show-diff-on-failure
+          make pre-commit

Makefile

@@ -1,4 +1,4 @@
-.PHONY: sync setup install dev-backend dev-frontend dev clean
+.PHONY: sync setup install dev-backend dev-frontend dev clean docker-build docker-run docker-stop docker-clean docker-logs
 
 # Sync all dependencies (Python + Node.js)
 sync:
@@ -23,6 +23,14 @@ dev-frontend:
 
 pre-commit:
 	uv run pre-commit run --all-files --show-diff-on-failure
+	make test
+
+# Run tests
+test:
+	cd cua2-core && uv run pytest tests/ -v
+
+test-coverage:
+	cd cua2-core && uv run pytest tests/ -v --cov=cua2_core --cov-report=html --cov-report=term
 
 clean:
 	find . -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true
@@ -30,3 +38,42 @@ clean:
 	find . -type d -name ".pytest_cache" -exec rm -rf {} + 2>/dev/null || true
 	cd cua2-front && rm -rf node_modules dist 2>/dev/null || true
 	@echo "✓ Cleaned!"
+
+# Docker commands
+docker-build:
+	@echo "Building Docker image..."
+	make docker-stop
+	docker build -t cua2:latest .
+	@echo "✓ Docker image built successfully!"
+
+docker-run:
+	@echo "Starting CUA2 container..."
+	@if [ -z "$$E2B_API_KEY" ]; then \
+		echo "Error: E2B_API_KEY environment variable is not set"; \
+		echo "Please set it with: export E2B_API_KEY=your-key"; \
+		exit 1; \
+	fi
+	@if [ -z "$$HF_TOKEN" ]; then \
+		echo "Error: HF_TOKEN environment variable is not set"; \
+		echo "Please set it with: export HF_TOKEN=your-token"; \
+		exit 1; \
+	fi
+	docker run -d --name cua2-app -p 7860:7860 \
+		-e E2B_API_KEY="$$E2B_API_KEY" \
+		-e HF_TOKEN="$$HF_TOKEN" \
+		cua2:latest
+	@echo "✓ Container started! Access at http://localhost:7860"
+
+docker-stop:
+	@echo "Stopping CUA2 container..."
+	docker stop cua2-app || true
+	docker rm cua2-app || true
+	@echo "✓ Container stopped!"
+
+docker-clean:
+	@echo "Removing CUA2 Docker images..."
+	docker rmi cua2:latest || true
+	@echo "✓ Docker images removed!"
+
+docker-logs:
+	docker logs -f cua2-app

cua2-core/src/cua2_core/app.py

@@ -1,3 +1,4 @@
+import os
 from contextlib import asynccontextmanager
 
 from cua2_core.services.agent_service import AgentService
@@ -17,6 +18,9 @@ async def lifespan(app: FastAPI):
     # Startup: Initialize services
     print("Initializing services...")
 
+    if not os.getenv("HF_TOKEN"):
+        raise ValueError("HF_TOKEN is not set")
+
     websocket_manager = WebSocketManager()
 
     sandbox_service = SandboxService()

cua2-core/src/cua2_core/websocket/websocket_manager.py

@@ -52,7 +52,7 @@ class WebSocketManager:
         try:
             await websocket.send_text(
                 json.dumps(
-                    message.model_dump(mode="json", context={"actions_as_json": True})
+                    message.model_dump(mode="json", context={"actions_as_json": False})
                 )
             )
         except Exception as e: