mirror of
https://github.com/trycua/computer.git
synced 2026-01-06 21:39:58 -06:00
133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
"""
|
|
Example usage of the proxy server and client requests.
|
|
"""
|
|
import dotenv
|
|
dotenv.load_dotenv()
|
|
|
|
import asyncio
|
|
import json
|
|
import os
|
|
import aiohttp
|
|
from typing import Dict, Any
|
|
|
|
|
|
async def test_http_endpoint():
|
|
"""Test the HTTP /responses endpoint."""
|
|
|
|
anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
|
|
assert isinstance(anthropic_api_key, str), "ANTHROPIC_API_KEY environment variable must be set"
|
|
|
|
# Test requests
|
|
base_url = "https://m-linux-96lcxd2c2k.containers.cloud.trycua.com:8443"
|
|
# base_url = "http://localhost:8000"
|
|
api_key = os.getenv("CUA_API_KEY")
|
|
assert isinstance(api_key, str), "CUA_API_KEY environment variable must be set"
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
for i, request_data in enumerate([
|
|
# ==== Request Body Examples ====
|
|
|
|
# Simple text request
|
|
{
|
|
"model": "anthropic/claude-3-5-sonnet-20241022",
|
|
"input": "Hello!",
|
|
"env": {
|
|
"ANTHROPIC_API_KEY": anthropic_api_key
|
|
}
|
|
},
|
|
|
|
# Multimodal request with image
|
|
# {
|
|
# "model": "anthropic/claude-3-5-sonnet-20241022",
|
|
# "input": [
|
|
# {
|
|
# "role": "user",
|
|
# "content": [
|
|
# {"type": "input_text", "text": "what is in this image?"},
|
|
# {
|
|
# "type": "input_image",
|
|
# "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
|
|
# }
|
|
# ]
|
|
# }
|
|
# ],
|
|
# "env": {
|
|
# "ANTHROPIC_API_KEY": anthropic_api_key
|
|
# }
|
|
# }
|
|
|
|
], 1):
|
|
print(f"\n--- Test {i} ---")
|
|
print(f"Request: {json.dumps(request_data, indent=2)}")
|
|
|
|
try:
|
|
print(f"Sending request to {base_url}/responses")
|
|
async with session.post(
|
|
f"{base_url}/responses",
|
|
json=request_data,
|
|
headers={"Content-Type": "application/json", "X-API-Key": api_key}
|
|
) as response:
|
|
result = await response.json()
|
|
print(f"Status: {response.status}")
|
|
print(f"Response: {json.dumps(result, indent=2)}")
|
|
print(f"Response Headers: {response.headers}")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
|
|
async def test_p2p_client():
|
|
"""Example P2P client using peerjs-python."""
|
|
try:
|
|
from peerjs import Peer, PeerOptions, ConnectionEventType
|
|
from aiortc import RTCConfiguration, RTCIceServer
|
|
|
|
# Set up client peer
|
|
options = PeerOptions(
|
|
host="0.peerjs.com",
|
|
port=443,
|
|
secure=True,
|
|
config=RTCConfiguration(
|
|
iceServers=[RTCIceServer(urls="stun:stun.l.google.com:19302")]
|
|
)
|
|
)
|
|
|
|
client_peer = Peer(id="test-client", peer_options=options)
|
|
await client_peer.start()
|
|
|
|
# Connect to proxy server
|
|
connection = client_peer.connect("computer-agent-proxy")
|
|
|
|
@connection.on(ConnectionEventType.Open)
|
|
async def connection_open():
|
|
print("Connected to proxy server")
|
|
|
|
# Send a test request
|
|
request = {
|
|
"model": "anthropic/claude-3-5-sonnet-20241022",
|
|
"input": "Hello from P2P client!"
|
|
}
|
|
await connection.send(json.dumps(request))
|
|
|
|
@connection.on(ConnectionEventType.Data)
|
|
async def connection_data(data):
|
|
print(f"Received response: {data}")
|
|
await client_peer.destroy()
|
|
|
|
# Wait for connection
|
|
await asyncio.sleep(10)
|
|
|
|
except ImportError:
|
|
print("P2P dependencies not available. Install peerjs-python for P2P testing.")
|
|
except Exception as e:
|
|
print(f"P2P test error: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
if len(sys.argv) > 1 and sys.argv[1] == "p2p":
|
|
asyncio.run(test_p2p_client())
|
|
else:
|
|
asyncio.run(test_http_endpoint())
|