antigravity-automation 1.5.0

Python client library for the Antigravity Automation API — automate clicks, send commands, stream live AI outputs.

antigravity-automation

The official Python SDK for the Antigravity Automation extension

🌐 Website  ·  📖 API Docs  ·  ⬇️ Install Extension  ·  ⭐ GitHub

---

Automate clicks, send commands, and stream live AI chat outputs from Python — all via the local REST API and WebSocket provided by the Antigravity Automation VS Code extension.

Install

pip install antigravity-automation

Quick Start

from antigravity_automation import AntigravityClient

client = AntigravityClient()

# Send a command to the AI chat
result = client.send_command("Refactor the auth module")
print(result)  # SendCommandResponse(status='queued', position=1, ...)

# Toggle auto-click automation
client.toggle_auto_run()

# Get usage stats
stats = client.get_stats()
print(f"Auto-Run clicks: {stats.autoRunClicks}")

WebSocket Stream

from antigravity_automation import AntigravityStream

stream = AntigravityStream()
stream.on_message(lambda msg: print(f"[{msg['title']}] {msg['content'][:80]}..."))
stream.on_connected(lambda: print("Connected!"))
stream.run()  # Blocking — runs asyncio event loop

API Reference

Method	Endpoint	Description
send_command(text)	POST /send_command	Send a prompt to the active AI chat
start_new_chat()	POST /start-new-chat	Start a new chat session
switch_chat(title)	POST /switch_chat	Switch to a conversation by title
toggle_auto_run()	POST /toggle_auto_run	Toggle auto-click Run button
toggle_auto_allow()	POST /toggle_auto_allow	Toggle auto-approve permissions
get_command()	GET /get_command	Poll for queued commands & state
get_stats()	GET /stats	Get cumulative usage statistics
update(title, content)	POST /update	Push chat content
track_action(action)	POST /track_action	Track a usage action

Configuration

client = AntigravityClient(
    base_url="http://localhost:5000",  # default
    timeout=10,                         # seconds, default
)

stream = AntigravityStream(
    url="ws://localhost:9812",   # default
    reconnect=True,              # auto-reconnect, default
    reconnect_delay=3.0,         # seconds, default
)

Local Development

pip install -e ".[dev]"          # Install in dev mode
python tests/test_import.py      # Run offline import tests
python tests/test_live.py        # Run live tests (bridge must be running)
python -m build                  # Build for PyPI
twine upload dist/*              # Publish to PyPI

Requirements

• Python ≥ 3.8
• Antigravity Automation extension installed and running in VS Code

License

MIT — Built by JoeCodeCreations