data-agent-tunnel-client 0.1.8

Data Agent Tunnel client - proxy local web services to public network

Data Agent Tunnel Client

Proxy local web services to the public network through Data Agent Tunnel.

Installation

pip install data-agent-tunnel-client

# or with uv
uv add data-agent-tunnel-client

Quick Start

Option 1: Simple Integration (Recommended)

The easiest way to connect - runs in a background thread, perfect for Flask/Django:

from flask import Flask
from data_agent_tunnel_client import connect_tunnel

app = Flask(__name__)

@app.route("/")
def hello():
    return "Hello from Flask!"

if __name__ == "__main__":
    # Start tunnel in background (auto prints connection info)
    connect_tunnel(
        tunnel_url="wss://...",
        local_url="http://localhost:5000",
        secret_key="your-secret-key",
        home_path="/initial_path",     # optional
    )

    # Start Flask
    app.run(port=5000)

Sync wait for connection (get return value yourself):

# Wait for connection and handle the result yourself
runner = connect_tunnel(
    tunnel_url="wss://...",
    local_url="http://localhost:5000",
    secret_key="your-secret-key",
    wait_for_connect=30,  # Wait up to 30 seconds, auto enables silent mode
)

if runner.is_connected:
    print(f"Public URL: {runner.public_url}")
    print(f"Session ID: {runner.connected_session_id}")
else:
    print("Connection failed or timeout")

app.run(port=5000)

Option 2: FastAPI Integration

For FastAPI, use create_tunnel_lifespan():

from fastapi import FastAPI
from data_agent_tunnel_client import create_tunnel_lifespan, get_tunnel_client

lifespan=create_tunnel_lifespan(
    tunnel_url="wss://...",
    local_url="http://localhost:8000",
    secret_key="your-secret-key",
    home_path="/dashboard",        # optional
)

app = FastAPI(lifespan=lifespan)

@app.get("/")
async def root():
    return {"message": "Hello from FastAPI!"}

@app.get("/tunnel-info")
async def tunnel_info():
    client = get_tunnel_client()
    return {"public_url": client.public_url if client else None}

API Reference

connect_tunnel()

Quick start function for synchronous frameworks:

from data_agent_tunnel_client import connect_tunnel

runner = connect_tunnel(
    tunnel_url="wss://...",          # Tunnel WebSocket URL
    local_url="http://...",          # Local service URL
    secret_key="your-secret-key",    # Auth key (required)
    home_path="/dashboard",          # Initial path appended to public URL (optional)
    wait_for_connect=None,           # Sync wait timeout in seconds (optional)
    silent=False,                    # Suppress default output (optional)
)

# Access connection info
print(runner.public_url)
print(runner.connected_session_id)
print(runner.is_connected)

home_path parameter:

The home_path parameter specifies the initial path to display after tunnel connection. This is useful when your app's entry point is not the root path.

# Example: If public_url is https://xxx.com?session=abc123&_tunnel_path=,
# With home_path="/dashboard", the displayed URL will be:
# https://xxx.com?session=abc123&_tunnel_path=/dashboard

create_tunnel_lifespan()

Create a lifespan context manager for FastAPI:

from data_agent_tunnel_client import create_tunnel_lifespan

lifespan = create_tunnel_lifespan(
    tunnel_url="wss://...",          # Tunnel WebSocket URL
    local_url="http://...",          # Local service URL
    secret_key="your-secret-key",    # Auth key (required)
    home_path="/",                   # Home path for display (optional)
    on_connect=None,                 # Connect callback (async or sync)
    on_disconnect=None,              # Disconnect callback (async or sync)
)

app = FastAPI(lifespan=lifespan)

Use get_tunnel_client() to access the client instance in your routes.

TunnelRunner

For more control over the background runner:

from data_agent_tunnel_client import TunnelRunner

runner = TunnelRunner(
    tunnel_url="wss://...",
    local_url="http://...",
    secret_key="your-secret-key",
    home_path="/",
    silent=False,  # Set True to suppress default output
    on_connect=lambda client: print(f"Connected: {client.public_url}"),
    on_disconnect=lambda client: print("Disconnected"),
)

runner.start()  # Non-blocking

# Sync wait for connection
if runner.wait_for_connect(timeout=30):
    print(f"Public URL: {runner.public_url}")
else:
    print("Connection failed")

TunnelClient

Low-level async client:

from data_agent_tunnel_client import TunnelClient

client = TunnelClient(
    tunnel_url="wss://...",             # Tunnel WebSocket URL
    local_url="http://...",             # Local service URL
    secret_key="your-secret-key",       # Auth key (required)
    session_id="",                      # Session ID (optional)
    reconnect=True,                     # Auto reconnect on disconnect
    reconnect_interval=5.0,             # Reconnect interval (seconds)
    ping_interval=30.0,                 # Heartbeat interval (seconds)
    request_timeout=300.0,              # Request timeout (seconds)
    max_concurrent_requests=100,        # Max concurrent requests
    on_connect=None,                    # Connect callback (async)
    on_disconnect=None,                 # Disconnect callback (async)
)

# Properties
client.public_url           # Public URL
client.connected_session_id # Session ID
client.is_connected         # Connection status

# Methods
await client.connect()      # Connect and start proxying
await client.disconnect()   # Disconnect

Proxy Support

The client automatically detects and uses system proxy settings (http_proxy, https_proxy, socks_proxy, etc.).

• If proxy is configured, it tries to connect via proxy first
• If proxy connection fails, it falls back to direct connection
• Use disable_proxy=True to skip proxy entirely

License

MIT