1
0

feat: 初始化 AI Gateway 项目

实现支持 OpenAI 和 Anthropic 双协议的统一大模型 API 网关 MVP 版本,包含:
- OpenAI 和 Anthropic 协议代理
- 供应商和模型管理
- 用量统计
- 前端配置界面
This commit is contained in:
2026-04-15 16:53:28 +08:00
commit 915b004924
53 changed files with 5662 additions and 0 deletions

37
frontend/src/App.tsx Normal file
View File

@@ -0,0 +1,37 @@
import { useState } from 'react';
import { ProvidersPage } from './pages/ProvidersPage';
import { StatsPage } from './pages/StatsPage';
import './App.css';
function App() {
const [currentPage, setCurrentPage] = useState<'providers' | 'stats'>('providers');
return (
<div className="app">
<nav className="navbar">
<h1>AI Gateway</h1>
<div className="nav-links">
<button
className={currentPage === 'providers' ? 'active' : ''}
onClick={() => setCurrentPage('providers')}
>
</button>
<button
className={currentPage === 'stats' ? 'active' : ''}
onClick={() => setCurrentPage('stats')}
>
</button>
</div>
</nav>
<main className="content">
{currentPage === 'providers' && <ProvidersPage />}
{currentPage === 'stats' && <StatsPage />}
</main>
</div>
);
}
export default App;

129
frontend/src/api/client.ts Normal file
View File

@@ -0,0 +1,129 @@
const API_BASE = 'http://localhost:9826/api';
export interface Provider {
id: string;
name: string;
api_key: string;
base_url: string;
enabled: boolean;
created_at: string;
updated_at: string;
}
export interface Model {
id: string;
provider_id: string;
model_name: string;
enabled: boolean;
created_at: string;
}
export interface UsageStats {
id: number;
provider_id: string;
model_name: string;
request_count: number;
date: string;
}
// Provider API
export async function listProviders(): Promise<Provider[]> {
const response = await fetch(`${API_BASE}/providers`);
if (!response.ok) throw new Error('Failed to fetch providers');
return response.json();
}
export async function createProvider(provider: Omit<Provider, 'created_at' | 'updated_at'>): Promise<Provider> {
const response = await fetch(`${API_BASE}/providers`, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(provider),
});
if (!response.ok) throw new Error('Failed to create provider');
return response.json();
}
export async function updateProvider(id: string, updates: Partial<Provider>): Promise<Provider> {
const response = await fetch(`${API_BASE}/providers/${id}`, {
method: 'PUT',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(updates),
});
if (!response.ok) throw new Error('Failed to update provider');
return response.json();
}
export async function deleteProvider(id: string): Promise<void> {
const response = await fetch(`${API_BASE}/providers/${id}`, { method: 'DELETE' });
if (!response.ok) throw new Error('Failed to delete provider');
}
// Model API
export async function listModels(providerId?: string): Promise<Model[]> {
const url = providerId ? `${API_BASE}/models?provider_id=${providerId}` : `${API_BASE}/models`;
const response = await fetch(url);
if (!response.ok) throw new Error('Failed to fetch models');
return response.json();
}
export async function createModel(model: Omit<Model, 'created_at'>): Promise<Model> {
const response = await fetch(`${API_BASE}/models`, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(model),
});
if (!response.ok) throw new Error('Failed to create model');
return response.json();
}
export async function updateModel(id: string, updates: Partial<Model>): Promise<Model> {
const response = await fetch(`${API_BASE}/models/${id}`, {
method: 'PUT',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(updates),
});
if (!response.ok) throw new Error('Failed to update model');
return response.json();
}
export async function deleteModel(id: string): Promise<void> {
const response = await fetch(`${API_BASE}/models/${id}`, { method: 'DELETE' });
if (!response.ok) throw new Error('Failed to delete model');
}
// Stats API
export async function getStats(params?: {
provider_id?: string;
model_name?: string;
start_date?: string;
end_date?: string;
}): Promise<UsageStats[]> {
const query = new URLSearchParams();
if (params?.provider_id) query.set('provider_id', params.provider_id);
if (params?.model_name) query.set('model_name', params.model_name);
if (params?.start_date) query.set('start_date', params.start_date);
if (params?.end_date) query.set('end_date', params.end_date);
const response = await fetch(`${API_BASE}/stats?${query}`);
if (!response.ok) throw new Error('Failed to fetch stats');
return response.json();
}
export async function getAggregatedStats(params?: {
provider_id?: string;
model_name?: string;
start_date?: string;
end_date?: string;
group_by?: 'provider' | 'model' | 'date';
}): Promise<any[]> {
const query = new URLSearchParams();
if (params?.provider_id) query.set('provider_id', params.provider_id);
if (params?.model_name) query.set('model_name', params.model_name);
if (params?.start_date) query.set('start_date', params.start_date);
if (params?.end_date) query.set('end_date', params.end_date);
if (params?.group_by) query.set('group_by', params.group_by);
const response = await fetch(`${API_BASE}/stats/aggregate?${query}`);
if (!response.ok) throw new Error('Failed to fetch aggregated stats');
return response.json();
}

Binary file not shown.

After

Width:  |  Height:  |  Size: 44 KiB

View File

@@ -0,0 +1 @@
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="35.93" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 228"><path fill="#00D8FF" d="M210.483 73.824a171.49 171.49 0 0 0-8.24-2.597c.465-1.9.893-3.777 1.273-5.621c6.238-30.281 2.16-54.676-11.769-62.708c-13.355-7.7-35.196.329-57.254 19.526a171.23 171.23 0 0 0-6.375 5.848a155.866 155.866 0 0 0-4.241-3.917C100.759 3.829 77.587-4.822 63.673 3.233C50.33 10.957 46.379 33.89 51.995 62.588a170.974 170.974 0 0 0 1.892 8.48c-3.28.932-6.445 1.924-9.474 2.98C17.309 83.498 0 98.307 0 113.668c0 15.865 18.582 31.778 46.812 41.427a145.52 145.52 0 0 0 6.921 2.165a167.467 167.467 0 0 0-2.01 9.138c-5.354 28.2-1.173 50.591 12.134 58.266c13.744 7.926 36.812-.22 59.273-19.855a145.567 145.567 0 0 0 5.342-4.923a168.064 168.064 0 0 0 6.92 6.314c21.758 18.722 43.246 26.282 56.54 18.586c13.731-7.949 18.194-32.003 12.4-61.268a145.016 145.016 0 0 0-1.535-6.842c1.62-.48 3.21-.974 4.76-1.488c29.348-9.723 48.443-25.443 48.443-41.52c0-15.417-17.868-30.326-45.517-39.844Zm-6.365 70.984c-1.4.463-2.836.91-4.3 1.345c-3.24-10.257-7.612-21.163-12.963-32.432c5.106-11 9.31-21.767 12.459-31.957c2.619.758 5.16 1.557 7.61 2.4c23.69 8.156 38.14 20.213 38.14 29.504c0 9.896-15.606 22.743-40.946 31.14Zm-10.514 20.834c2.562 12.94 2.927 24.64 1.23 33.787c-1.524 8.219-4.59 13.698-8.382 15.893c-8.067 4.67-25.32-1.4-43.927-17.412a156.726 156.726 0 0 1-6.437-5.87c7.214-7.889 14.423-17.06 21.459-27.246c12.376-1.098 24.068-2.894 34.671-5.345a134.17 134.17 0 0 1 1.386 6.193ZM87.276 214.515c-7.882 2.783-14.16 2.863-17.955.675c-8.075-4.657-11.432-22.636-6.853-46.752a156.923 156.923 0 0 1 1.869-8.499c10.486 2.32 22.093 3.988 34.498 4.994c7.084 9.967 14.501 19.128 21.976 27.15a134.668 134.668 0 0 1-4.877 4.492c-9.933 8.682-19.886 14.842-28.658 17.94ZM50.35 144.747c-12.483-4.267-22.792-9.812-29.858-15.863c-6.35-5.437-9.555-10.836-9.555-15.216c0-9.322 13.897-21.212 37.076-29.293c2.813-.98 5.757-1.905 8.812-2.773c3.204 10.42 7.406 21.315 12.477 32.332c-5.137 11.18-9.399 22.249-12.634 32.792a134.718 134.718 0 0 1-6.318-1.979Zm12.378-84.26c-4.811-24.587-1.616-43.134 6.425-47.789c8.564-4.958 27.502 2.111 47.463 19.835a144.318 144.318 0 0 1 3.841 3.545c-7.438 7.987-14.787 17.08-21.808 26.988c-12.04 1.116-23.565 2.908-34.161 5.309a160.342 160.342 0 0 1-1.76-7.887Zm110.427 27.268a347.8 347.8 0 0 0-7.785-12.803c8.168 1.033 15.994 2.404 23.343 4.08c-2.206 7.072-4.956 14.465-8.193 22.045a381.151 381.151 0 0 0-7.365-13.322Zm-45.032-43.861c5.044 5.465 10.096 11.566 15.065 18.186a322.04 322.04 0 0 0-30.257-.006c4.974-6.559 10.069-12.652 15.192-18.18ZM82.802 87.83a323.167 323.167 0 0 0-7.227 13.238c-3.184-7.553-5.909-14.98-8.134-22.152c7.304-1.634 15.093-2.97 23.209-3.984a321.524 321.524 0 0 0-7.848 12.897Zm8.081 65.352c-8.385-.936-16.291-2.203-23.593-3.793c2.26-7.3 5.045-14.885 8.298-22.6a321.187 321.187 0 0 0 7.257 13.246c2.594 4.48 5.28 8.868 8.038 13.147Zm37.542 31.03c-5.184-5.592-10.354-11.779-15.403-18.433c4.902.192 9.899.29 14.978.29c5.218 0 10.376-.117 15.453-.343c-4.985 6.774-10.018 12.97-15.028 18.486Zm52.198-57.817c3.422 7.8 6.306 15.345 8.596 22.52c-7.422 1.694-15.436 3.058-23.88 4.071a382.417 382.417 0 0 0 7.859-13.026a347.403 347.403 0 0 0 7.425-13.565Zm-16.898 8.101a358.557 358.557 0 0 1-12.281 19.815a329.4 329.4 0 0 1-23.444.823c-7.967 0-15.716-.248-23.178-.732a310.202 310.202 0 0 1-12.513-19.846h.001a307.41 307.41 0 0 1-10.923-20.627a310.278 310.278 0 0 1 10.89-20.637l-.001.001a307.318 307.318 0 0 1 12.413-19.761c7.613-.576 15.42-.876 23.31-.876H128c7.926 0 15.743.303 23.354.883a329.357 329.357 0 0 1 12.335 19.695a358.489 358.489 0 0 1 11.036 20.54a329.472 329.472 0 0 1-11 20.722Zm22.56-122.124c8.572 4.944 11.906 24.881 6.52 51.026c-.344 1.668-.73 3.367-1.15 5.09c-10.622-2.452-22.155-4.275-34.23-5.408c-7.034-10.017-14.323-19.124-21.64-27.008a160.789 160.789 0 0 1 5.888-5.4c18.9-16.447 36.564-22.941 44.612-18.3ZM128 90.808c12.625 0 22.86 10.235 22.86 22.86s-10.235 22.86-22.86 22.86s-22.86-10.235-22.86-22.86s10.235-22.86 22.86-22.86Z"></path></svg>

After

Width:  |  Height:  |  Size: 4.0 KiB

File diff suppressed because one or more lines are too long

After

Width:  |  Height:  |  Size: 8.5 KiB

View File

@@ -0,0 +1,117 @@
import { useState } from 'react';
import * as api from '../api/client';
interface ModelFormProps {
model?: api.Model;
providers: api.Provider[];
onSave: () => void;
onCancel: () => void;
}
export function ModelForm({ model, providers, onSave, onCancel }: ModelFormProps) {
const [id, setId] = useState(model?.id || '');
const [providerId, setProviderId] = useState(model?.provider_id || (providers[0]?.id || ''));
const [modelName, setModelName] = useState(model?.model_name || '');
const [enabled, setEnabled] = useState(model?.enabled ?? true);
const [loading, setLoading] = useState(false);
const [error, setError] = useState<string | null>(null);
const isEdit = !!model;
async function handleSubmit(e: React.FormEvent) {
e.preventDefault();
setLoading(true);
setError(null);
try {
if (isEdit) {
const updates: any = {};
if (providerId !== model.provider_id) updates.provider_id = providerId;
if (modelName !== model.model_name) updates.model_name = modelName;
if (enabled !== model.enabled) updates.enabled = enabled;
if (Object.keys(updates).length > 0) {
await api.updateModel(model.id, updates);
}
} else {
await api.createModel({
id,
provider_id: providerId,
model_name: modelName,
enabled,
});
}
onSave();
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
} finally {
setLoading(false);
}
}
return (
<div className="modal">
<div className="modal-content">
<h2>{isEdit ? '编辑模型' : '添加模型'}</h2>
{error && <div className="error">{error}</div>}
<form onSubmit={handleSubmit}>
<div className="form-group">
<label>ID</label>
<input
type="text"
value={id}
onChange={e => setId(e.target.value)}
disabled={isEdit}
required
/>
</div>
<div className="form-group">
<label></label>
<select
value={providerId}
onChange={e => setProviderId(e.target.value)}
required
>
{providers.map(p => (
<option key={p.id} value={p.id}>{p.name}</option>
))}
</select>
</div>
<div className="form-group">
<label></label>
<input
type="text"
value={modelName}
onChange={e => setModelName(e.target.value)}
required
/>
</div>
<div className="form-group">
<label>
<input
type="checkbox"
checked={enabled}
onChange={e => setEnabled(e.target.checked)}
/>
</label>
</div>
<div className="form-actions">
<button type="submit" disabled={loading}>
{loading ? '保存中...' : '保存'}
</button>
<button type="button" onClick={onCancel} disabled={loading}>
</button>
</div>
</form>
</div>
</div>
);
}

View File

@@ -0,0 +1,130 @@
import { useState } from 'react';
import * as api from '../api/client';
interface ProviderFormProps {
provider?: api.Provider;
onSave: () => void;
onCancel: () => void;
}
export function ProviderForm({ provider, onSave, onCancel }: ProviderFormProps) {
const [id, setId] = useState(provider?.id || '');
const [name, setName] = useState(provider?.name || '');
const [apiKey, setApiKey] = useState('');
const [baseUrl, setBaseUrl] = useState(provider?.base_url || '');
const [enabled, setEnabled] = useState(provider?.enabled ?? true);
const [loading, setLoading] = useState(false);
const [error, setError] = useState<string | null>(null);
const isEdit = !!provider;
async function handleSubmit(e: React.FormEvent) {
e.preventDefault();
setLoading(true);
setError(null);
try {
if (isEdit) {
const updates: any = {};
if (name !== provider.name) updates.name = name;
if (apiKey) updates.api_key = apiKey;
if (baseUrl !== provider.base_url) updates.base_url = baseUrl;
if (enabled !== provider.enabled) updates.enabled = enabled;
if (Object.keys(updates).length > 0) {
await api.updateProvider(provider.id, updates);
}
} else {
await api.createProvider({
id,
name,
api_key: apiKey,
base_url: baseUrl,
enabled,
});
}
onSave();
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
} finally {
setLoading(false);
}
}
return (
<div className="modal">
<div className="modal-content">
<h2>{isEdit ? '编辑供应商' : '添加供应商'}</h2>
{error && <div className="error">{error}</div>}
<form onSubmit={handleSubmit}>
<div className="form-group">
<label>ID</label>
<input
type="text"
value={id}
onChange={e => setId(e.target.value)}
disabled={isEdit}
required
/>
</div>
<div className="form-group">
<label></label>
<input
type="text"
value={name}
onChange={e => setName(e.target.value)}
required
/>
</div>
<div className="form-group">
<label>API Key {isEdit && '(留空则不修改)'}</label>
<input
type="password"
value={apiKey}
onChange={e => setApiKey(e.target.value)}
required={!isEdit}
/>
</div>
<div className="form-group">
<label>Base URL</label>
<input
type="url"
value={baseUrl}
onChange={e => setBaseUrl(e.target.value)}
placeholder="例如: https://api.openai.com/v1 或 https://open.bigmodel.cn/api/paas/v4"
required
/>
<small style={{color: '#666', fontSize: '0.85rem'}}>
API /chat/completions
</small>
</div>
<div className="form-group">
<label>
<input
type="checkbox"
checked={enabled}
onChange={e => setEnabled(e.target.checked)}
/>
</label>
</div>
<div className="form-actions">
<button type="submit" disabled={loading}>
{loading ? '保存中...' : '保存'}
</button>
<button type="button" onClick={onCancel} disabled={loading}>
</button>
</div>
</form>
</div>
</div>
);
}

10
frontend/src/main.tsx Normal file
View File

@@ -0,0 +1,10 @@
import { StrictMode } from 'react'
import { createRoot } from 'react-dom/client'
import './index.css'
import App from './App.tsx'
createRoot(document.getElementById('root')!).render(
<StrictMode>
<App />
</StrictMode>,
)

View File

@@ -0,0 +1,182 @@
import { useState, useEffect } from 'react';
import * as api from '../api/client';
import { ProviderForm } from '../components/ProviderForm';
import { ModelForm } from '../components/ModelForm';
export function ProvidersPage() {
const [providers, setProviders] = useState<api.Provider[]>([]);
const [models, setModels] = useState<api.Model[]>([]);
const [loading, setLoading] = useState(true);
const [error, setError] = useState<string | null>(null);
// 表单状态
const [showProviderForm, setShowProviderForm] = useState(false);
const [editingProvider, setEditingProvider] = useState<api.Provider | null>(null);
const [showModelForm, setShowModelForm] = useState(false);
const [editingModel, setEditingModel] = useState<api.Model | null>(null);
useEffect(() => {
loadData();
}, []);
async function loadData() {
try {
setLoading(true);
const [providersData, modelsData] = await Promise.all([
api.listProviders(),
api.listModels(),
]);
setProviders(providersData);
setModels(modelsData);
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
} finally {
setLoading(false);
}
}
async function handleDeleteProvider(id: string) {
if (!confirm('确定要删除这个供应商吗?关联的模型也会被删除。')) return;
try {
await api.deleteProvider(id);
loadData();
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
}
}
async function handleDeleteModel(id: string) {
if (!confirm('确定要删除这个模型吗?')) return;
try {
await api.deleteModel(id);
loadData();
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
}
}
if (loading) return <div className="loading">...</div>;
if (error) return <div className="error">{error}</div>;
return (
<div className="providers-page">
<h1></h1>
<div className="section">
<h2></h2>
<button onClick={() => {
setEditingProvider(null);
setShowProviderForm(true);
}}>
</button>
<table>
<thead>
<tr>
<th>ID</th>
<th></th>
<th>API Key</th>
<th>Base URL</th>
<th></th>
<th></th>
</tr>
</thead>
<tbody>
{providers.map(p => (
<tr key={p.id}>
<td>{p.id}</td>
<td>{p.name}</td>
<td>{p.api_key}</td>
<td>{p.base_url}</td>
<td>{p.enabled ? '启用' : '禁用'}</td>
<td>
<button onClick={() => {
setEditingProvider(p);
setShowProviderForm(true);
}}></button>
<button onClick={() => handleDeleteProvider(p.id)}></button>
</td>
</tr>
))}
</tbody>
</table>
</div>
<div className="section">
<h2></h2>
<button onClick={() => {
setEditingModel(null);
setShowModelForm(true);
}}>
</button>
<table>
<thead>
<tr>
<th>ID</th>
<th></th>
<th></th>
<th></th>
<th></th>
</tr>
</thead>
<tbody>
{models.map(m => {
const provider = providers.find(p => p.id === m.provider_id);
return (
<tr key={m.id}>
<td>{m.id}</td>
<td>{provider?.name || m.provider_id}</td>
<td>{m.model_name}</td>
<td>{m.enabled ? '启用' : '禁用'}</td>
<td>
<button onClick={() => {
setEditingModel(m);
setShowModelForm(true);
}}></button>
<button onClick={() => handleDeleteModel(m.id)}></button>
</td>
</tr>
);
})}
</tbody>
</table>
</div>
{/* 供应商表单 */}
{showProviderForm && (
<ProviderForm
provider={editingProvider || undefined}
onSave={() => {
setShowProviderForm(false);
setEditingProvider(null);
loadData();
}}
onCancel={() => {
setShowProviderForm(false);
setEditingProvider(null);
}}
/>
)}
{/* 模型表单 */}
{showModelForm && (
<ModelForm
model={editingModel || undefined}
providers={providers}
onSave={() => {
setShowModelForm(false);
setEditingModel(null);
loadData();
}}
onCancel={() => {
setShowModelForm(false);
setEditingModel(null);
}}
/>
)}
</div>
);
}

View File

@@ -0,0 +1,110 @@
import { useState, useEffect } from 'react';
import * as api from '../api/client';
export function StatsPage() {
const [stats, setStats] = useState<api.UsageStats[]>([]);
const [providers, setProviders] = useState<api.Provider[]>([]);
const [loading, setLoading] = useState(true);
const [error, setError] = useState<string | null>(null);
// 过滤条件
const [providerId, setProviderId] = useState('');
const [modelName, setModelName] = useState('');
const [startDate, setStartDate] = useState('');
const [endDate, setEndDate] = useState('');
useEffect(() => {
loadData();
}, []);
async function loadData() {
try {
setLoading(true);
const [statsData, providersData] = await Promise.all([
api.getStats({
provider_id: providerId || undefined,
model_name: modelName || undefined,
start_date: startDate || undefined,
end_date: endDate || undefined,
}),
api.listProviders(),
]);
setStats(statsData);
setProviders(providersData);
} catch (err) {
setError(err instanceof Error ? err.message : 'Unknown error');
} finally {
setLoading(false);
}
}
function handleFilter(e: React.FormEvent) {
e.preventDefault();
loadData();
}
if (loading) return <div className="loading">...</div>;
if (error) return <div className="error">{error}</div>;
return (
<div className="stats-page">
<h1></h1>
<form onSubmit={handleFilter} className="filter-form">
<select value={providerId} onChange={e => setProviderId(e.target.value)}>
<option value=""></option>
{providers.map(p => (
<option key={p.id} value={p.id}>{p.name}</option>
))}
</select>
<input
type="text"
placeholder="模型名称"
value={modelName}
onChange={e => setModelName(e.target.value)}
/>
<input
type="date"
placeholder="开始日期"
value={startDate}
onChange={e => setStartDate(e.target.value)}
/>
<input
type="date"
placeholder="结束日期"
value={endDate}
onChange={e => setEndDate(e.target.value)}
/>
<button type="submit"></button>
</form>
<table>
<thead>
<tr>
<th></th>
<th></th>
<th></th>
<th></th>
</tr>
</thead>
<tbody>
{stats.map(s => {
const provider = providers.find(p => p.id === s.provider_id);
return (
<tr key={s.id}>
<td>{provider?.name || s.provider_id}</td>
<td>{s.model_name}</td>
<td>{s.date}</td>
<td>{s.request_count}</td>
</tr>
);
})}
</tbody>
</table>
</div>
);
}