Skip to content

askui/vision-agent

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ€– AskUI Vision Agent

⚑ Automate computer tasks in Python ⚑

Release Notes PyPI - License

Join the AskUI Discord.

AskUI_VisionAgentsforEnterprise.1.mp4

πŸ”§ Setup

1. Install AskUI Agent OS

Agent OS is a device controller that allows agents to take screenshots, move the mouse, click, and type on the keyboard across any operating system.

Windows
AMD64

AskUI Installer for AMD64

ARM64

AskUI Installer for ARM64

Linux

⚠️ Warning: Agent OS currently does not work on Wayland. Switch to XOrg to use it.

AMD64
curl -L -o /tmp/AskUI-Suite-Latest-User-Installer-Linux-x64-Full.run https://files.askui.com/releases/Installer/Latest/AskUI-Suite-Latest-User-Installer-Linux-x64-Full.run
bash /tmp/AskUI-Suite-Latest-User-Installer-Linux-x64-Full.run
ARM64
curl -L -o /tmp/AskUI-Suite-Latest-User-Installer-Linux-ARM64-Full.run https://files.askui.com/releases/Installer/Latest/AskUI-Suite-Latest-User-Installer-Linux-ARM64-Full.run
bash /tmp/AskUI-Suite-Latest-User-Installer-Linux-ARM64-Full.run
MacOS
curl -L -o /tmp/AskUI-Suite-Latest-User-Installer-MacOS-ARM64-Full.run https://files.askui.com/releases/Installer/Latest/AskUI-Suite-Latest-User-Installer-MacOS-ARM64-Full.run
bash /tmp/AskUI-Suite-Latest-User-Installer-MacOS-ARM64-Full.run

2. Install vision-agent in your Python environment

pip install askui

Note: Requires Python version >=3.10.

3a. Authenticate with an AI Model Provider

AskUI INFO Anthropic INFO
ENV Variables ASKUI_WORKSPACE_ID, ASKUI_TOKEN ANTHROPIC_API_KEY
Supported Commands click() click(), get(), act()
Description Faster Inference, European Server, Enterprise Ready Supports complex actions

To get started, set the environment variables required to authenticate with your chosen model provider.

How to set an environment variable?

Linux & MacOS

Use export to set an evironment variable:

export ANTHROPIC_API_KEY=<your-api-key-here>
Windows PowerShell

Set an environment variable with $env:

$env:ANTHROPIC_API_KEY="<your-api-key-here>"

3b. Test with πŸ€— Hugging Face AI Models (Spaces API)

You can test the Vision Agent with Hugging Face models via their Spaces API. Please note that the API is rate-limited so for production use cases, it is recommended to choose step 3a.

Note: Hugging Face Spaces host model demos provided by individuals not associated with Hugging Face or AskUI. Don't use these models on screens with sensible information.

Supported Models:

Example Code:

agent.click("search field", model_name="OS-Copilot/OS-Atlas-Base-7B")

3c. Host your own AI Models

UI-TARS

You can use Vision Agent with UI-TARS if you provide your own UI-TARS API endpoint.

  1. Step: Host the model locally or in the cloud. More information about hosting UI-TARS can be found here.

  2. Step: Provide the TARS_URL and TARS_API_KEY environment variables to Vision Agent.

  3. Step: Use the model_name="tars" parameter in your click(), get() and act() commands.

▢️ Start Building

from askui import VisionAgent

# Initialize your agent context manager
with VisionAgent() as agent:
    # Use the webbrowser tool to start browsing
    agent.tools.webbrowser.open_new("http://www.google.com")

    # Start to automate individual steps
    agent.click("url bar")
    agent.type("http://www.google.com")
    agent.keyboard("enter")

    # Extract information from the screen
    datetime = agent.get("What is the datetime at the top of the screen?")
    print(datetime)

    # Or let the agent work on its own
    agent.act("search for a flight from Berlin to Paris in January")

πŸŽ›οΈ Model Selection

Instead of relying on the default model for the entire automation script, you can specify a model for each click command using the model_name parameter.

AskUI Anthropic
click() askui-combo, askui-pta, askui-ocr anthropic-claude-3-5-sonnet-20241022

Example: agent.click("Preview", model_name="askui-combo")

Antrophic AI Models

Supported commands are: click(), type(), mouse_move(), get(), act()

Model Name Info Production Ready? Enterprise?
anthropic-claude-3-5-sonnet-20241022 The Computer Use model from Antrophic is an Large Action model (LAM), which can autonoumsly achive goals. e.g. "Book me a flight from Berlin to Rom" ❌ ❌

Note: Configure your Antrophic Model Provider here

AskUI AI Models

Supported commands are: click(), type(), mouse_move()

Model Name Info Production Ready? Enterprise? Teachable?
askui-pta PTA-1 (Prompt-to-Automation) is a vision language model (VLM) trained by AskUI which is trained to address all kindes of UI elements by a textual description e.g. "Login button", "Text login" βœ… βœ… βœ…
askui-ocr AskUI OCR is an OCR model trained to address texts on UI Screens e.g. "Login", "Search" βœ… βœ… βœ…
askui-combo AskUI Combo is an combination from the askui-pta and the askui-ocr model to improve the accuracy. βœ… βœ… βœ…
askui-ai-element AskUI AI Element allows you to address visual elements like icons or images by demonstrating what you looking for. Therfore you have to crop out the element and give it a name. βœ… βœ… βœ…

Note: Configure your AskUI Model Provider here

Huggingface AI Models (Spaces API)

Supported commands are: click(), type(), mouse_move()

Model Name Info Production Ready? Enterprise?
AskUI/PTA-1 PTA-1 (Prompt-to-Automation) is a vision language model (VLM) trained by AskUI which is trained to address all kindes of UI elements by a textual description e.g. "Login button", "Text login" ❌ ❌
OS-Copilot/OS-Atlas-Base-7B OS-Atlas-Base-7B is a Large Action Model (LAM), which can autonoumsly achive goals. e.g. "Please help me modify VS Code setting to hide all folders in the explorer view". This model is not in the act() command available ❌ ❌
showlab/ShowUI-2B showlab/ShowUI-2B is a Large Action Model (LAM), which can autonoumsly achive goals. e.g. "Search in google maps for Nahant". This model is not in the act() command available ❌ ❌
Qwen/Qwen2-VL-2B-Instruct Qwen/Qwen2-VL-2B-Instruct is a Visual Language Model (VLM) pre-trained on multiple dataset including UI data. This model is not in the act() command available ❌ ❌
Qwen/Qwen2-VL-7B-Instruct [Qwen/Qwen2-VL-7B-Instruct](https://github.com/QwenLM/Qwen2.5-VLB) is a Visual Language Model (VLM) pre-trained on multiple dataset including UI data. This model is not in the act()` command available ❌ ❌

Note: No authentication required! But rate-limited!

Self Hosted UI Models

Supported commands are: click(), type(), mouse_move(), get(), act()

Model Name Info Production Ready? Enterprise?
tars UI-Tars is a Large Action Model (LAM) based on Qwen2 and fine-tuned by ByteDance on UI data e.g. "Book me a flight to rom" ❌ ❌

Note: These models needs to been self hosted by yourself. (See here)

πŸ› οΈ Direct Tool Use

Under the hood agents are using a set of tools. You can also directly access these tools.

Agent OS

The controller for the operating system.

agent.tools.os.click("left", 2) # clicking
agent.tools.os.mouse(100, 100) # mouse movement
agent.tools.os.keyboard_tap("v", modifier_keys=["control"]) # Paste
# and many more

Web browser

The webbrowser tool powered by webbrowser allows you to directly access webbrowsers in your environment.

agent.tools.webbrowser.open_new("http://www.google.com")
# also check out open and open_new_tab

Clipboard

The clipboard tool powered by pyperclip allows you to interact with the clipboard.

agent.tools.clipboard.copy("...")
result = agent.tools.clipboard.paste()

πŸ“œ Logging & Reporting

You want a better understanding of what you agent is doing? Set the log_level to DEBUG. You can also generate a report of the automation run by setting enable_report to True.

import logging

with VisionAgent(log_level=logging.DEBUG, enable_report=True) as agent:
    agent...

πŸ–₯️ Multi-Monitor Support

You have multiple monitors? Choose which one to automate by setting display to 1 or 2.

with VisionAgent(display=1) as agent:
    agent...

What is AskUI Vision Agent?

AskUI Vision Agent is a versatile AI powered framework that enables you to automate computer tasks in Python.

It connects Agent OS with powerful computer use models like Anthropic's Claude Sonnet 3.5 v2 and the AskUI Prompt-to-Action series. It is your entry point for building complex automation scenarios with detailed instructions or let the agent explore new challenges on its own.

image

Agent OS is a custom-built OS controller designed to enhance your automation experience.

It offers powerful features like

  • multi-screen support,
  • support for all major operating systems (incl. Windows, MacOS and Linux),
  • process visualizations,
  • real Unicode character typing

and more exciting features like application selection, in background automation and video streaming are to be released soon.