@yeaft/webchat-agent 0.1.398 → 0.1.408

package/unify/models.js

@@ -0,0 +1,167 @@
+/**
+ * models.js — Model ID registry for Yeaft Unify
+ *
+ * Maps model IDs (e.g. "gpt-5", "claude-sonnet-4-20250514") to their
+ * adapter type, API base URL, and capabilities.
+ *
+ * Yeaft does not provide its own models. The "model" field is always a
+ * model ID from an external provider. This registry lets Yeaft auto-detect
+ * the correct adapter and endpoint from just the model ID, so users only
+ * need to set YEAFT_MODEL=gpt-5 without configuring adapter/baseUrl separately.
+ *
+ * Unknown model IDs return null — caller falls back to env-based detection.
+ */
+
+/**
+ * @typedef {Object} ModelInfo
+ * @property {'anthropic' | 'chat-completions'} adapter — Which adapter to use
+ * @property {string} baseUrl — API endpoint base URL
+ * @property {number} contextWindow — Max context tokens
+ * @property {number} maxOutputTokens — Max output tokens
+ * @property {string} displayName — Human-readable model name
+ */
+
+/** @type {Map<string, ModelInfo>} */
+export const MODEL_REGISTRY = new Map([
+  // ── Anthropic ──────────────────────────────────────────────────
+  ['claude-sonnet-4-20250514', {
+    adapter: 'anthropic',
+    baseUrl: 'https://api.anthropic.com',
+    contextWindow: 200000,
+    maxOutputTokens: 16384,
+    displayName: 'Claude Sonnet 4',
+  }],
+  ['claude-opus-4-20250514', {
+    adapter: 'anthropic',
+    baseUrl: 'https://api.anthropic.com',
+    contextWindow: 200000,
+    maxOutputTokens: 16384,
+    displayName: 'Claude Opus 4',
+  }],
+  ['claude-haiku-3-20250414', {
+    adapter: 'anthropic',
+    baseUrl: 'https://api.anthropic.com',
+    contextWindow: 200000,
+    maxOutputTokens: 8192,
+    displayName: 'Claude Haiku 3',
+  }],
+
+  // ── OpenAI ─────────────────────────────────────────────────────
+  ['gpt-5', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 256000,
+    maxOutputTokens: 16384,
+    displayName: 'GPT-5',
+  }],
+  ['gpt-5.4', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 272000,
+    maxOutputTokens: 16384,
+    displayName: 'GPT-5.4',
+  }],
+  ['gpt-4.1', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 1047576,
+    maxOutputTokens: 32768,
+    displayName: 'GPT-4.1',
+  }],
+  ['gpt-4.1-mini', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 1047576,
+    maxOutputTokens: 16384,
+    displayName: 'GPT-4.1 Mini',
+  }],
+  ['gpt-4.1-nano', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 1047576,
+    maxOutputTokens: 16384,
+    displayName: 'GPT-4.1 Nano',
+  }],
+  ['o3', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 200000,
+    maxOutputTokens: 100000,
+    displayName: 'o3',
+  }],
+  ['o4-mini', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.openai.com/v1',
+    contextWindow: 200000,
+    maxOutputTokens: 100000,
+    displayName: 'o4-mini',
+  }],
+
+  // ── DeepSeek ───────────────────────────────────────────────────
+  ['deepseek-chat', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.deepseek.com',
+    contextWindow: 131072,
+    maxOutputTokens: 8192,
+    displayName: 'DeepSeek Chat',
+  }],
+  ['deepseek-reasoner', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://api.deepseek.com',
+    contextWindow: 131072,
+    maxOutputTokens: 8192,
+    displayName: 'DeepSeek Reasoner',
+  }],
+
+  // ── Google (via OpenAI-compatible API) ─────────────────────────
+  ['gemini-2.5-pro', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://generativelanguage.googleapis.com/v1beta/openai',
+    contextWindow: 1048576,
+    maxOutputTokens: 65536,
+    displayName: 'Gemini 2.5 Pro',
+  }],
+  ['gemini-2.5-flash', {
+    adapter: 'chat-completions',
+    baseUrl: 'https://generativelanguage.googleapis.com/v1beta/openai',
+    contextWindow: 1048576,
+    maxOutputTokens: 65536,
+    displayName: 'Gemini 2.5 Flash',
+  }],
+]);
+
+/**
+ * Resolve a model name to its registry info.
+ *
+ * @param {string} modelName — The model name (e.g., 'gpt-5', 'claude-sonnet-4-20250514')
+ * @returns {ModelInfo | null} — Model info, or null if not in registry
+ */
+export function resolveModel(modelName) {
+  if (!modelName) return null;
+  const info = MODEL_REGISTRY.get(modelName);
+  // Return a shallow copy so callers can't mutate the registry
+  return info ? { ...info } : null;
+}
+
+/**
+ * List all known models.
+ *
+ * @returns {{ name: string, adapter: string, baseUrl: string, contextWindow: number, maxOutputTokens: number, displayName: string }[]}
+ */
+export function listModels() {
+  const result = [];
+  for (const [name, info] of MODEL_REGISTRY) {
+    result.push({ name, ...info });
+  }
+  return result;
+}
+
+/**
+ * Check if a model name is in the registry.
+ *
+ * @param {string} modelName
+ * @returns {boolean}
+ */
+export function isKnownModel(modelName) {
+  return MODEL_REGISTRY.has(modelName);
+}

package/unify/prompts.js

@@ -0,0 +1,61 @@
+/**
+ * prompts.js — Bilingual system prompt templates
+ *
+ * Single source of truth for system prompts. Both engine.js and cli.js
+ * import buildSystemPrompt() from here. Supports 'en' and 'zh'.
+ *
+ * To add a new language: add a new key to PROMPTS with all required fields.
+ */
+
+// ─── Prompt Templates ─────────────────────────────────────────
+
+const PROMPTS = {
+  en: {
+    identity: 'You are Yeaft, a helpful AI assistant.',
+    mode: (mode) => `Current mode: ${mode}`,
+    date: (d) => `Date: ${d}`,
+    work: 'You are in work mode. Break tasks into steps, execute them using tools, and report progress.',
+    dream: 'You are in dream mode. Reflect on past conversations and consolidate memories.',
+    tools: (names) => `Available tools: ${names}`,
+  },
+  zh: {
+    identity: '你是 Yeaft，一个有用的 AI 助手。',
+    mode: (mode) => `当前模式：${mode}`,
+    date: (d) => `日期：${d}`,
+    work: '你处于工作模式。将任务分解为步骤，使用工具执行，并报告进度。',
+    dream: '你处于梦境模式。回顾过去的对话，整理和巩固记忆。',
+    tools: (names) => `可用工具：${names}`,
+  },
+};
+
+/** Supported language codes. */
+export const SUPPORTED_LANGUAGES = Object.keys(PROMPTS);
+
+/**
+ * Build the system prompt for a given language and mode.
+ *
+ * @param {{ language?: string, mode?: string, toolNames?: string[] }} params
+ * @returns {string}
+ */
+export function buildSystemPrompt({ language = 'en', mode = 'chat', toolNames = [] } = {}) {
+  // Fallback to English for unknown languages
+  const lang = PROMPTS[language] || PROMPTS.en;
+
+  const parts = [
+    lang.identity,
+    lang.mode(mode),
+    lang.date(new Date().toISOString().split('T')[0]),
+  ];
+
+  if (mode === 'work') {
+    parts.push(lang.work);
+  } else if (mode === 'dream') {
+    parts.push(lang.dream);
+  }
+
+  if (toolNames.length > 0) {
+    parts.push(lang.tools(toolNames.join(', ')));
+  }
+
+  return parts.join('\n\n');
+}