ugc-lens 0.0.1 → 0.0.2

package/README.md

@@ -15,7 +15,15 @@ npm install -g ugc-lens
 
 ## 配置模型
 
-在运行命令的目录里放一个 `.env`（或设置同名环境变量），选一个提供商并填它的 API key：
+最简单的方式：
+
+```
+ugc-lens init
+```
+
+交互式地选提供商、从 `/models` 列表里选模型、填 API key 并发一次测试调用，写入全局配置 `~/.ugc-lens/config.json`（权限 600），所有目录下都能用。
+
+也可以手动配置：在运行命令的目录里放一个 `.env`（或设置同名环境变量），选一个提供商并填它的 API key：
 
 ```
 UGC_LENS_PROVIDER=deepseek
@@ -23,11 +31,13 @@ UGC_LENS_MODEL=deepseek-flash
 DEEPSEEK_API_KEY=你的 key
 ```
 
-支持 DeepSeek、通义千问、Kimi、智谱、MiniMax、OpenAI、Claude、Gemini、Grok、OpenRouter 和本地模型，变量名和差异见 [模型提供商](docs/providers.md)。
+支持 DeepSeek、通义千问、Kimi、智谱、MiniMax、OpenAI、Claude、Gemini、Grok、OpenRouter 和本地模型，变量名和差异见 [模型提供商](docs/providers.md)。优先级：环境变量 > 当前目录的 `.env` / `.ugc-lens/config.json` > 全局配置。
 
 ## 使用
 
 ```
+ugc-lens init                                     # 初始化模型配置
+
 # 分析一个文件：报告打印到标准输出，同时保存到 .ugc-lens/reports/
 ugc-lens analyze bilibili_all.json > report.md
 ugc-lens analyze bilibili_all.json --focus 价格,售后     # 只看这些方向

package/dist/cli/index.js

@@ -9,8 +9,10 @@ import { ImportError } from '../ingest/index.js';
 import { ANALYZE_USAGE, runAnalyze } from './analyze.js';
 import { ASK_USAGE, runAsk } from './ask.js';
 import { UsageError } from './common.js';
+import { INIT_USAGE, runInit } from './init.js';
 import { runWatch, WATCH_USAGE } from './watch.js';
 const COMMANDS = {
+    init: { run: runInit, usage: INIT_USAGE },
     analyze: { run: runAnalyze, usage: ANALYZE_USAGE },
     ask: { run: runAsk, usage: ASK_USAGE },
     watch: { run: runWatch, usage: WATCH_USAGE },
@@ -18,6 +20,7 @@ const COMMANDS = {
 const MAIN_USAGE = `ugc-lens：分析评论、弹幕等用户互动内容
 
 用法：
+  ugc-lens init                         初始化：选择提供商和模型，写入全局配置
   ugc-lens analyze <文件>               生成 Markdown 分析报告
   ugc-lens ask <文件> <问题> [追问...]   追问，回答带 [ID] 引用
   ugc-lens watch <目录>                 监听目录，新的 *_all.json 写完后自动分析

package/dist/cli/init.d.ts

@@ -0,0 +1,23 @@
+import { type LlmConfigFile } from '../config.js';
+export declare const INIT_USAGE = "\u7528\u6CD5\uFF1Augc-lens init\n\n\u4EA4\u4E92\u5F0F\u521D\u59CB\u5316\uFF1A\u9009\u62E9\u63D0\u4F9B\u5546\u548C\u6A21\u578B\u3001\u586B API key\u3001\u53D1\u4E00\u6B21\u6D4B\u8BD5\u8C03\u7528\uFF0C\n\u5199\u5165\u5168\u5C40\u914D\u7F6E ~/.ugc-lens/config.json\uFF08\u6743\u9650 600\uFF09\uFF0C\u6240\u6709\u76EE\u5F55\u4E0B\u751F\u6548\u3002\n\u9879\u76EE\u76EE\u5F55\u7684 .env \u6216 .ugc-lens/config.json \u4ECD\u7136\u53EF\u4EE5\u8986\u76D6\u5B83\u3002";
+/** 交互的输入输出，测试时换成假的 */
+export interface InitIo {
+    ask(question: string): Promise<string>;
+    print(text: string): void;
+}
+export interface InitDeps {
+    io: InitIo;
+    homeDir: string;
+    /** 同时读取这个目录的 .env，默认当前目录：init 也能发现项目里已有的 key */
+    cwd?: string;
+    /** 列出提供商的模型；默认真的请求 GET {baseUrl}/models */
+    fetchModels?: (config: {
+        baseUrl: string;
+        apiKey?: string;
+    }) => Promise<string[]>;
+    /** 发一次测试调用，返回错误消息；undefined 表示通过。默认用真实客户端 */
+    verifyModel?: (config: LlmConfigFile) => Promise<string | undefined>;
+}
+/** 交互式初始化流程；返回写入的配置，用户取消时返回 undefined */
+export declare function runInitFlow(deps: InitDeps): Promise<LlmConfigFile | undefined>;
+export declare function runInit(): Promise<void>;

package/dist/cli/init.js

@@ -0,0 +1,185 @@
+import { createInterface } from 'node:readline/promises';
+import { loadDotEnv, loadLlmConfig, readGlobalConfig, writeGlobalConfig } from '../config.js';
+import { isProviderId, PROVIDERS, resolveBaseUrl } from '../llm/providers.js';
+import { createChatClient } from '../llm/client.js';
+import { createLlmModel } from '../agent/llm_model.js';
+import { UsageError } from './common.js';
+export const INIT_USAGE = `用法：ugc-lens init
+
+交互式初始化：选择提供商和模型、填 API key、发一次测试调用，
+写入全局配置 ~/.ugc-lens/config.json（权限 600），所有目录下生效。
+项目目录的 .env 或 .ugc-lens/config.json 仍然可以覆盖它。`;
+const PROVIDER_IDS = [...Object.keys(PROVIDERS), 'custom'];
+/** GET {baseUrl}/models，10 秒超时；返回模型 id 列表 */
+async function fetchModelsReal({ baseUrl, apiKey }) {
+    const response = await fetch(`${baseUrl}/models`, {
+        headers: apiKey ? { authorization: `Bearer ${apiKey}` } : {},
+        signal: AbortSignal.timeout(10_000),
+    });
+    if (!response.ok)
+        throw new Error(`HTTP ${response.status}`);
+    const body = (await response.json());
+    const ids = (body.data ?? []).map((item) => item.id).filter((id) => typeof id === 'string');
+    if (ids.length === 0)
+        throw new Error('返回的列表是空的');
+    return ids;
+}
+/** 一次最小的调用，并要求调用一个假工具，顺便验证 function call 支持 */
+async function verifyModelReal(file) {
+    try {
+        const config = loadLlmConfig({ ...file, timeout_ms: 30_000, max_retries: 0 });
+        if (!config.model)
+            return '没有选模型';
+        const model = createLlmModel(createChatClient(config), { temperature: 0, maxTokens: 30, disableThinking: true });
+        const result = await model.complete({
+            messages: [{ role: 'user', content: '现在几点？用工具回答。' }],
+            tools: [
+                {
+                    type: 'function',
+                    function: {
+                        name: 'get_current_time',
+                        description: '获取当前时间',
+                        parameters: { type: 'object', properties: {}, additionalProperties: false },
+                    },
+                },
+            ],
+            toolChoice: { type: 'function', function: { name: 'get_current_time' } },
+            signal: AbortSignal.timeout(35_000),
+        });
+        if (!result.message.tool_calls?.length)
+            return '模型返回了文字而不是工具调用，这个模型可能不支持 function call';
+        return undefined;
+    }
+    catch (error) {
+        return error.message;
+    }
+}
+function pickProviderId(input) {
+    const value = input.trim().toLowerCase();
+    const byNumber = PROVIDER_IDS[Number(value) - 1];
+    if (byNumber)
+        return byNumber;
+    return isProviderId(value) ? value : undefined;
+}
+/** 交互式初始化流程；返回写入的配置，用户取消时返回 undefined */
+export async function runInitFlow(deps) {
+    const { io } = deps;
+    loadDotEnv(deps.cwd ?? process.cwd());
+    const existing = readGlobalConfig(deps.homeDir);
+    const fetchModels = deps.fetchModels ?? fetchModelsReal;
+    const verifyModel = deps.verifyModel ?? verifyModelReal;
+    if (existing.provider) {
+        io.print(`当前全局配置：${existing.provider}${existing.model ? ` / ${existing.model}` : ''}${existing.api_key ? '（已保存 key）' : ''}`);
+        if ((await io.ask('重新设置？[y/N] ')).toLowerCase() !== 'y')
+            return undefined;
+    }
+    io.print('\n选择模型提供商：');
+    PROVIDER_IDS.forEach((id, i) => {
+        io.print(`  ${i + 1}. ${id === 'custom' ? '自定义（任意 OpenAI 兼容接口）' : PROVIDERS[id].name}（${id}）`);
+    });
+    let provider;
+    while (!provider) {
+        provider = pickProviderId(await io.ask(`输入编号或 id${existing.provider ? `（回车保持 ${existing.provider}）` : ''}：`));
+        if (!provider && existing.provider)
+            provider = existing.provider;
+        if (!provider)
+            io.print('没认出来，再试一次。');
+    }
+    const preset = provider === 'custom' ? undefined : PROVIDERS[provider];
+    const file = { provider };
+    let region = 'cn';
+    if (preset && typeof preset.baseUrl !== 'string') {
+        const answer = await io.ask(`国内还是国际接口？[1] 国内 [2] 国际（回车默认国内）：`);
+        region = answer.trim() === '2' ? 'global' : 'cn';
+        if (region !== 'cn')
+            file.region = region;
+    }
+    if (provider === 'custom') {
+        const baseUrl = (await io.ask('接口地址（如 http://localhost:8000/v1）：')).trim();
+        if (!baseUrl)
+            throw new UsageError('自定义提供商必须填接口地址');
+        file.base_url = baseUrl;
+    }
+    const keyEnv = preset?.apiKeyEnv;
+    const needsKey = Boolean(keyEnv) && provider !== 'lmstudio';
+    if (provider !== 'ollama') {
+        const envKey = keyEnv ? process.env[keyEnv] : undefined;
+        const hints = [
+            envKey ? `环境变量 ${keyEnv} 里已经有，回车用它` : '',
+            existing.api_key ? '回车保持已保存的 key' : '',
+            needsKey ? '' : '没有就回车跳过',
+        ].filter(Boolean);
+        const key = (await io.ask(`API key${hints.length > 0 ? `（${hints.join('；')}）` : ''}：`)).trim();
+        if (key)
+            file.api_key = key;
+        else if (existing.api_key)
+            file.api_key = existing.api_key;
+        if (!file.api_key && !envKey && needsKey) {
+            io.print(`注意：没有 key，运行前还需要设置 ${keyEnv}。`);
+        }
+    }
+    const baseUrl = file.base_url ?? (preset ? resolveBaseUrl(preset, region) : '');
+    const effectiveKey = file.api_key ?? (keyEnv ? process.env[keyEnv] : undefined);
+    try {
+        io.print('正在获取模型列表……');
+        const models = await fetchModels({ baseUrl, apiKey: effectiveKey });
+        const shown = models.slice(0, 30);
+        shown.forEach((id, i) => io.print(`  ${i + 1}. ${id}`));
+        if (models.length > shown.length)
+            io.print(`  ……共 ${models.length} 个，也可以直接输入名字`);
+        while (!file.model) {
+            const answer = (await io.ask(`选择模型，输入编号或名字${existing.model ? `（回车保持 ${existing.model}）` : ''}：`)).trim();
+            if (!answer && existing.model)
+                file.model = existing.model;
+            else if (models[Number(answer) - 1])
+                file.model = models[Number(answer) - 1];
+            else if (answer)
+                file.model = answer;
+            else
+                io.print('需要一个模型。');
+        }
+    }
+    catch (error) {
+        io.print(`获取模型列表失败：${error.message}，手动输入。`);
+        while (!file.model) {
+            const answer = (await io.ask(`模型名字${existing.model ? `（回车保持 ${existing.model}）` : ''}：`)).trim();
+            if (answer)
+                file.model = answer;
+            else if (existing.model)
+                file.model = existing.model;
+            else
+                io.print('需要一个模型。');
+        }
+    }
+    const verify = (await io.ask('发一次测试调用验证模型和 key？[Y/n] ')).trim().toLowerCase();
+    if (verify !== 'n') {
+        const problem = await verifyModel({ ...existing, ...file });
+        if (problem) {
+            io.print(`测试调用失败：${problem}`);
+            const save = (await io.ask('仍然保存配置？[y/N] ')).trim().toLowerCase();
+            if (save !== 'y')
+                return undefined;
+        }
+        else {
+            io.print('测试调用通过。');
+        }
+    }
+    const path = writeGlobalConfig(file, deps.homeDir);
+    io.print(`已写入 ${path}（权限 600）。某个项目要用别的配置时，在该目录放 .env 或 .ugc-lens/config.json 即可覆盖。`);
+    return file;
+}
+export async function runInit() {
+    if (!process.stdin.isTTY)
+        throw new UsageError('init 需要在终端里交互运行');
+    // 整个流程共用一个接口：每问新建会把用户一次粘贴的多行回答丢掉
+    const rl = createInterface({ input: process.stdin, output: process.stdout });
+    try {
+        await runInitFlow({
+            io: { ask: (question) => rl.question(question), print: (text) => console.log(text) },
+            homeDir: process.env.HOME ?? process.cwd(),
+        });
+    }
+    finally {
+        rl.close();
+    }
+}

package/dist/config.d.ts

@@ -6,6 +6,8 @@ export interface LlmConfigFile {
     region?: Region;
     base_url?: string | null;
     api_key_env?: string | null;
+    /** 直接写 key，只建议放在全局配置 ~/.ugc-lens/config.json 里 */
+    api_key?: string | null;
     timeout_ms?: number;
     max_retries?: number;
 }
@@ -22,7 +24,14 @@ export interface LlmConfig {
 export declare class ConfigError extends Error {
     name: string;
 }
+/** 读取 cwd/.env 进环境变量（不覆盖已有的），没有就什么都不做 */
+export declare function loadDotEnv(cwd: string): void;
+export declare function globalConfigPath(homeDir?: string): string;
+/** 读取全局配置（~/.ugc-lens/config.json）里的 llm 字段；没有时返回空对象 */
+export declare function readGlobalConfig(homeDir?: string): LlmConfigFile;
+/** 写入全局配置：替换 llm 字段，其他顶层字段保留；权限 600，因为里面有 key */
+export declare function writeGlobalConfig(llm: LlmConfigFile, homeDir?: string): string;
 /**
- * 读取模型配置。优先级：参数 overrides > 环境变量 > 配置文件 > 提供商预设。
+ * 读取模型配置。每一项的优先级：参数 overrides > 环境变量 > 项目配置（cwd/.ugc-lens）> 全局配置（~/.ugc-lens）> 提供商预设。
  */
-export declare function loadLlmConfig(overrides?: LlmConfigFile, cwd?: string): LlmConfig;
+export declare function loadLlmConfig(overrides?: LlmConfigFile, cwd?: string, homeDir?: string): LlmConfig;

package/dist/config.js

@@ -1,11 +1,13 @@
-import { existsSync, readFileSync } from 'node:fs';
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
+import { homedir } from 'node:os';
 import { join } from 'node:path';
 import { PROVIDERS, isProviderId, resolveBaseUrl, } from './llm/providers.js';
 export class ConfigError extends Error {
     name = 'ConfigError';
 }
 const CONFIG_DIR = '.ugc-lens';
-function loadDotEnv(cwd) {
+/** 读取 cwd/.env 进环境变量（不覆盖已有的），没有就什么都不做 */
+export function loadDotEnv(cwd) {
     const file = join(cwd, '.env');
     if (!existsSync(file))
         return;
@@ -15,6 +17,33 @@ function loadDotEnv(cwd) {
     // 已存在的环境变量不会被 .env 覆盖，保证命令行里临时指定的值优先
     process.loadEnvFile(file);
 }
+export function globalConfigPath(homeDir = homedir()) {
+    return join(homeDir, CONFIG_DIR, 'config.json');
+}
+/** 读取全局配置（~/.ugc-lens/config.json）里的 llm 字段；没有时返回空对象 */
+export function readGlobalConfig(homeDir = homedir()) {
+    return readConfigFile(homeDir);
+}
+/** 写入全局配置：替换 llm 字段，其他顶层字段保留；权限 600，因为里面有 key */
+export function writeGlobalConfig(llm, homeDir = homedir()) {
+    const path = globalConfigPath(homeDir);
+    let rest = {};
+    if (existsSync(path)) {
+        try {
+            const parsed = JSON.parse(readFileSync(path, 'utf8'));
+            if (parsed && typeof parsed === 'object') {
+                const { llm: _old, ...others } = parsed;
+                rest = others;
+            }
+        }
+        catch (error) {
+            throw new ConfigError(`无法解析 ${path}: ${error.message}`);
+        }
+    }
+    mkdirSync(join(homeDir, CONFIG_DIR), { recursive: true });
+    writeFileSync(path, `${JSON.stringify({ ...rest, llm }, null, 2)}\n`, { mode: 0o600 });
+    return path;
+}
 function readConfigFile(cwd) {
     const file = join(cwd, CONFIG_DIR, 'config.json');
     if (!existsSync(file))
@@ -36,42 +65,43 @@ function pick(...values) {
     return undefined;
 }
 /**
- * 读取模型配置。优先级：参数 overrides > 环境变量 > 配置文件 > 提供商预设。
+ * 读取模型配置。每一项的优先级：参数 overrides > 环境变量 > 项目配置（cwd/.ugc-lens）> 全局配置（~/.ugc-lens）> 提供商预设。
  */
-export function loadLlmConfig(overrides = {}, cwd = process.cwd()) {
+export function loadLlmConfig(overrides = {}, cwd = process.cwd(), homeDir = homedir()) {
     loadDotEnv(cwd);
     const file = readConfigFile(cwd);
+    const global = existsSync(globalConfigPath(homeDir)) ? readConfigFile(homeDir) : {};
     const env = process.env;
-    const provider = pick(overrides.provider, env.UGC_LENS_PROVIDER, file.provider);
+    const provider = pick(overrides.provider, env.UGC_LENS_PROVIDER, file.provider, global.provider);
     if (!provider) {
-        throw new ConfigError('未选择模型提供商：设置 UGC_LENS_PROVIDER 或配置文件中的 llm.provider');
+        throw new ConfigError('未选择模型提供商：设置 UGC_LENS_PROVIDER，或运行 ugc-lens init 写入全局配置');
     }
     if (!isProviderId(provider)) {
         const ids = [...Object.keys(PROVIDERS), 'custom'].join(', ');
         throw new ConfigError(`未知的提供商 "${provider}"，可选：${ids}`);
     }
-    const region = (pick(overrides.region, env.UGC_LENS_REGION, file.region) ?? 'cn');
+    const region = (pick(overrides.region, env.UGC_LENS_REGION, file.region, global.region) ?? 'cn');
     if (region !== 'cn' && region !== 'global') {
         throw new ConfigError(`region 只能是 cn 或 global，当前为 "${region}"`);
     }
     const preset = provider === 'custom' ? undefined : PROVIDERS[provider];
-    const baseUrl = pick(overrides.base_url, env.UGC_LENS_BASE_URL, file.base_url, preset && resolveBaseUrl(preset, region));
+    const baseUrl = pick(overrides.base_url, env.UGC_LENS_BASE_URL, file.base_url, global.base_url, preset && resolveBaseUrl(preset, region));
     if (!baseUrl) {
         throw new ConfigError('custom 提供商必须设置 base_url（UGC_LENS_BASE_URL 或 llm.base_url）');
     }
-    const apiKeyEnv = pick(overrides.api_key_env, file.api_key_env, preset?.apiKeyEnv);
-    const apiKey = apiKeyEnv ? pick(env[apiKeyEnv]) : undefined;
+    const apiKeyEnv = pick(overrides.api_key_env, file.api_key_env, global.api_key_env, preset?.apiKeyEnv);
+    const apiKey = pick(overrides.api_key, apiKeyEnv ? env[apiKeyEnv] : undefined, file.api_key, global.api_key);
     // 本地服务（ollama、lmstudio、custom）可以没有 key
     if (preset?.apiKeyEnv && !apiKey && provider !== 'lmstudio') {
-        throw new ConfigError(`缺少 API key：请在环境变量或 .env 中设置 ${apiKeyEnv}`);
+        throw new ConfigError(`缺少 API key：设置 ${apiKeyEnv}，或运行 ugc-lens init 写入全局配置`);
     }
-    const model = pick(overrides.model, env.UGC_LENS_MODEL, file.model) ?? '';
+    const model = pick(overrides.model, env.UGC_LENS_MODEL, file.model, global.model) ?? '';
     return {
         provider,
         model,
         baseUrl: baseUrl.replace(/\/+$/, ''),
         apiKey,
-        timeoutMs: overrides.timeout_ms ?? file.timeout_ms ?? 120_000,
-        maxRetries: overrides.max_retries ?? file.max_retries ?? 3,
+        timeoutMs: overrides.timeout_ms ?? file.timeout_ms ?? global.timeout_ms ?? 120_000,
+        maxRetries: overrides.max_retries ?? file.max_retries ?? global.max_retries ?? 3,
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ugc-lens",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "面向评论、弹幕等用户互动内容的轻量分析 Agent",
   "keywords": [
     "ugc",