@axiom-lattice/core 2.1.77 → 2.1.78

package/dist/index.mjs

@@ -1665,6 +1665,7 @@ var InMemoryProjectStore = class {
       workspaceId,
       name: data.name,
       description: data.description,
+      config: data.config,
       createdAt: now,
       updatedAt: now
     };
@@ -1672,6 +1673,13 @@ var InMemoryProjectStore = class {
     this.projects.set(key, project);
     return project;
   }
+  /**
+   * Update an existing project
+   *
+   * @remarks
+   * - The `config` field uses **replace** semantics: if provided, it completely
+   *   overwrites the existing config. To preserve the current config, omit this field.
+   */
   async updateProject(tenantId, id, updates) {
     const key = this.getKey(tenantId, id);
     const existing = this.projects.get(key);
@@ -2679,6 +2687,87 @@ var InMemoryBindingStore = class {
   }
 };
 
+// src/store_lattice/InMemoryA2AApiKeyStore.ts
+import { randomUUID as randomUUID2 } from "crypto";
+function generateApiKey() {
+  return `a2a_${randomUUID2().replace(/-/g, "")}`;
+}
+var InMemoryA2AApiKeyStore = class {
+  constructor() {
+    this.keys = /* @__PURE__ */ new Map();
+  }
+  async findByKey(key) {
+    for (const record of this.keys.values()) {
+      if (record.key === key && record.enabled) return record;
+    }
+    return null;
+  }
+  async list(params) {
+    let records = Array.from(this.keys.values());
+    if (params.tenantId) {
+      records = records.filter((r) => r.tenantId === params.tenantId);
+    }
+    const offset = params.offset || 0;
+    const limit = params.limit;
+    records = records.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
+    return limit ? records.slice(offset, offset + limit) : records.slice(offset);
+  }
+  async create(input) {
+    const now = /* @__PURE__ */ new Date();
+    const record = {
+      id: randomUUID2(),
+      key: generateApiKey(),
+      tenantId: input.tenantId,
+      projectId: input.projectId,
+      workspaceId: input.workspaceId,
+      label: input.label,
+      enabled: true,
+      createdAt: now,
+      updatedAt: now
+    };
+    this.keys.set(record.id, record);
+    return record;
+  }
+  async disable(id) {
+    const record = this.keys.get(id);
+    if (!record) throw new Error(`A2A API key not found: ${id}`);
+    const updated = { ...record, enabled: false, updatedAt: /* @__PURE__ */ new Date() };
+    this.keys.set(id, updated);
+    return updated;
+  }
+  async enable(id) {
+    const record = this.keys.get(id);
+    if (!record) throw new Error(`A2A API key not found: ${id}`);
+    const updated = { ...record, enabled: true, updatedAt: /* @__PURE__ */ new Date() };
+    this.keys.set(id, updated);
+    return updated;
+  }
+  async rotate(id) {
+    const record = this.keys.get(id);
+    if (!record) throw new Error(`A2A API key not found: ${id}`);
+    const updated = { ...record, key: generateApiKey(), updatedAt: /* @__PURE__ */ new Date() };
+    this.keys.set(id, updated);
+    return updated;
+  }
+  async delete(id) {
+    this.keys.delete(id);
+  }
+  async loadIntoMap() {
+    const map = /* @__PURE__ */ new Map();
+    for (const record of this.keys.values()) {
+      if (record.enabled) {
+        map.set(record.key, {
+          key: record.key,
+          tenantId: record.tenantId,
+          projectId: record.projectId,
+          workspaceId: record.workspaceId
+        });
+      }
+    }
+    return map;
+  }
+};
+
 // src/store_lattice/StoreLatticeManager.ts
 var StoreLatticeManager = class _StoreLatticeManager extends BaseLatticeManager {
   /**
@@ -2853,6 +2942,12 @@ storeLatticeManager.registerLattice(
   "channelBinding",
   defaultChannelBindingStore
 );
+var defaultA2AApiKeyStore = new InMemoryA2AApiKeyStore();
+storeLatticeManager.registerLattice(
+  "default",
+  "a2aApiKey",
+  defaultA2AApiKeyStore
+);
 
 // src/tool_lattice/manage_binding/index.ts
 function getInstallationStore() {
@@ -6949,6 +7044,7 @@ import {
   isDeepAgentConfig,
   isProcessingAgentConfig,
   isTeamAgentConfig,
+  isA2ARemoteAgentConfig,
   getToolsFromConfig,
   getSubAgentsFromConfig
 } from "@axiom-lattice/protocols";
@@ -18093,6 +18189,227 @@ var ProcessingAgentGraphBuilder = class {
   }
 };
 
+// src/agent_lattice/builders/RemoteAgentGraphBuilder.ts
+import { StateGraph as StateGraph2, MessagesAnnotation } from "@langchain/langgraph";
+import { AIMessage as AIMessage4 } from "@langchain/core/messages";
+
+// src/services/a2a-client.ts
+import { v4 as v42 } from "uuid";
+var A2ARemoteError = class extends Error {
+  constructor(message, statusCode, body) {
+    super(message);
+    this.statusCode = statusCode;
+    this.body = body;
+    this.name = "A2ARemoteError";
+  }
+};
+var A2ARemoteClient = class {
+  constructor(config) {
+    this.config = config;
+    this.jsonRpcUrl = null;
+    this.card = null;
+  }
+  // ── Resolution ───────────────────────────────────────────────────────────
+  /**
+   * Fetch the agent card and resolve the JSON-RPC endpoint.
+   */
+  async resolve() {
+    if (this.card) return this.card;
+    const resp = await fetch(this.config.agentCardUrl, {
+      signal: AbortSignal.timeout(this.config.timeout ?? 3e4)
+    });
+    if (!resp.ok) {
+      throw new A2ARemoteError(
+        `Failed to fetch agent card: HTTP ${resp.status}`,
+        resp.status
+      );
+    }
+    this.card = await resp.json();
+    const card = this.card;
+    const jsonRpc = card.additionalInterfaces?.find(
+      (i) => i.transport === "JSONRPC"
+    );
+    this.jsonRpcUrl = jsonRpc?.url ?? card.url;
+    return this.card;
+  }
+  // ── Execute ──────────────────────────────────────────────────────────────
+  /**
+   * Send a text prompt to the remote A2A agent and return the response.
+   *
+   * Supports SSE streaming (parses status-update events) and falls back
+   * to polling the task status for non-streaming agents.
+   */
+  async sendMessage(text) {
+    await this.resolve();
+    const taskId = v42();
+    const body = JSON.stringify({
+      jsonrpc: "2.0",
+      method: "tasks/send",
+      params: {
+        id: taskId,
+        message: {
+          role: "user",
+          parts: [{ kind: "text", text }]
+        }
+      },
+      id: taskId
+    });
+    const headers = {
+      "Content-Type": "application/json",
+      "Accept": "text/event-stream"
+    };
+    if (this.config.apiKey) {
+      headers["Authorization"] = `Bearer ${this.config.apiKey}`;
+    }
+    const resp = await fetch(this.jsonRpcUrl, {
+      method: "POST",
+      headers,
+      body,
+      signal: AbortSignal.timeout(this.config.timeout ?? 3e5)
+    });
+    if (!resp.ok) {
+      const text2 = await resp.text().catch(() => "");
+      throw new A2ARemoteError(
+        `A2A request failed: HTTP ${resp.status}`,
+        resp.status,
+        text2
+      );
+    }
+    const contentType = resp.headers.get("content-type") ?? "";
+    if (contentType.includes("text/event-stream")) {
+      return this.parseSSE(resp);
+    }
+    const json = await resp.json();
+    return this.extractArtifactText(json);
+  }
+  // ── SSE Parsing ──────────────────────────────────────────────────────────
+  async parseSSE(resp) {
+    if (!resp.body) {
+      throw new A2ARemoteError("Empty response body");
+    }
+    const reader = resp.body.getReader();
+    const decoder = new TextDecoder();
+    let buffer2 = "";
+    let accumulatedText = "";
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer2 += decoder.decode(value, { stream: true });
+        const lines = buffer2.split("\n");
+        buffer2 = lines.pop() ?? "";
+        for (const line of lines) {
+          if (line.startsWith("data: ")) {
+            const data = line.slice(6).trim();
+            if (!data) continue;
+            try {
+              const event = JSON.parse(data);
+              this.handleSSEEvent(event, (t) => {
+                accumulatedText += t;
+              });
+            } catch {
+            }
+          }
+        }
+      }
+      if (buffer2.startsWith("data: ")) {
+        const data = buffer2.slice(6).trim();
+        if (data) {
+          try {
+            const event = JSON.parse(data);
+            this.handleSSEEvent(event, (t) => {
+              accumulatedText += t;
+            });
+          } catch {
+          }
+        }
+      }
+    } finally {
+      reader.releaseLock();
+    }
+    return accumulatedText;
+  }
+  handleSSEEvent(event, onText) {
+    const e = event;
+    const messageParts = e.status?.message?.parts;
+    if (messageParts) {
+      for (const part of messageParts) {
+        if (part.text) onText(part.text);
+      }
+    }
+  }
+  // ── Artifact Extraction ──────────────────────────────────────────────────
+  extractArtifactText(task) {
+    const artifacts = task.artifacts ?? [];
+    return artifacts.flatMap((a) => a.parts ?? []).filter((p) => p.text).map((p) => p.text).join("\n");
+  }
+};
+
+// src/agent_lattice/builders/RemoteAgentGraphBuilder.ts
+import { AgentType as AgentType2 } from "@axiom-lattice/protocols";
+var RemoteAgentGraphBuilder = class {
+  async build(agentLattice, params) {
+    if (agentLattice.config.type !== AgentType2.A2A_REMOTE) {
+      throw new Error(
+        `RemoteAgentGraphBuilder received wrong agent type: ${agentLattice.config.type}`
+      );
+    }
+    const config = agentLattice.config;
+    const client = new A2ARemoteClient({
+      agentCardUrl: config.agentCardUrl,
+      apiKey: config.apiKey,
+      timeout: config.timeout
+    });
+    await client.resolve();
+    const remoteCallNode = async (state) => {
+      const messages = state.messages ?? [];
+      const text = extractLastHumanMessage(messages);
+      if (!text) {
+        return {
+          messages: [
+            new AIMessage4("No text input provided to remote agent.")
+          ]
+        };
+      }
+      try {
+        const fullPrompt = params.prompt ? `${params.prompt}
+
+User request:
+${text}` : text;
+        const response = await client.sendMessage(fullPrompt);
+        return {
+          messages: [new AIMessage4(response)]
+        };
+      } catch (error) {
+        const msg = error.message ?? String(error);
+        return {
+          messages: [
+            new AIMessage4(`Remote A2A agent error: ${msg}`)
+          ]
+        };
+      }
+    };
+    const workflow = new StateGraph2(MessagesAnnotation).addNode("remote_call", remoteCallNode).addEdge("__start__", "remote_call").addEdge("remote_call", "__end__");
+    return workflow.compile();
+  }
+};
+function extractLastHumanMessage(messages) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const msg = messages[i];
+    if (msg._getType() === "human") {
+      const content = msg.content;
+      if (typeof content === "string") return content;
+      if (Array.isArray(content)) {
+        return content.filter(
+          (c) => typeof c === "object" && c !== null && c.type === "text"
+        ).map((c) => c.text).join("\n");
+      }
+      return String(content);
+    }
+  }
+  return "";
+}
+
 // src/agent_lattice/builders/AgentGraphBuilderFactory.ts
 var AgentGraphBuilderFactory = class _AgentGraphBuilderFactory {
   constructor() {
@@ -18116,6 +18433,7 @@ var AgentGraphBuilderFactory = class _AgentGraphBuilderFactory {
     this.builders.set(AgentType.DEEP_AGENT, new DeepAgentGraphBuilder());
     this.builders.set(AgentType.TEAM, new TeamAgentGraphBuilder());
     this.builders.set(AgentType.PROCESSING, new ProcessingAgentGraphBuilder());
+    this.builders.set(AgentType.A2A_REMOTE, new RemoteAgentGraphBuilder());
   }
   /**
    * 注册自定义Builder
@@ -18792,8 +19110,8 @@ ${body}` : `${frontmatter}
 
 // src/agent_lattice/agentArchitectTools.ts
 import z55 from "zod";
-import { v4 as v42 } from "uuid";
-import { AgentType as AgentType2 } from "@axiom-lattice/protocols";
+import { v4 as v43 } from "uuid";
+import { AgentType as AgentType3 } from "@axiom-lattice/protocols";
 function getTenantId(exeConfig) {
   const runConfig = exeConfig?.configurable?.runConfig || {};
   return runConfig.tenantId || "default";
@@ -18901,7 +19219,7 @@ registerToolLattice(
         key: id,
         name: input.name,
         description: input.description || "",
-        type: input.type === "deep_agent" ? AgentType2.DEEP_AGENT : AgentType2.REACT,
+        type: input.type === "deep_agent" ? AgentType3.DEEP_AGENT : AgentType3.REACT,
         prompt: input.prompt,
         ...input.tools && input.tools.length > 0 ? { tools: input.tools } : {},
         ...input.middleware && input.middleware.length > 0 ? { middleware: input.middleware } : {},
@@ -18990,7 +19308,7 @@ registerToolLattice(
         key: id,
         name: input.name,
         description: input.description || "",
-        type: AgentType2.PROCESSING,
+        type: AgentType3.PROCESSING,
         prompt: input.prompt,
         ...input.tools && input.tools.length > 0 ? { tools: input.tools } : {},
         middleware,
@@ -19058,7 +19376,7 @@ registerToolLattice(
         return JSON.stringify({ error: `Agent '${input.id}' not found` });
       }
       const existingConfig = existing.graphDefinition || {};
-      if (existingConfig.type !== AgentType2.PROCESSING) {
+      if (existingConfig.type !== AgentType3.PROCESSING) {
         return JSON.stringify({ error: `Agent '${input.id}' is not a PROCESSING agent (type: ${existingConfig.type})` });
       }
       const normalize = (s) => s.toLowerCase().replace(/[_\-\s]/g, "");
@@ -19266,7 +19584,7 @@ registerToolLattice(
       if (!existing) {
         return JSON.stringify({ error: `Agent '${id}' not found` });
       }
-      const threadId = v42();
+      const threadId = v43();
       const agent = new Agent({
         tenant_id: tenantId,
         assistant_id: id,
@@ -19290,7 +19608,7 @@ registerToolLattice(
 );
 
 // src/agent_lattice/agentArchitectConfig.ts
-import { AgentType as AgentType4 } from "@axiom-lattice/protocols";
+import { AgentType as AgentType5 } from "@axiom-lattice/protocols";
 
 // src/agent_lattice/agentArchitectPrompt.ts
 var AGENT_ARCHITECT_PROMPT = `# Agent Architect
@@ -19930,7 +20248,7 @@ Updates a PROCESSING agent's topology edges, name, prompt, or sub-agents. Unlike
 `;
 
 // src/agent_lattice/agentReviewerConfig.ts
-import { AgentType as AgentType3 } from "@axiom-lattice/protocols";
+import { AgentType as AgentType4 } from "@axiom-lattice/protocols";
 
 // src/agent_lattice/agentReviewerPrompt.ts
 var AGENT_REVIEWER_PROMPT = `# Agent Reviewer
@@ -19993,7 +20311,7 @@ var agentReviewerConfig = {
   key: AGENT_REVIEWER_KEY,
   name: "Agent Reviewer",
   description: "Review and test AI agents. Use this agent to check agent configurations for correctness and to test agents by invoking them with realistic messages.",
-  type: AgentType3.REACT,
+  type: AgentType4.REACT,
   prompt: AGENT_REVIEWER_PROMPT,
   tools: [
     "invoke_agent",
@@ -20019,7 +20337,7 @@ var agentArchitectConfig = {
   key: AGENT_ARCHITECT_KEY,
   name: "Agent Architect",
   description: "Design and manage AI agents through natural language conversation. Use this agent when you want to create a new agent, modify an existing agent, or manage your agent collection (list, view, update, delete).",
-  type: AgentType4.DEEP_AGENT,
+  type: AgentType5.DEEP_AGENT,
   prompt: AGENT_ARCHITECT_PROMPT,
   tools: [
     "list_agents",
@@ -22849,6 +23167,7 @@ export {
   FileSystemSkillStore,
   FilesystemBackend,
   HumanMessage3 as HumanMessage,
+  InMemoryA2AApiKeyStore,
   InMemoryAssistantStore,
   InMemoryBindingStore,
   InMemoryChannelInstallationStore,