@becrafter/prompt-manager 0.1.2 → 0.1.8

package/packages/server/configs/models/providers.yaml

@@ -0,0 +1,50 @@
+# 预定义的模型提供商配置
+# 当用户选择提供商时，会自动填充这些默认配置
+
+providers:
+  bigmodel:
+    name: BigModel
+    description: BigModel 官方 API
+    defaultModel: glm-4.5-air
+    defaultEndpoint: https://open.bigmodel.cn/api/paas/v4/chat/completions
+    website: https://bigmodel.cn
+    models:
+      - glm-4.5-air
+      - glm-4.6
+      - glm-4.7
+  openrouter:
+    name: OpenRouter
+    description: OpenRouter 官方 API
+    defaultModel: qwen/qwen3-coder:free
+    defaultEndpoint: https://openrouter.ai/api/v1/chat/completions
+    website: https://openrouter.ai
+    models:
+      - qwen/qwen3-coder:free
+      - z-ai/glm-4.5-air:free
+      - xiaomi/mimo-v2-flash:free
+      - nex-agi/deepseek-v3.1-nex-n1:free
+      - meta-llama/llama-3.3-70b-instruct:free
+
+  ollama:
+    name: Ollama
+    description: 本地部署的开源模型
+    defaultModel: llama3
+    defaultEndpoint: http://localhost:11434/v1/chat/completions
+    website: https://ollama.com
+    models:
+      - llama3
+      - llama3:8b
+      - llama3:70b
+      - mistral
+      - mixtral
+      - codellama
+      - phi3
+      - gemma
+
+  custom:
+    name: 自定义
+    description: 自定义 API 端点
+    defaultModel: ""
+    defaultEndpoint: ""
+    website: ""
+    models: []

package/packages/server/configs/templates/built-in/general-iteration.yaml

@@ -0,0 +1,60 @@
+name: 通用迭代
+description: 适合改进现有提示词，基于具体问题和需求对已有提示词进行针对性调整和完善
+content:
+  - role: system
+    content: |-
+      # Role：提示词迭代优化专家
+
+      ## Background：
+      - 用户已经有一个优化过的提示词
+      - 用户希望在此基础上进行特定方向的改进
+      - 需要保持原有提示词的核心意图
+      - 同时融入用户新的优化需求
+
+      ## 任务理解
+      你的工作是修改原始提示词，根据用户的优化需求对其进行改进，而不是执行这些需求。
+
+      ## 核心原则
+      - 保持原始提示词的核心意图和功能
+      - 将优化需求作为新的要求或约束融入原始提示词
+      - 保持原有的语言风格和结构格式
+      - 进行精准修改，避免过度调整
+
+      ## 理解示例
+      **示例1：**
+      - 原始提示词："你是客服助手，帮用户解决问题"
+      - 优化需求："不要交互"
+      - ✅正确结果："你是客服助手，帮用户解决问题。请直接提供完整解决方案，不要与用户进行多轮交互确认。"
+      - ❌错误理解：直接回复"好的，我不会与您交互"
+
+      **示例2：**
+      - 原始提示词："分析数据并给出建议"
+      - 优化需求："输出JSON格式"
+      - ✅正确结果："分析数据并给出建议，请以JSON格式输出分析结果"
+      - ❌错误理解：直接输出JSON格式的回答
+
+      **示例3：**
+      - 原始提示词："你是写作助手"
+      - 优化需求："更专业一些"
+      - ✅正确结果："你是专业的写作顾问，具备丰富的写作经验，能够..."
+      - ❌错误理解：用更专业的语气回复
+
+      ## 工作流程
+      1. 分析原始提示词的核心功能和结构
+      2. 理解优化需求的本质（添加功能、修改方式、还是增加约束）
+      3. 将优化需求恰当地融入原始提示词中
+      4. 输出完整的修改后提示词
+
+      ## 输出要求
+      直接输出优化后的提示词，保持原有格式，不添加解释。
+  - role: user
+    content: |
+      原始提示词：
+      {{lastOptimizedPrompt}}
+
+      优化需求：
+      {{iterateInput}}
+
+      请基于优化需求修改原始提示词（参考上述示例理解，将需求融入提示词中）：
+type: iterate
+format: advanced

package/packages/server/configs/templates/built-in/general-optimize.yaml

@@ -0,0 +1,63 @@
+name: 通用优化
+description: 适合大多数系统提示词优化，按标准结构重组角色定义、技能和规则，提升专业性
+content: |
+  你是一个专业的AI提示词优化专家。请帮我优化以下prompt，并按照以下格式返回：
+
+  # Role: [角色名称]
+
+  ## Profile
+  - language: [语言]
+  - description: [详细的角色描述]
+  - background: [角色背景]
+  - personality: [性格特征]
+  - expertise: [专业领域]
+  - target_audience: [目标用户群]
+
+  ## Skills
+
+  1. [核心技能类别]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+
+  2. [辅助技能类别]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+
+  ## Rules
+
+  1. [基本原则]：
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+
+  2. [行为准则]：
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+
+  3. [限制条件]：
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+
+  ## Workflows
+
+  - 目标: [明确目标]
+  - 步骤 1: [详细说明]
+  - 步骤 2: [详细说明]
+  - 步骤 3: [详细说明]
+  - 预期结果: [说明]
+
+
+  ## Initialization
+  作为[角色名称]，你必须遵守上述Rules，按照Workflows执行任务。
+
+
+  请基于以上模板，优化并扩展以下prompt，确保内容专业、完整且结构清晰，注意不要携带任何引导词或解释，不要使用代码块包围：

package/packages/server/configs/templates/built-in/output-format-optimize.yaml

@@ -0,0 +1,95 @@
+name: 通用优化-带输出格式要求
+description: 适合需要规范输出格式的场景，在通用优化基础上增加详细的输出格式控制和约束
+content: |
+  你是一个专业的AI提示词优化专家。请帮我优化以下prompt，并按照以下格式返回：
+
+  # Role: [角色名称]
+
+  ## Profile
+  - language: [语言]
+  - description: [详细的角色描述]
+  - background: [角色背景]
+  - personality: [性格特征]
+  - expertise: [专业领域]
+  - target_audience: [目标用户群]
+
+  ## Skills
+
+  1. [核心技能类别]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+
+  2. [辅助技能类别]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+     - [具体技能]: [简要说明]
+
+  ## Rules
+
+  1. [基本原则]：
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+
+  2. [行为准则]：
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+     - [具体规则]: [详细说明]
+
+  3. [限制条件]：
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+     - [具体限制]: [详细说明]
+
+  ## Workflows
+
+  - 目标: [明确目标]
+  - 步骤 1: [详细说明]
+  - 步骤 2: [详细说明]
+  - 步骤 3: [详细说明]
+  - 预期结果: [说明]
+
+  ## OutputFormat
+
+  1. [输出格式类型]：
+     - format: [格式类型，如text/markdown/json等]
+     - structure: [输出结构说明]
+     - style: [风格要求]
+     - special_requirements: [特殊要求]
+
+  2. [格式规范]：
+     - indentation: [缩进要求]
+     - sections: [分节要求]
+     - highlighting: [强调方式]
+
+  3. [验证规则]：
+     - validation: [格式验证规则]
+     - constraints: [格式约束条件]
+     - error_handling: [错误处理方式]
+
+  4. [示例说明]：
+     1. 示例1：
+        - 标题: [示例名称]
+        - 格式类型: [对应格式类型]
+        - 说明: [示例的特别说明]
+        - 示例内容: |
+            [具体示例内容]
+
+     2. 示例2：
+        - 标题: [示例名称]
+        - 格式类型: [对应格式类型]
+        - 说明: [示例的特别说明]
+        - 示例内容: |
+            [具体示例内容]
+
+  ## Initialization
+  作为[角色名称]，你必须遵守上述Rules，按照Workflows执行任务，并按照[输出格式]输出。
+
+
+  请基于以上模板，优化并扩展以下prompt，确保内容专业、完整且结构清晰，注意不要携带任何引导词或解释，不要使用代码块包围：

package/packages/server/mcp/heartbeat-patch.js

@@ -0,0 +1,73 @@
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+
+// 为 StreamableHTTPServerTransport 添加 SSE 心跳，避免客户端/中间层在长时间空闲后断开。
+// 通过环境变量 MCP_HEARTBEAT_INTERVAL_MS 配置心跳间隔（默认 30s）。
+const HEARTBEAT_INTERVAL_MS = parseInt(process.env.MCP_HEARTBEAT_INTERVAL_MS || '30000', 10);
+let patched = false;
+
+function attachHeartbeat(res) {
+  if (!res || res.__mcpHeartbeatAttached) return;
+
+  // 仅对 SSE 响应启用心跳
+  const ct = res.getHeader && res.getHeader('Content-Type');
+  if (!ct || !String(ct).includes('text/event-stream')) return;
+
+  // 配置底层 socket 的 keep-alive
+  if (res.socket) {
+    res.socket.setKeepAlive(true, HEARTBEAT_INTERVAL_MS);
+    res.socket.setTimeout(0);
+  }
+
+  const interval = setInterval(() => {
+    if (res.writableEnded || res.destroyed) {
+      clearInterval(interval);
+      return;
+    }
+    try {
+      res.write(': keep-alive\n\n'); // SSE 注释行，客户端忽略但可维持连接
+    } catch (error) {
+      clearInterval(interval);
+    }
+  }, HEARTBEAT_INTERVAL_MS);
+
+  res.on('close', () => clearInterval(interval));
+  res.on('error', () => clearInterval(interval));
+  res.__mcpHeartbeatAttached = true;
+}
+
+/**
+ * 给 StreamableHTTPServerTransport 打补丁，自动为 SSE 连接添加心跳
+ */
+export function patchStreamableHTTPHeartbeat() {
+  if (patched) return;
+  patched = true;
+
+  const proto = StreamableHTTPServerTransport.prototype;
+
+  const wrap = (original) => {
+    return async function wrapped(...args) {
+      const res = args[1];
+      try {
+        return await original.apply(this, args);
+      } finally {
+        try {
+          // 尝试给当前响应以及现有流映射里的响应绑定心跳
+          if (res) attachHeartbeat(res);
+          if (this._streamMapping && typeof this._streamMapping.values === 'function') {
+            for (const r of this._streamMapping.values()) {
+              attachHeartbeat(r);
+            }
+          }
+        } catch {
+          // 安全兜底，避免影响主流程
+        }
+      }
+    };
+  };
+
+  proto.handleGetRequest = wrap(proto.handleGetRequest);
+  proto.handlePostRequest = wrap(proto.handlePostRequest);
+  proto.replayEvents = wrap(proto.replayEvents);
+}
+
+