account system、custom prompt

2026-07-06 12:30:59 +08:00 · 2025-10-31 03:42:01 +08:00
parent bbe1e1f929
commit ceb2f7b435
32 changed files with 3873 additions and 465 deletions
--- a/web/src/lib/api.ts
+++ b/web/src/lib/api.ts
@@ -10,14 +10,31 @@ import type {
  CreateTraderRequest,
  UpdateModelConfigRequest,
  UpdateExchangeConfigRequest,
+  CompetitionData,
 } from '../types';

 const API_BASE = '/api';

+// Helper function to get auth headers
+function getAuthHeaders(): Record<string, string> {
+  const token = localStorage.getItem('auth_token');
+  const headers: Record<string, string> = {
+    'Content-Type': 'application/json',
+  };
+  
+  if (token) {
+    headers['Authorization'] = `Bearer ${token}`;
+  }
+  
+  return headers;
+}
+
 export const api = {
  // AI交易员管理接口
  async getTraders(): Promise<TraderInfo[]> {
-    const res = await fetch(`${API_BASE}/traders`);
+    const res = await fetch(`${API_BASE}/traders`, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取trader列表失败');
    return res.json();
  },
@@ -25,7 +42,7 @@ export const api = {
  async createTrader(request: CreateTraderRequest): Promise<TraderInfo> {
    const res = await fetch(`${API_BASE}/traders`, {
      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
+      headers: getAuthHeaders(),
      body: JSON.stringify(request),
    });
    if (!res.ok) throw new Error('创建交易员失败');
@@ -35,6 +52,7 @@ export const api = {
  async deleteTrader(traderId: string): Promise<void> {
    const res = await fetch(`${API_BASE}/traders/${traderId}`, {
      method: 'DELETE',
+      headers: getAuthHeaders(),
    });
    if (!res.ok) throw new Error('删除交易员失败');
  },
@@ -42,6 +60,7 @@ export const api = {
  async startTrader(traderId: string): Promise<void> {
    const res = await fetch(`${API_BASE}/traders/${traderId}/start`, {
      method: 'POST',
+      headers: getAuthHeaders(),
    });
    if (!res.ok) throw new Error('启动交易员失败');
  },
@@ -49,21 +68,40 @@ export const api = {
  async stopTrader(traderId: string): Promise<void> {
    const res = await fetch(`${API_BASE}/traders/${traderId}/stop`, {
      method: 'POST',
+      headers: getAuthHeaders(),
    });
    if (!res.ok) throw new Error('停止交易员失败');
  },

+  async updateTraderPrompt(traderId: string, customPrompt: string): Promise<void> {
+    const res = await fetch(`${API_BASE}/traders/${traderId}/prompt`, {
+      method: 'PUT',
+      headers: getAuthHeaders(),
+      body: JSON.stringify({ custom_prompt: customPrompt }),
+    });
+    if (!res.ok) throw new Error('更新自定义策略失败');
+  },
+
  // AI模型配置接口
  async getModelConfigs(): Promise<AIModel[]> {
-    const res = await fetch(`${API_BASE}/models`);
+    const res = await fetch(`${API_BASE}/models`, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取模型配置失败');
    return res.json();
  },

+  // 获取系统支持的AI模型列表（无需认证）
+  async getSupportedModels(): Promise<AIModel[]> {
+    const res = await fetch(`${API_BASE}/supported-models`);
+    if (!res.ok) throw new Error('获取支持的模型失败');
+    return res.json();
+  },
+
  async updateModelConfigs(request: UpdateModelConfigRequest): Promise<void> {
    const res = await fetch(`${API_BASE}/models`, {
      method: 'PUT',
-      headers: { 'Content-Type': 'application/json' },
+      headers: getAuthHeaders(),
      body: JSON.stringify(request),
    });
    if (!res.ok) throw new Error('更新模型配置失败');
@@ -71,15 +109,24 @@ export const api = {

  // 交易所配置接口
  async getExchangeConfigs(): Promise<Exchange[]> {
-    const res = await fetch(`${API_BASE}/exchanges`);
+    const res = await fetch(`${API_BASE}/exchanges`, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取交易所配置失败');
    return res.json();
  },

+  // 获取系统支持的交易所列表（无需认证）
+  async getSupportedExchanges(): Promise<Exchange[]> {
+    const res = await fetch(`${API_BASE}/supported-exchanges`);
+    if (!res.ok) throw new Error('获取支持的交易所失败');
+    return res.json();
+  },
+
  async updateExchangeConfigs(request: UpdateExchangeConfigRequest): Promise<void> {
    const res = await fetch(`${API_BASE}/exchanges`, {
      method: 'PUT',
-      headers: { 'Content-Type': 'application/json' },
+      headers: getAuthHeaders(),
      body: JSON.stringify(request),
    });
    if (!res.ok) throw new Error('更新交易所配置失败');
@@ -90,7 +137,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/status?trader_id=${traderId}`
      : `${API_BASE}/status`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取系统状态失败');
    return res.json();
  },
@@ -103,6 +152,7 @@ export const api = {
    const res = await fetch(url, {
      cache: 'no-store',
      headers: {
+        ...getAuthHeaders(),
        'Cache-Control': 'no-cache',
      },
    });
@@ -117,7 +167,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/positions?trader_id=${traderId}`
      : `${API_BASE}/positions`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取持仓列表失败');
    return res.json();
  },
@@ -127,7 +179,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/decisions?trader_id=${traderId}`
      : `${API_BASE}/decisions`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取决策日志失败');
    return res.json();
  },
@@ -137,7 +191,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/decisions/latest?trader_id=${traderId}`
      : `${API_BASE}/decisions/latest`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取最新决策失败');
    return res.json();
  },
@@ -147,7 +203,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/statistics?trader_id=${traderId}`
      : `${API_BASE}/statistics`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取统计信息失败');
    return res.json();
  },
@@ -157,7 +215,9 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/equity-history?trader_id=${traderId}`
      : `${API_BASE}/equity-history`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取历史数据失败');
    return res.json();
  },
@@ -167,8 +227,19 @@ export const api = {
    const url = traderId
      ? `${API_BASE}/performance?trader_id=${traderId}`
      : `${API_BASE}/performance`;
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      headers: getAuthHeaders(),
+    });
    if (!res.ok) throw new Error('获取AI学习数据失败');
    return res.json();
  },
+
+  // 获取竞赛数据
+  async getCompetition(): Promise<CompetitionData> {
+    const res = await fetch(`${API_BASE}/competition`, {
+      headers: getAuthHeaders(),
+    });
+    if (!res.ok) throw new Error('获取竞赛数据失败');
+    return res.json();
+  },
 };
--- a/web/src/lib/config.ts
+++ b/web/src/lib/config.ts
@@ -0,0 +1,27 @@
+export interface SystemConfig {
+  admin_mode: boolean;
+}
+
+let configPromise: Promise<SystemConfig> | null = null;
+let cachedConfig: SystemConfig | null = null;
+
+export function getSystemConfig(): Promise<SystemConfig> {
+  if (cachedConfig) {
+    return Promise.resolve(cachedConfig);
+  }
+  if (configPromise) {
+    return configPromise;
+  }
+  configPromise = fetch('/api/config')
+    .then((res) => res.json())
+    .then((data: SystemConfig) => {
+      cachedConfig = data;
+      return data;
+    })
+    .finally(() => {
+      // Keep cachedConfig for reuse; allow re-fetch via explicit invalidation if added later
+    });
+  return configPromise;
+}
+
+