agens-studio 0.1.0

package/config.js

@@ -0,0 +1,146 @@
+import path from 'node:path';
+import fs from 'node:fs';
+import os from 'node:os';
+import { fileURLToPath } from 'node:url';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+
+/**
+ * 外部配置文件路径
+ * - CLI 分发模式：~/.agens-cli/config.json（由首次运行的 setup wizard 创建）
+ * - 开发模式：同目录 config.local.json（手动创建，gitignore）
+ *
+ * 优先级：外部配置文件 > 环境变量 > 内置默认值
+ */
+function loadExternalConfig() {
+  // 1. ~/.agens-cli/config.json（分发模式）
+  const homeConfig = path.join(os.homedir(), '.agens-cli', 'config.json');
+  // 2. 项目本地 config.local.json（开发模式）
+  const localConfig = path.join(__dirname, 'config.local.json');
+
+  for (const cfgPath of [homeConfig, localConfig]) {
+    try {
+      const raw = fs.readFileSync(cfgPath, 'utf8');
+      const parsed = JSON.parse(raw);
+      if (parsed && typeof parsed === 'object' && Object.keys(parsed).length > 0) {
+        return { data: parsed, source: cfgPath };
+      }
+    } catch { /* 不存在或解析失败，尝试下一个 */ }
+  }
+  return { data: {}, source: null };
+}
+
+const ext = loadExternalConfig();
+const ext_cfg = ext.data;
+
+/**
+ * 全局配置
+ */
+export const config = {
+  // 配置文件来源（调试用）
+  _configSource: ext.source,
+
+  // 服务端口
+  port: process.env.PORT || 5173,
+
+  // agens API 接入信息
+  agens: {
+    // 生图（agnes-image-2.1-flash）
+    image: {
+      baseUrl: ext_cfg.agens_image_base_url || process.env.AGENS_IMAGE_BASE_URL || 'https://apihub.agnes-ai.com/v1',
+      apiKey: ext_cfg.agens_api_key || process.env.AGENS_IMAGE_API_KEY || '',
+      model: ext_cfg.agens_image_model || 'agnes-image-2.1-flash',
+      endpoint: '/images/generations',
+    },
+    // 视频（agnes-video-v2.0）
+    video: {
+      baseUrl: ext_cfg.agens_video_base_url || process.env.AGENS_VIDEO_BASE_URL || 'https://apihub.agnes-ai.com/v1',
+      apiKey: ext_cfg.agens_api_key || process.env.AGENS_VIDEO_API_KEY || '',
+      model: ext_cfg.agens_video_model || 'agnes-video-v2.0',
+      createEndpoint: '/videos',
+      queryBaseUrl: process.env.AGENS_VIDEO_QUERY_BASE_URL || 'https://apihub.agnes-ai.com',
+      queryPath: '/agnesapi',
+    },
+    // 视频轮询配置
+    pollInterval: 5000,
+    pollMaxAttempts: 120,
+  },
+
+  // mimo 大模型（用于提示词优化）
+  mimo: {
+    baseUrl: ext_cfg.mimo_base_url || process.env.MIMO_BASE_URL || 'https://token-plan-cn.xiaomimimo.com/v1',
+    apiKey: ext_cfg.mimo_api_key || '',
+    model: ext_cfg.mimo_model || 'mimo-v2.5-pro',
+    visionModel: ext_cfg.mimo_vision_model || 'mimo-v2.5',
+    endpoint: '/chat/completions',
+    maxTokens: 4096,
+    temperature: 0.8,
+    topP: 0.95,
+  },
+
+  // 识图结果缓存
+  visionCache: {
+    file: path.join(__dirname, 'data', 'vision_cache.json'),
+  },
+
+  // 提示词记忆库
+  promptMemory: {
+    file: path.join(__dirname, 'data', 'prompt_memory.json'),
+    maxSamples: 500,
+    feedSamples: 5,
+    minSamplesForPrefs: 8,
+  },
+
+  // 目录
+  dirs: {
+    root: __dirname,
+    public: path.join(__dirname, 'public'),
+    assets: path.join(__dirname, 'assets'),
+    images: path.join(__dirname, 'assets', 'images'),
+    videos: path.join(__dirname, 'assets', 'videos'),
+    data: path.join(__dirname, 'data'),
+    indexFile: path.join(__dirname, 'data', 'index.json'),
+    uploads: path.join(__dirname, 'data', 'uploads'),
+  },
+
+  // 上传参考图限制
+  upload: {
+    maxBytes: 20 * 1024 * 1024,
+    allowedMime: ['image/png', 'image/jpeg', 'image/webp', 'image/jpg'],
+  },
+};
+
+/**
+ * 用户配置目录（~/.agens-cli/）
+ * 分发模式下 API Key 和配置存在这里
+ */
+export const USER_CONFIG_DIR = path.join(os.homedir(), '.agens-cli');
+export const USER_CONFIG_FILE = path.join(USER_CONFIG_DIR, 'config.json');
+
+/**
+ * 保存用户配置到 ~/.agens-cli/config.json
+ * @param {object} cfg 配置对象
+ */
+export function saveUserConfig(cfg) {
+  fs.mkdirSync(USER_CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(USER_CONFIG_FILE, JSON.stringify(cfg, null, 2), 'utf8');
+}
+
+/**
+ * 读取用户配置（如果存在）
+ * @returns {object|null}
+ */
+export function readUserConfig() {
+  try {
+    return JSON.parse(fs.readFileSync(USER_CONFIG_FILE, 'utf8'));
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 检查 API Key 是否已配置（非默认空值）
+ */
+export function isApiConfigured() {
+  return !!(config.agens.image.apiKey || config.mimo.apiKey);
+}

package/package.json

@@ -0,0 +1,35 @@
+{
+  "name": "agens-studio",
+  "version": "0.1.0",
+  "description": "Agens 创作工作台 —— AI 图片/视频生成 CLI + Web 工作台",
+  "type": "module",
+  "main": "server.js",
+  "bin": {
+    "agens-cli": "./cli.js"
+  },
+  "files": [
+    "cli.js",
+    "config.js",
+    "src/",
+    "README-CLI.md"
+  ],
+  "scripts": {
+    "start": "node server.js",
+    "dev": "node --watch server.js",
+    "cli": "node cli.js"
+  },
+  "keywords": [
+    "agens",
+    "ai",
+    "image-generation",
+    "video-generation",
+    "cli"
+  ],
+  "license": "MIT",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "express": "^5.2.1"
+  }
+}

package/server.js

@@ -0,0 +1,74 @@
+import express from 'express';
+import path from 'node:path';
+import { config } from './config.js';
+import generateRouter from './src/routes/generate.js';
+import taskRouter from './src/routes/task.js';
+import assetsRouter from './src/routes/assets.js';
+import promptRouter from './src/routes/prompt.js';
+import { init as initAssetStore } from './src/services/assetStore.js';
+import { init as initPromptMemory } from './src/services/promptMemory.js';
+import { init as initVisionCache } from './src/services/mimoVision.js';
+import { isMockMode } from './src/services/agensClient.js';
+
+const app = express();
+
+// 中间件
+app.use(express.json({ limit: '2mb' }));
+
+// 开发期禁用静态文件缓存，保证改前端即时生效
+function noCache(req, res, next) {
+  res.set('Cache-Control', 'no-store, no-cache, must-revalidate, proxy-revalidate');
+  res.set('Pragma', 'no-cache');
+  res.set('Expires', '0');
+  next();
+}
+
+// 静态目录：前端 + 生成物 + 上传
+app.use(noCache, express.static(config.dirs.public));
+app.use('/assets', express.static(config.dirs.assets));
+app.use('/uploads', express.static(config.dirs.uploads));
+
+// 路由
+app.use('/api/generate', generateRouter);
+app.use('/api/task', taskRouter);
+app.use('/api/assets', assetsRouter);
+app.use('/api/prompt', promptRouter);
+
+// 健康检查 / 配置信息（前端用来判断是否 mock 模式）
+app.get('/api/health', (req, res) => {
+  res.json({
+    ok: true,
+    mockMode: isMockMode(),
+    agensConfigured: !isMockMode(),
+  });
+});
+
+// 兜底：所有非 /api、非静态文件的请求都回 index.html（便于前端单页逻辑）
+// 注：Express 5 不再支持 '*' 通配符，需使用命名捕获参数
+app.get('{*path}', (req, res) => {
+  res.sendFile(path.join(config.dirs.public, 'index.html'));
+});
+
+async function main() {
+  await initAssetStore();
+  await initPromptMemory();
+  await initVisionCache();
+
+  app.listen(config.port, () => {
+    console.log('\n  ┌──────────────────────────────────────────────┐');
+    console.log('  │   agens 本地创作工作台已启动                   │');
+    console.log(`  │   地址: http://localhost:${config.port}              │`);
+    if (isMockMode()) {
+      console.log('  │   ⚠️  Mock 模式（未配置 agens API，使用占位结果） │');
+      console.log('  │      配好 config.js 里的 baseUrl / apiKey 后生效 │');
+    } else {
+      console.log('  │   ✅ 已接入 agens API                          │');
+    }
+    console.log('  └──────────────────────────────────────────────┘\n');
+  });
+}
+
+main().catch((err) => {
+  console.error('启动失败：', err);
+  process.exit(1);
+});

package/src/routes/assets.js

@@ -0,0 +1,146 @@
+import { Router } from 'express';
+import fs from 'node:fs/promises';
+import path from 'node:path';
+import { randomUUID } from 'node:crypto';
+import { config } from '../../config.js';
+import * as store from '../services/assetStore.js';
+
+const router = Router();
+
+/** GET /api/assets —— 列资产 */
+router.get('/', async (req, res) => {
+  const result = await store.listAssets({
+    type: req.query.type || 'all',
+    search: req.query.search || '',
+    keptOnly: req.query.keptOnly === '1' || req.query.keptOnly === 'true',
+    page: parseInt(req.query.page, 10) || 1,
+    pageSize: parseInt(req.query.pageSize, 10) || 60,
+  });
+  res.json(result);
+});
+
+/** GET /api/assets/:id —— 单个资产详情 */
+router.get('/:id', async (req, res) => {
+  const item = await store.getAsset(req.params.id);
+  if (!item) return res.status(404).json({ error: '资产不存在' });
+  res.json(item);
+});
+
+/** PUT /api/assets/:id —— 更新（kept 保留标记 / note 备注） */
+router.put('/:id', async (req, res) => {
+  const { kept, note } = req.body || {};
+  const patch = {};
+  if (typeof kept === 'boolean') patch.kept = kept;
+  if (typeof note === 'string') patch.note = note;
+  const item = await store.updateAsset(req.params.id, patch);
+  if (!item) return res.status(404).json({ error: '资产不存在' });
+  res.json(item);
+});
+
+/** DELETE /api/assets/:id —— 删除（文件 + 索引） */
+router.delete('/:id', async (req, res) => {
+  const ok = await store.deleteAsset(req.params.id);
+  if (!ok) return res.status(404).json({ error: '资产不存在' });
+  res.json({ ok: true });
+});
+
+/**
+ * POST /api/upload —— 上传参考图（用于图生/多图/关键帧）
+ * multipart/form-data, 字段名 file
+ * 返回 { url } —— 供 /api/generate 的 imageUrls 使用
+ *
+ * 这里手写一个最简的 multipart 解析，避免引入 multer。
+ * 仅支持单文件、非流式，本地单机够用。
+ */
+router.post('/upload', async (req, res) => {
+  try {
+    const contentType = req.headers['content-type'] || '';
+    if (!contentType.startsWith('multipart/form-data')) {
+      return res.status(400).json({ error: '需要 multipart/form-data' });
+    }
+
+    const boundary = extractBoundary(contentType);
+    if (!boundary) return res.status(400).json({ error: '无法解析 boundary' });
+
+    const chunks = [];
+    for await (const c of req) chunks.push(c);
+    const buf = Buffer.concat(chunks);
+    if (buf.length > config.upload.maxBytes) {
+      return res.status(413).json({ error: '文件过大（>20MB）' });
+    }
+
+    const parsed = parseMultipart(buf, boundary);
+    const file = parsed.find((p) => p.filename);
+    if (!file) return res.status(400).json({ error: '未找到文件' });
+
+    if (!config.upload.allowedMime.includes(file.contentType)) {
+      return res.status(415).json({ error: '仅支持 png/jpg/webp' });
+    }
+
+    const ext = guessExt(file.filename, file.contentType);
+    const id = randomUUID();
+    const filename = `${id}${ext}`;
+    const absPath = path.join(config.dirs.uploads, filename);
+    await fs.writeFile(absPath, file.data);
+
+    // 浏览器可访问的路径
+    const url = `/uploads/${filename}`;
+    res.json({ url, filename, size: file.data.length });
+  } catch (e) {
+    res.status(500).json({ error: '上传失败：' + e.message });
+  }
+});
+
+function extractBoundary(ct) {
+  const m = ct.match(/boundary=(?:"([^"]+)"|([^;]+))/i);
+  return m ? (m[1] || m[2]) : null;
+}
+
+function parseMultipart(buf, boundary) {
+  const sep = Buffer.from('--' + boundary);
+  const parts = [];
+  let start = 0;
+  while (true) {
+    const i = buf.indexOf(sep, start);
+    if (i === -1) break;
+    const next = buf.indexOf(sep, i + sep.length);
+    if (next === -1) break;
+    const block = buf.slice(i + sep.length, next);
+    parts.push(block);
+    start = next;
+  }
+  const result = [];
+  for (const block of parts) {
+    // 跳过结尾的 --
+    if (block.slice(0, 2).toString() === '--') continue;
+    const headerEnd = block.indexOf('\r\n\r\n');
+    if (headerEnd === -1) continue;
+    const headerText = block.slice(0, headerEnd).toString('utf8');
+    let body = block.slice(headerEnd + 4);
+    // 去掉末尾的 \r\n
+    if (body.length >= 2 && body[body.length - 2] === 0x0d && body[body.length - 1] === 0x0a) {
+      body = body.slice(0, -2);
+    }
+    const nameMatch = headerText.match(/name="([^"]+)"/);
+    const fileMatch = headerText.match(/filename="([^"]*)"/);
+    const ctMatch = headerText.match(/Content-Type:\s*([^\r\n]+)/i);
+    result.push({
+      name: nameMatch ? nameMatch[1] : null,
+      filename: fileMatch ? fileMatch[1] : null,
+      contentType: ctMatch ? ctMatch[1].trim() : 'application/octet-stream',
+      data: body,
+    });
+  }
+  return result;
+}
+
+function guessExt(filename, mime) {
+  if (filename) {
+    const m = filename.toLowerCase().match(/\.(png|jpe?g|webp)$/);
+    if (m) return '.' + m[1].replace('jpeg', 'jpg');
+  }
+  const map = { 'image/png': '.png', 'image/jpeg': '.jpg', 'image/jpg': '.jpg', 'image/webp': '.webp' };
+  return map[mime?.toLowerCase()] || '.png';
+}
+
+export default router;

package/src/routes/generate.js

@@ -0,0 +1,64 @@
+import { Router } from 'express';
+import { submitGenerateJob } from '../services/taskManager.js';
+
+const router = Router();
+
+const IMAGE_MODES = ['image', 'img2img'];
+const VIDEO_MODES = ['text2video', 'image2video', 'multi2video', 'keyframe'];
+
+/**
+ * POST /api/generate
+ * body: {
+ *   type: 'image' | 'video',
+ *   mode:  生图 'image' | 'img2img'
+ *          视频 'text2video' | 'image2video' | 'multi2video' | 'keyframe',
+ *   prompt: string,
+ *   params: object,         // 比例、风格、时长等
+ *   imageUrls: string[],    // 图生/多图/关键帧需要的参考图（已上传后的本地路径）
+ * }
+ * 立即返回本地 taskId，前端轮询 /api/task/:id
+ */
+router.post('/', async (req, res) => {
+  const { type, mode, prompt, params, imageUrls } = req.body || {};
+  if (!type || !['image', 'video'].includes(type)) {
+    return res.status(400).json({ error: 'type 必须是 image 或 video' });
+  }
+  if (!mode) {
+    return res.status(400).json({ error: '请指定 mode' });
+  }
+
+  // 模式合法性校验
+  if (type === 'image' && !IMAGE_MODES.includes(mode)) {
+    return res.status(400).json({ error: '未知的生图模式：' + mode });
+  }
+  if (type === 'video' && !VIDEO_MODES.includes(mode)) {
+    return res.status(400).json({ error: '未知的视频模式：' + mode });
+  }
+
+  // 参考图校验
+  const hasImages = Array.isArray(imageUrls) && imageUrls.length > 0;
+  if (mode === 'img2img' && !hasImages) {
+    return res.status(400).json({ error: '图生图需要上传一张原图' });
+  }
+  if (mode === 'image2video' && !hasImages) {
+    return res.status(400).json({ error: '该模式需要上传至少一张参考图' });
+  }
+  if (mode === 'multi2video' && (!hasImages || imageUrls.length < 2)) {
+    return res.status(400).json({ error: '多图生视频至少需要 2 张参考图' });
+  }
+  if (mode === 'keyframe' && (!hasImages || imageUrls.length < 2)) {
+    return res.status(400).json({ error: '关键帧动画至少需要首帧和尾帧（共 2 张）' });
+  }
+
+  const task = await submitGenerateJob({
+    type,
+    mode,
+    prompt: prompt || '',
+    params: params || {},
+    imageUrls: imageUrls || [],
+  });
+
+  res.json({ taskId: task.id });
+});
+
+export default router;