Spaces:
Running
Running
File size: 8,238 Bytes
09d390b f0f6e5c 29d02e9 09d390b 6a07eb6 5da6291 6a07eb6 09d390b 6a07eb6 5da6291 6a07eb6 f0f6e5c 719511c f0f6e5c 719511c f0f6e5c 719511c f0f6e5c 09d390b f0f6e5c f590683 f0f6e5c 09d390b f0f6e5c 701b92f 91df1e2 f0f6e5c 29d02e9 f0f6e5c 09d390b b44d59e 09d390b f0f6e5c 09d390b 719511c 514e5bf 5902c97 514e5bf 719511c b287eb9 f0f6e5c 09d390b 514e5bf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 |
<div align="center">
<img src="assets/proxy-lite.png" alt="Proxy Lite logo" width="600" height="auto" style="margin-bottom: 20px;" />
<h2>
A mini, open-weights, version of our Proxy assistant.
</h2>
<!-- Badges -->
<p>
<a href="https://github.com/convergence-ai/proxy-lite/contributors">
<img src="https://img.shields.io/github/contributors/convergence-ai/proxy-lite" alt="contributors" />
</a>
<a href="">
<img src="https://img.shields.io/github/last-commit/convergence-ai/proxy-lite" alt="last update" />
</a>
<a href="https://github.com/convergence-ai/proxy-lite/network/members">
<img src="https://img.shields.io/github/forks/convergence-ai/proxy-lite" alt="forks" />
</a>
<a href="https://github.com/convergence-ai/proxy-lite/stargazers">
<img src="https://img.shields.io/github/stars/convergence-ai/proxy-lite" alt="stars" />
</a>
<a href="https://github.com/convergence-ai/proxy-lite/issues/">
<img src="https://img.shields.io/github/issues/convergence-ai/proxy-lite" alt="open issues" />
</a>
<a href="https://github.com/convergence-ai/proxy-lite/blob/master/LICENSE">
<img src="https://img.shields.io/github/license/convergence-ai/proxy-lite.svg" alt="license" />
</a>
</p>
</div>
<div align="center">
<img src="assets/demo.gif" alt="Proxy Lite Demo" />
</div>
## Installation
Clone the repository:
```bash
git clone https://github.com/convergence-ai/proxy-lite.git
```
Set-up the environment with:
```bash
make proxy
```
Or do it manually:
```bash
pip install uv
uv venv --python 3.11 --python-preference managed
uv sync
uv pip install -e .
playwright install
```
## Usage
```bash
proxy --help
```
You can directly run Proxy Lite on a task with:
```bash
proxy "Book a table for 2 at an Italian restaurant in Kings Cross tonight at 7pm."
```
Alternatively you can run the local web ui with:
```bash
make app
```
### Proxy Lite Endpoint
By default, Proxy Lite will point to an endpoint set up on HuggingFace spaces.
> ❗ This is a demo endpoint and is not suitable for production, or even frequent hobbyist, use; it may be very slow when under even moderate load.
We recommend hosting your own endpoint with vLLM, you can use the following command:
```bash
vllm serve --model convergence-ai/proxy-lite \
--trust-remote-code \
--enable-auto-tool-choice \
--tool-call-parser hermes \
--port 8008 \
```
The tool arguments are **very important** for parsing the tool calls from the model appropriately.
> **Important:** To serve the model locally, install vLLM and transformers with `uv sync --all-extras`. Qwen-2.5-VL support is not yet available in the latest release of `transformers` so installation from source is required (the appropriate revision is specified in the `pyproject.toml` file).
You can set the `api_base` to point to your local endpoint when calling Proxy Lite:
```bash
proxy --api-base http://localhost:8008/v1 "Book a table...
```
or by setting the environment variable:
```bash
export PROXY_LITE_API_BASE=http://localhost:8008/v1
```
### Scaffolding Proxy Lite in Python
We use the `RunnerConfig` to control the setup of the task.
The library is designed to be modular and extendable, you can easily swap the environment, solver, or agent.
Example:
```python
import asyncio
from proxy_lite import Runner, RunnerConfig
config = RunnerConfig.from_dict(
{
"environment": {
"name": "webbrowser",
"homepage": "https://www.google.com",
"headless": True, # Don't show the browser
},
"solver": {
"name": "simple",
"agent": {
"name": "proxy_lite",
"client": {
"name": "convergence",
"model_id": "convergence-ai/proxy-lite",
"api_base": "https://convergence-ai-demo-api.hf.space/v1",
},
},
},
"max_steps": 50,
"action_timeout": 1800,
"environment_timeout": 1800,
"task_timeout": 18000,
"logger_level": "DEBUG",
},
)
proxy = Runner(config=config)
result = asyncio.run(
proxy.run("Book a table for 2 at an Italian restaurant in Kings Cross tonight at 7pm.")
)
```
The `Runner` sets the solver and environment off in a loop, like in a traditional reinforcement learning setup.
<div align="center">
<img src="assets/loop.png" alt="Runner Loop" width="700" height="auto" style="margin-bottom: 20px;" />
</div>
When it comes to prompting Proxy Lite, the model expects a message history of the form:
```python
message_history = [
{
"role": "system",
"content": "You are Proxy Lite...", # Full system prompt in src/proxy_lite/agents/proxy_lite_agent.py
}, # System prompt
{
"role": "user",
"content": "Book a table for 2 at an Italian restaurant in Kings Cross tonight at 7pm.",
}, # Set the task
{
"role": "user",
"content": [
{"type": "image_url", "image_url": {base64_encoded_screenshot} },
{"type": "text", "text": "URL: https://www.google.com/ \n- [0] <a>About</a> \n- [1] <a>Store</a>...."}
] # This is the observation from the environment
},
]
```
This would then build up the message history, alternating between the assistant (action) and the user (observation), although for new calls, all the last observations other than the current one are discarded.
The chat template will format this automatically, but also expects the appropriate `Tools` to be passed in so that the model is aware of the available actions. You can do this with `transformers`:
```python
from qwen_vl_utils import process_vision_info
from transformers import AutoProcessor
from proxy_lite.tools import ReturnValueTool, BrowserTool
from proxy_lite.serializer import OpenAICompatableSerializer
processor = AutoProcessor.from_pretrained("convergence-ai/proxy-lite")
tools = OpenAICompatableSerializer().serialize_tools([ReturnValueTool(), BrowserTool(session=None)])
templated_messages = processor.apply_chat_template(
message_history, tokenize=False, add_generation_prompt=True, tools=tools
)
image_inputs, video_inputs = process_vision_info(message_history)
batch = processor(
text=[templated_messages],
images=image_inputs,
videos=video_inputs,
padding=True,
return_tensors="pt",
)
```
Or you can send to the endpoint directly, which will handle the formatting:
```python
from openai import OpenAI
client = OpenAI(api_base="http://convergence-ai-demo-api.hf.space/v1")
response = client.chat.completions.create(
model="convergence-ai/proxy-lite",
messages=message_history,
tools=tools,
tool_choice="auto",
)
```
### Webbrowser Environment
The `webbrowser` environment is a simple environment that uses the `playwright` library to navigate the web.
We launch a Chromium browser and navigate to the `homepage` provided in the `RunnerConfig`.
Actions in an environment are defined through available tool calls, which in the browser case are set as default in the `BrowserTool` class. This allows the model to click, type, etc. at relevant `mark_id` elements on the page. These elements are extracted using JavaScript injected into the page in order to make interaction easier for the models.
**Note:** We use `playwright_stealth` to lower the chance of detection by anti-bot services, but this isn't foolproof and Proxy Lite may still get blocked with captchas or other anti-bot measures, especially when using the `headless` flag. We recommend using network proxies to avoid this issue.
## Limitations
This model has not currently been designed to act as a full assistant that can interact with the user, and is instead designed to as a tool that will go out and *autonomously* complete the task set.
As such, it will struggle with tasks that require credentials or user interaction such as actually purchasing items if you don't give all the required details in the prompt.
## Citation
```bibtex
@article{proxy-lite,
title={Proxy Lite - A Mini, Open-weights, Autonomous Assistant},
author={Convergence AI},
url={https://github.com/convergence-ai/proxy-lite},
year={2025}
}
``` |