Hugging Face's logo

Spaces:
dedw231
/
Antigravity2api
Running

App Files Community
Antigravity2api
File size: 3,564 Bytes 
b88ce1b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
import fs from 'fs/promises';
import path from 'path';
import logger from '../utils/logger.js';
import { reloadConfig } from '../config/config.js';

const CONFIG_FILE = path.join(process.cwd(), 'config.json');

// 加载设置
export async function loadSettings() {
  try {
    const data = await fs.readFile(CONFIG_FILE, 'utf-8');
    return JSON.parse(data);
  } catch (error) {
    logger.error('读取配置文件失败:', error);
    // 返回默认配置
    return {
      server: { port: 8045, host: '0.0.0.0' },
      api: {
        url: 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse',
        modelsUrl: 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels',
        host: 'daily-cloudcode-pa.sandbox.googleapis.com',
        userAgent: 'antigravity/1.11.3 windows/amd64'
      },
      defaults: { temperature: 1, top_p: 0.85, top_k: 50, max_tokens: 8096 },
      security: { maxRequestSize: '50mb', apiKey: 'sk-text', adminPassword: 'admin123' },
      systemInstruction: '你是聊天机器人，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演，也可以提供数学或者代码上的建议'
    };
  }
}

// 保存设置
export async function saveSettings(newSettings) {
  try {
    // 读取现有配置
    let config;
    try {
      const data = await fs.readFile(CONFIG_FILE, 'utf-8');
      config = JSON.parse(data);
    } catch {
      config = {};
    }

    // 合并设置
    config.server = config.server || {};
    config.security = config.security || {};
    config.defaults = config.defaults || {};

    // 更新服务器配置
    if (newSettings.server) {
      config.server.port = parseInt(newSettings.server.port) || config.server.port;
      config.server.host = newSettings.server.host || config.server.host;
    }

    // 更新安全配置
    if (newSettings.security) {
      config.security.apiKey = newSettings.security.apiKey || config.security.apiKey;
      config.security.adminPassword = newSettings.security.adminPassword || config.security.adminPassword;
      config.security.maxRequestSize = newSettings.security.maxRequestSize || config.security.maxRequestSize;
    }

    // 更新默认参数
    if (newSettings.defaults) {
      const temp = parseFloat(newSettings.defaults.temperature);
      if (!isNaN(temp)) config.defaults.temperature = temp;

      const topP = parseFloat(newSettings.defaults.top_p);
      if (!isNaN(topP)) config.defaults.top_p = topP;

      const topK = parseInt(newSettings.defaults.top_k);
      if (!isNaN(topK)) config.defaults.top_k = topK;

      const maxTokens = parseInt(newSettings.defaults.max_tokens);
      if (!isNaN(maxTokens)) config.defaults.max_tokens = maxTokens;
    }

    // 更新系统指令
    if (newSettings.systemInstruction !== undefined) {
      config.systemInstruction = newSettings.systemInstruction;
    }

    // 写入文件
    logger.info('Saving config:', JSON.stringify(config, null, 2));
    await fs.writeFile(CONFIG_FILE, JSON.stringify(config, null, 2), 'utf-8');
    logger.info('配置文件已保存');

    // 触发热重载
    const reloaded = reloadConfig();
    const message = reloaded
      ? '设置已保存并生效（端口等核心配置需重启）'
      : '设置已保存，但热重载失败，请重启服务器';

    return { success: true, message };
  } catch (error) {
    logger.error('保存配置文件失败:', error);
    throw new Error('保存配置失败: ' + error.message);
  }
}