File size: 5,778 Bytes
fc0b00d
 
 
 
 
 
 
 
ed887c7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
title: APIarium
emoji: 🐝
colorFrom: blue
colorTo: green
sdk: docker
app_port: 1111
---

# 🐝 APIarium

OpenAI-compatible LLM API gateway with multi-provider routing, encrypted configuration, and admin management.

## Features

- **Multi-Model Routing** β€” Route requests to multiple LLM providers through a unified OpenAI-compatible API
- **Encrypted Config** β€” Sensitive values (API keys, tokens) stored as AES-256-GCM encrypted blobs in the repo
- **Admin Dashboard** β€” Web UI for managing API keys, providers, and balance monitoring
- **Cloudflare Turnstile** β€” Bot protection for admin endpoints
- **VPN Detection** β€” Free VPN/proxy detection via proxycheck.io
- **IP Blocking** β€” Built-in IP blocklist support
- **GitHub Gist Storage** β€” API key persistence via encrypted GitHub Gists
- **HuggingFace Dataset Logging** β€” Request/response logging to HF Datasets
- **Dynamic Providers** β€” Add/remove upstream providers at runtime

## Architecture

```
Client β†’ APIarium (FastAPI) β†’ Multiple LLM Upstreams
                  ↓
           Admin Panel (Turnstile protected)
                  ↓
           GitHub Gist (encrypted key storage)
           HF Dataset (request logging)
```

## Supported Models

Models are exposed as aliases through the `/v1/models` endpoint. Each alias routes to one or more upstream providers configured in `app/config.py`.

## API Endpoints

### Proxy

| Method | Endpoint | Description |
|--------|----------|-------------|
| `POST` | `/v1/chat/completions` | OpenAI-compatible chat completion |
| `POST` | `/v1/completions` | Text completions |
| `GET` | `/v1/models` | List available models |

### Admin

| Method | Endpoint | Description |
|--------|----------|-------------|
| `GET` | `/admin` | Admin dashboard (Turnstile protected) |
| `GET` | `/admin/keys` | API key management |
| `GET` | `/admin/providers` | Provider management |
| `GET` | `/admin/balance` | Balance monitoring |
| `POST` | `/admin/verify-turnstile` | Turnstile token verification |

### Health

| Method | Endpoint | Description |
|--------|----------|-------------|
| `GET` | `/health` | Health check |

## Environment Variables

| Variable | Required | Description |
|----------|----------|-------------|
| `AES_KEY` | βœ… Yes | Base64-encoded 32-byte AES-256 key for decrypting config secrets |
| `HF_TOKEN` | Optional | HuggingFace token for dataset logging |
| `TURNSTILE_SITE_KEY` | Optional | Cloudflare Turnstile site key for admin protection |
| `TURNSTILE_SECRET_KEY` | Optional | Cloudflare Turnstile secret key for server-side verification |
| `UPSTREAM_API_KEY` | Optional | Fallback API key for upstreams without specific keys |

## Configuration

### Encrypted Secrets

Sensitive values are encrypted using AES-256-GCM and stored in `app/config.py`:

```bash
# Encrypt a value locally
python encrypt_config.py "your-secret-value"
# Output: enc:v1:nonce:ciphertext
```

In code:

```python
GITHUB_TOKEN = _decrypt("enc:v1:...")
```

The `AES_KEY` environment variable must be set at runtime for decryption.

### Generating an AES Key

```python
import base64, os
key = base64.b64encode(os.urandom(32)).decode()
print(key)
```

Store this value as the `AES_KEY` secret in your deployment environment.

### Upstream Routing

Models are mapped to upstream endpoints in the `UPSTREAMS` dict inside `app/config.py`:

```python
UPSTREAMS = {
    "model-alias": [
        {"url": "https://upstream.example.com/v1", "model": "real-model-name"},
    ],
}
```

### Per-Upstream API Keys

```python
UPSTREAM_API_KEYS = {
    "https://upstream-a.example.com/v1": "sk-...",
    "https://upstream-b.example.com/v1": "sk-...",
}
```

If no key is configured for a URL, the `UPSTREAM_API_KEY` env var is used as fallback.

### Balance Endpoints

Provider balance checking via `BALANCE_ENDPOINTS`:

```python
BALANCE_ENDPOINTS = {
    "provider-name": "https://api.provider.com/v1/balance",
}
```

## Local Development

```bash
# Clone
git clone https://huggingface.co/spaces/rnilkyway/APIarium
cd APIarium

# Create virtual environment
python -m venv venv
source venv/bin/activate  # Windows: venv\Scripts\activate

# Install dependencies
pip install -r requirements.txt

# Set required env vars
export AES_KEY="your-base64-encoded-32-byte-key"

# Run
uvicorn app.index:app --host 0.0.0.0 --port 1111 --reload
```

## Docker

```bash
docker build -t apiarium .
docker run -p 1111:1111 -e AES_KEY="your-key" apiarium
```

## Deployment on HuggingFace Spaces

This space runs on **HuggingFace Spaces** using the Docker SDK (`app_port: 1111`).

### Setting Secrets

1. Go to **Space Settings β†’ Variables and secrets**
2. Add `AES_KEY` with your base64-encoded 32-byte key
3. Optionally add `TURNSTILE_SITE_KEY` and `TURNSTILE_SECRET_KEY`
4. Optionally add `HF_TOKEN` for dataset logging

### Triggering a Rebuild

Any commit to the `main` branch automatically triggers a rebuild.

## Project Structure

```
APIarium/
β”œβ”€β”€ app/
β”‚   β”œβ”€β”€ index.py        # Main FastAPI application
β”‚   β”œβ”€β”€ config.py       # Configuration & encrypted secrets
β”‚   └── templates.py    # HTML templates for admin UI
β”œβ”€β”€ encrypt_config.py   # AES encryption utility
β”œβ”€β”€ test_gist.py        # Gist storage integration tests
β”œβ”€β”€ Dockerfile          # Container definition
β”œβ”€β”€ requirements.txt    # Python dependencies
└── README.md           # This file
```

## Security

- All sensitive config values are AES-256-GCM encrypted at rest
- Master key authentication for admin access
- Cloudflare Turnstile bot protection on admin endpoints
- Free VPN/proxy detection and automatic blocking
- IP blocklist support
- CORS configured for allowed origins
- Encrypted GitHub Gist storage for API key persistence