@templmf/temp-solf-lmf 0.0.114 → 0.0.116

package/agent.js

@@ -0,0 +1,218 @@
+/**
+ * agent.js
+ * 基于 LangGraph 1.3.4 + @langchain/anthropic 1.4.0 的 GitLab 意图识别 Agent
+ *
+ * 架构：
+ *   START → intent_router
+ *               ├── YES → gitlab_agent ⟷ gitlab_tools  (ReAct 循环)
+ *               └── NO  → general_agent → END
+ */
+
+"use strict";
+
+const {
+  StateGraph,
+  MessagesAnnotation,
+  Annotation,
+  START,
+  END,
+  MemorySaver,
+} = require("@langchain/langgraph");
+const { ToolNode, toolsCondition } = require("@langchain/langgraph/prebuilt");
+const { ChatAnthropic } = require("@langchain/anthropic");
+const { HumanMessage, SystemMessage } = require("@langchain/core/messages");
+const { GITLAB_TOOLS } = require("./gitlabTools");
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 1. State 定义
+//    在 MessagesAnnotation.spec 基础上加 intent 字段
+// ─────────────────────────────────────────────────────────────────────────────
+const AgentAnnotation = Annotation.Root({
+  ...MessagesAnnotation.spec,
+  intent: Annotation({
+    reducer: (_, newVal) => newVal,  // 直接覆盖
+    default: () => "",
+  }),
+});
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 2. 模型初始化
+// ─────────────────────────────────────────────────────────────────────────────
+const MODEL = "claude-sonnet-4-5";
+
+/** 意图识别：不绑定 tools，只返回 YES/NO */
+const intentLLM = new ChatAnthropic({
+  model: MODEL,
+  temperature: 0,
+  maxTokens: 16,
+});
+
+/** GitLab Agent：绑定全部 14 个工具 */
+const gitlabLLM = new ChatAnthropic({
+  model: MODEL,
+  temperature: 0,
+  maxTokens: 4096,
+}).bindTools(GITLAB_TOOLS);
+
+/** 通用对话：不携带任何工具 */
+const generalLLM = new ChatAnthropic({
+  model: MODEL,
+  temperature: 0.7,
+  maxTokens: 2048,
+});
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 3. System Prompts
+// ─────────────────────────────────────────────────────────────────────────────
+const INTENT_SYSTEM = `你是一个意图分类器。
+判断用户消息是否与 GitLab 操作相关（项目、Issue、MR、流水线、分支、仓库文件、成员等）。
+只回复 YES 或 NO，不要解释，不要其他内容。`;
+
+const GITLAB_SYSTEM = `你是一名专业的 GitLab 助手，可以通过调用工具来操作 GitLab。
+可用操作包括：项目管理、Issue 管理、Merge Request、CI/CD 流水线、分支与文件查询、成员管理等。
+请根据用户需求选择合适的工具；如需多步操作，请逐步完成并给出清晰说明。
+工具调用结果如有错误，请如实告知用户并提供建议。`;
+
+const GENERAL_SYSTEM = `你是一个全能 AI 助手，回答用户各类问题。
+回答应简洁准确、有帮助。如果问题涉及 GitLab 操作，提示用户可以直接让你操作 GitLab。`;
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 4. 节点函数
+// ─────────────────────────────────────────────────────────────────────────────
+
+/** 意图识别节点 */
+async function intentRouterNode(state) {
+  const lastHuman = [...state.messages]
+    .reverse()
+    .find((m) => m._getType?.() === "human" || m.constructor?.name === "HumanMessage");
+
+  if (!lastHuman) return { intent: "general" };
+
+  const response = await intentLLM.invoke([
+    new SystemMessage(INTENT_SYSTEM),
+    new HumanMessage(String(lastHuman.content)),
+  ]);
+
+  const intent = response.content.toUpperCase().includes("YES") ? "gitlab" : "general";
+  return { intent };
+}
+
+/** GitLab Agent 节点：携带工具推理 */
+async function gitlabAgentNode(state) {
+  const messages = [new SystemMessage(GITLAB_SYSTEM), ...state.messages];
+  const response = await gitlabLLM.invoke(messages);
+  return { messages: [response] };
+}
+
+/** 通用对话节点 */
+async function generalAgentNode(state) {
+  const messages = [new SystemMessage(GENERAL_SYSTEM), ...state.messages];
+  const response = await generalLLM.invoke(messages);
+  return { messages: [response] };
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 5. 条件路由
+// ─────────────────────────────────────────────────────────────────────────────
+
+function routeByIntent(state) {
+  return state.intent === "gitlab" ? "gitlab_agent" : "general_agent";
+}
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 6. 构建 Graph
+// ─────────────────────────────────────────────────────────────────────────────
+
+function buildGraph() {
+  const graph = new StateGraph(AgentAnnotation);
+
+  // 节点注册
+  graph.addNode("intent_router",  intentRouterNode);
+  graph.addNode("gitlab_agent",   gitlabAgentNode);
+  graph.addNode("general_agent",  generalAgentNode);
+  graph.addNode("gitlab_tools",   new ToolNode(GITLAB_TOOLS));
+
+  // 边：入口
+  graph.addEdge(START, "intent_router");
+
+  // 边：意图路由
+  graph.addConditionalEdges("intent_router", routeByIntent, {
+    gitlab_agent: "gitlab_agent",
+    general_agent: "general_agent",
+  });
+
+  // 边：GitLab Agent → tool_calls? YES=tools, NO=END (ReAct 循环)
+  graph.addConditionalEdges("gitlab_agent", toolsCondition, {
+    tools: "gitlab_tools",
+    __end__: END,
+  });
+
+  // 边：工具执行完成 → 回到 GitLab Agent
+  graph.addEdge("gitlab_tools", "gitlab_agent");
+
+  // 边：通用对话直接结束
+  graph.addEdge("general_agent", END);
+
+  return graph.compile({
+    checkpointer: new MemorySaver(),  // 多轮对话内存持久化
+  });
+}
+
+const compiledGraph = buildGraph();
+
+// ─────────────────────────────────────────────────────────────────────────────
+// 7. 便捷调用接口
+// ─────────────────────────────────────────────────────────────────────────────
+
+/**
+ * 发送消息，返回 Agent 最终回复文本
+ * @param {string} userMessage
+ * @param {string} threadId    - 相同 ID = 同一多轮对话
+ * @returns {Promise<string>}
+ */
+async function chat(userMessage, threadId = "default") {
+  const config = { configurable: { thread_id: threadId } };
+  const input = { messages: [new HumanMessage(userMessage)] };
+
+  const result = await compiledGraph.invoke(input, config);
+  const last = result.messages[result.messages.length - 1];
+  return typeof last.content === "string" ? last.content : JSON.stringify(last.content);
+}
+
+/**
+ * 发送消息，返回完整执行细节（意图 + 工具调用链 + 最终回复）
+ * @param {string} userMessage
+ * @param {string} threadId
+ * @returns {Promise<{intent: string, toolCalls: Array, finalAnswer: string}>}
+ */
+async function chatWithDetails(userMessage, threadId = "default") {
+  const config = { configurable: { thread_id: threadId } };
+  const input = { messages: [new HumanMessage(userMessage)] };
+
+  const result = await compiledGraph.invoke(input, config);
+
+  const toolCalls = [];
+  let finalAnswer = "";
+
+  for (const msg of result.messages) {
+    if (msg.constructor?.name === "AIMessage" || msg._getType?.() === "ai") {
+      if (msg.tool_calls?.length) {
+        for (const tc of msg.tool_calls) {
+          toolCalls.push({ tool: tc.name, args: tc.args });
+        }
+      } else if (msg.content) {
+        finalAnswer = typeof msg.content === "string"
+          ? msg.content
+          : JSON.stringify(msg.content);
+      }
+    }
+  }
+
+  return {
+    intent: result.intent ?? "unknown",
+    toolCalls,
+    finalAnswer,
+  };
+}
+
+module.exports = { chat, chatWithDetails, compiledGraph };

package/demo.js

@@ -0,0 +1,119 @@
+/**
+ * demo.js
+ * GitLab Intent Agent 演示脚本
+ *
+ * 运行前设置环境变量：
+ *   export ANTHROPIC_API_KEY=sk-ant-...
+ *
+ * 运行：
+ *   node demo.js
+ */
+
+"use strict";
+
+if (!process.env.ANTHROPIC_API_KEY) {
+  console.error("⚠️  请先设置 ANTHROPIC_API_KEY 环境变量");
+  process.exit(1);
+}
+
+const { chat, chatWithDetails } = require("./src/agent");
+
+const SEP = "─".repeat(60);
+
+async function demoCase(title, message, threadId = "demo") {
+  console.log(`\n${SEP}`);
+  console.log(`📌 ${title}`);
+  console.log(`👤 用户: ${message}`);
+  console.log(SEP);
+
+  const details = await chatWithDetails(message, threadId);
+
+  const intentIcon = details.intent === "gitlab" ? "🦊 GitLab" : "💬 通用对话";
+  console.log(`🎯 意图识别: ${intentIcon}`);
+
+  if (details.toolCalls.length > 0) {
+    console.log("🔧 工具调用链:");
+    details.toolCalls.forEach((tc, i) => {
+      const argsStr = Object.entries(tc.args)
+        .map(([k, v]) => `${k}=${JSON.stringify(v)}`)
+        .join(", ");
+      console.log(`   ${i + 1}. ${tc.tool}(${argsStr})`);
+    });
+  }
+
+  console.log(`\n🤖 Agent 回复:\n${details.finalAnswer}`);
+}
+
+async function runDemos() {
+  console.log("=".repeat(60));
+  console.log("       GitLab Intent Agent (Node.js) — 演示");
+  console.log("=".repeat(60));
+
+  // 通用问题 → general_agent，不触发任何工具
+  await demoCase(
+    "通用问题（非 GitLab）",
+    "JavaScript 中 Promise 和 async/await 有什么区别？",
+    "general-1"
+  );
+
+  // GitLab：列出项目
+  await demoCase(
+    "GitLab：列出我的项目",
+    "帮我查看一下我所有的 GitLab 项目",
+    "gitlab-1"
+  );
+
+  // GitLab：查询 Issue
+  await demoCase(
+    "GitLab：查询 bug 类 Issue",
+    "列出项目 1 里所有打开的 bug 类型 Issue",
+    "gitlab-2"
+  );
+
+  // GitLab：创建 Issue
+  await demoCase(
+    "GitLab：创建 Issue",
+    "在项目 2 里创建一个标题为"修复首页加载慢"、标签为 bug 的 Issue，指派给 alice",
+    "gitlab-3"
+  );
+
+  // GitLab：查询失败流水线 + 重试
+  await demoCase(
+    "GitLab：查询失败的流水线",
+    "查看项目 1 所有失败的 CI 流水线，并告诉我能不能重试",
+    "gitlab-4"
+  );
+
+  // GitLab：列出 MR
+  await demoCase(
+    "GitLab：列出合并请求",
+    "项目 1 现在有哪些打开的 MR？",
+    "gitlab-5"
+  );
+
+  // ── 多轮对话演示（同一 thread_id）──
+  console.log(`\n${SEP}`);
+  console.log("📌 多轮对话演示（同一 thread）");
+  console.log(SEP);
+
+  const tid = "multi-turn";
+
+  const q1 = "帮我列出项目 1 的所有分支";
+  console.log(`👤 第1轮: ${q1}`);
+  const r1 = await chat(q1, tid);
+  console.log(`🤖 回复: ${r1.slice(0, 200)}\n`);
+
+  const q2 = "刚才你说的那些分支里，有没有已经合并了的？";
+  console.log(`👤 第2轮: ${q2}`);
+  const r2 = await chat(q2, tid);
+  console.log(`🤖 回复: ${r2.slice(0, 200)}`);
+
+  console.log(`\n${SEP}`);
+  console.log("✅ 演示完成");
+  console.log(SEP);
+}
+
+runDemos().catch((err) => {
+  console.error("❌ 运行出错:", err.message);
+  process.exit(1);
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@templmf/temp-solf-lmf",
-  "version": "0.0.114",
+  "version": "0.0.116",
   "description": "",
   "main": "index.js",
   "scripts": {