@axiom-lattice/gateway 2.1.15 → 2.1.17

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/gateway@2.1.15 build /home/runner/work/agentic/agentic/packages/gateway
+> @axiom-lattice/gateway@2.1.17 build /home/runner/work/agentic/agentic/packages/gateway
 > tsup src/index.ts --format cjs,esm --dts --clean --sourcemap
 
 CLI Building entry: src/index.ts
@@ -9,13 +9,13 @@
 CLI Cleaning output folder
 CJS Build start
 ESM Build start
-ESM dist/index.mjs     56.17 KB
-ESM dist/index.mjs.map 121.06 KB
-ESM ⚡️ Build success in 143ms
-CJS dist/index.js     58.53 KB
-CJS dist/index.js.map 121.16 KB
-CJS ⚡️ Build success in 145ms
+ESM dist/index.mjs     55.49 KB
+ESM dist/index.mjs.map 119.35 KB
+ESM ⚡️ Build success in 105ms
+CJS dist/index.js     57.99 KB
+CJS dist/index.js.map 119.45 KB
+CJS ⚡️ Build success in 107ms
 DTS Build start
-DTS ⚡️ Build success in 8167ms
+DTS ⚡️ Build success in 7551ms
 DTS dist/index.d.ts  3.32 KB
 DTS dist/index.d.mts 3.32 KB

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @axiom-lattice/gateway
 
+## 2.1.17
+
+### Patch Changes
+
+- dceba32: add health
+
+## 2.1.16
+
+### Patch Changes
+
+- 7d8f3f5: enhance lattice
+- Updated dependencies [7d8f3f5]
+  - @axiom-lattice/protocols@2.1.6
+  - @axiom-lattice/core@2.1.12
+  - @axiom-lattice/queue-redis@1.0.5
+
 ## 2.1.15
 
 ### Patch Changes

package/dist/index.js

@@ -248,73 +248,10 @@ async function resume_stream({
   };
 }
 
-// src/stores/assistant_store.ts
-var AssistantStore = class {
-  constructor() {
-    this.assistants = /* @__PURE__ */ new Map();
-  }
-  /**
-   * Get all assistants
-   */
-  getAllAssistants() {
-    return Array.from(this.assistants.values());
-  }
-  /**
-   * Get assistant by ID
-   */
-  getAssistantById(id) {
-    return this.assistants.get(id);
-  }
-  /**
-   * Create a new assistant
-   */
-  createAssistant(id, data) {
-    const now = /* @__PURE__ */ new Date();
-    const assistant = {
-      id,
-      name: data.name,
-      description: data.description,
-      graphDefinition: data.graphDefinition,
-      createdAt: now,
-      updatedAt: now
-    };
-    this.assistants.set(id, assistant);
-    return assistant;
-  }
-  /**
-   * Update an existing assistant
-   */
-  updateAssistant(id, updates) {
-    const existing = this.assistants.get(id);
-    if (!existing) {
-      return null;
-    }
-    const updated = {
-      ...existing,
-      ...updates,
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    this.assistants.set(id, updated);
-    return updated;
-  }
-  /**
-   * Delete an assistant by ID
-   */
-  deleteAssistant(id) {
-    return this.assistants.delete(id);
-  }
-  /**
-   * Check if assistant exists
-   */
-  hasAssistant(id) {
-    return this.assistants.has(id);
-  }
-};
-var assistantStore = new AssistantStore();
-
 // src/controllers/assistant.ts
-var import_crypto = require("crypto");
 var import_core2 = require("@axiom-lattice/core");
+var import_crypto = require("crypto");
+var import_core3 = require("@axiom-lattice/core");
 function convertAgentConfigToAssistant(config) {
   return {
     id: config.key,
@@ -329,11 +266,13 @@ function convertAgentConfigToAssistant(config) {
   };
 }
 async function getAssistantList(request, reply) {
-  const agentConfigs = await (0, import_core2.getAllAgentConfigs)();
+  const agentConfigs = await (0, import_core3.getAllAgentConfigs)();
   const codeConfiguredAssistants = agentConfigs.map(
     convertAgentConfigToAssistant
   );
-  const storedAssistants = assistantStore.getAllAssistants();
+  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const storedAssistants = await assistantStore.getAllAssistants();
   const assistantMap = /* @__PURE__ */ new Map();
   codeConfiguredAssistants.forEach((assistant) => {
     assistantMap.set(assistant.id, assistant);
@@ -353,9 +292,11 @@ async function getAssistantList(request, reply) {
 }
 async function getAssistant(request, reply) {
   const { id } = request.params;
-  let assistant = assistantStore.getAssistantById(id);
+  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const assistantStore = storeLattice.store;
+  let assistant = await assistantStore.getAssistantById(id);
   if (!assistant) {
-    const agentConfigs = await (0, import_core2.getAllAgentConfigs)();
+    const agentConfigs = await (0, import_core3.getAllAgentConfigs)();
     const agentConfig = agentConfigs.find((config) => config.key === id);
     if (agentConfig) {
       assistant = convertAgentConfigToAssistant(agentConfig);
@@ -388,7 +329,9 @@ async function createAssistant(request, reply) {
     });
   }
   const id = (0, import_crypto.randomUUID)();
-  const newAssistant = assistantStore.createAssistant(id, data);
+  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const newAssistant = await assistantStore.createAssistant(id, data);
   return reply.status(201).send({
     success: true,
     message: "Successfully created assistant",
@@ -398,7 +341,7 @@ async function createAssistant(request, reply) {
 async function updateAssistant(request, reply) {
   const { id } = request.params;
   const updates = request.body;
-  const agentConfigs = await (0, import_core2.getAllAgentConfigs)();
+  const agentConfigs = await (0, import_core3.getAllAgentConfigs)();
   const isCodeConfigured = agentConfigs.some((config) => config.key === id);
   if (isCodeConfigured) {
     return reply.status(403).send({
@@ -406,13 +349,16 @@ async function updateAssistant(request, reply) {
       message: "Cannot update code-configured assistant. Only stored assistants can be updated."
     });
   }
-  if (!assistantStore.hasAssistant(id)) {
+  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found"
     });
   }
-  const updatedAssistant = assistantStore.updateAssistant(id, updates);
+  const updatedAssistant = await assistantStore.updateAssistant(id, updates);
   if (!updatedAssistant) {
     return reply.status(500).send({
       success: false,
@@ -427,7 +373,7 @@ async function updateAssistant(request, reply) {
 }
 async function deleteAssistant(request, reply) {
   const { id } = request.params;
-  const agentConfigs = await (0, import_core2.getAllAgentConfigs)();
+  const agentConfigs = await (0, import_core3.getAllAgentConfigs)();
   const isCodeConfigured = agentConfigs.some((config) => config.key === id);
   if (isCodeConfigured) {
     return reply.status(403).send({
@@ -435,13 +381,16 @@ async function deleteAssistant(request, reply) {
       message: "Cannot delete code-configured assistant. Only stored assistants can be deleted."
     });
   }
-  if (!assistantStore.hasAssistant(id)) {
+  const storeLattice = (0, import_core2.getStoreLattice)("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found"
     });
   }
-  const deleted = assistantStore.deleteAssistant(id);
+  const deleted = await assistantStore.deleteAssistant(id);
   if (!deleted) {
     return reply.status(500).send({
       success: false,
@@ -736,7 +685,7 @@ var clearMemory = async (request, reply) => {
 };
 
 // src/controllers/agent_task.ts
-var import_core3 = require("@axiom-lattice/core");
+var import_core4 = require("@axiom-lattice/core");
 var triggerAgentTask = async (request, reply) => {
   try {
     const { assistant_id, thread_id, input, command } = request.body;
@@ -755,7 +704,7 @@ var triggerAgentTask = async (request, reply) => {
       });
       return;
     }
-    const agentManager = import_core3.AgentManager.getInstance();
+    const agentManager = import_core4.AgentManager.getInstance();
     const result = await agentManager.callAgentInQueue({
       assistant_id,
       thread_id,
@@ -774,102 +723,14 @@ var triggerAgentTask = async (request, reply) => {
   }
 };
 
-// src/stores/thread_store.ts
-var ThreadStore = class {
-  constructor() {
-    // Map<assistantId, Map<threadId, Thread>>
-    this.threads = /* @__PURE__ */ new Map();
-  }
-  /**
-   * Get all threads for a specific assistant
-   */
-  getThreadsByAssistantId(assistantId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return [];
-    }
-    return Array.from(assistantThreads.values());
-  }
-  /**
-   * Get a thread by ID for a specific assistant
-   */
-  getThreadById(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return void 0;
-    }
-    return assistantThreads.get(threadId);
-  }
-  /**
-   * Create a new thread for an assistant
-   */
-  createThread(assistantId, threadId, data) {
-    const now = /* @__PURE__ */ new Date();
-    const thread = {
-      id: threadId,
-      assistantId,
-      metadata: data.metadata || {},
-      createdAt: now,
-      updatedAt: now
-    };
-    if (!this.threads.has(assistantId)) {
-      this.threads.set(assistantId, /* @__PURE__ */ new Map());
-    }
-    const assistantThreads = this.threads.get(assistantId);
-    assistantThreads.set(threadId, thread);
-    return thread;
-  }
-  /**
-   * Update an existing thread
-   */
-  updateThread(assistantId, threadId, updates) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return null;
-    }
-    const existing = assistantThreads.get(threadId);
-    if (!existing) {
-      return null;
-    }
-    const updated = {
-      ...existing,
-      metadata: {
-        ...existing.metadata,
-        ...updates.metadata || {}
-      },
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    assistantThreads.set(threadId, updated);
-    return updated;
-  }
-  /**
-   * Delete a thread by ID
-   */
-  deleteThread(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.delete(threadId);
-  }
-  /**
-   * Check if thread exists
-   */
-  hasThread(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.has(threadId);
-  }
-};
-var threadStore = new ThreadStore();
-
 // src/controllers/threads.ts
+var import_core5 = require("@axiom-lattice/core");
 var import_crypto2 = require("crypto");
 async function getThreadList(request, reply) {
   const { assistantId } = request.params;
-  const threads = threadStore.getThreadsByAssistantId(assistantId);
+  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const threadStore = storeLattice.store;
+  const threads = await threadStore.getThreadsByAssistantId(assistantId);
   return {
     success: true,
     message: "Successfully retrieved thread list",
@@ -881,7 +742,9 @@ async function getThreadList(request, reply) {
 }
 async function getThread(request, reply) {
   const { assistantId, threadId } = request.params;
-  const thread = threadStore.getThreadById(assistantId, threadId);
+  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const threadStore = storeLattice.store;
+  const thread = await threadStore.getThreadById(assistantId, threadId);
   if (!thread) {
     return reply.status(404).send({
       success: false,
@@ -898,7 +761,9 @@ async function createThread(request, reply) {
   const { assistantId } = request.params;
   const data = request.body;
   const threadId = (0, import_crypto2.randomUUID)();
-  const newThread = threadStore.createThread(assistantId, threadId, data);
+  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const threadStore = storeLattice.store;
+  const newThread = await threadStore.createThread(assistantId, threadId, data);
   return reply.status(201).send({
     success: true,
     message: "Successfully created thread",
@@ -908,13 +773,16 @@ async function createThread(request, reply) {
 async function updateThread(request, reply) {
   const { assistantId, threadId } = request.params;
   const updates = request.body;
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const threadStore = storeLattice.store;
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found"
     });
   }
-  const updatedThread = threadStore.updateThread(
+  const updatedThread = await threadStore.updateThread(
     assistantId,
     threadId,
     updates
@@ -933,13 +801,16 @@ async function updateThread(request, reply) {
 }
 async function deleteThread(request, reply) {
   const { assistantId, threadId } = request.params;
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const storeLattice = (0, import_core5.getStoreLattice)("default", "thread");
+  const threadStore = storeLattice.store;
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found"
     });
   }
-  const deleted = threadStore.deleteThread(assistantId, threadId);
+  const deleted = await threadStore.deleteThread(assistantId, threadId);
   if (!deleted) {
     return reply.status(500).send({
       success: false,
@@ -1025,7 +896,7 @@ var ConfigService = class {
 var configService = new ConfigService();
 
 // src/services/queue_service.ts
-var import_core4 = require("@axiom-lattice/core");
+var import_core6 = require("@axiom-lattice/core");
 var import_protocols = require("@axiom-lattice/protocols");
 var import_queue_redis = require("@axiom-lattice/queue-redis");
 var DEFAULT_QUEUE_KEY = "default";
@@ -1044,8 +915,8 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     } : void 0
   };
-  if (import_core4.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
-    import_core4.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
+  if (import_core6.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+    import_core6.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
   }
   let client;
   if (type === "redis") {
@@ -1054,13 +925,13 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     });
   }
-  (0, import_core4.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
+  (0, import_core6.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
 };
 var getQueueService = () => {
-  if (!import_core4.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+  if (!import_core6.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
     setQueueServiceType(queueServiceType);
   }
-  return (0, import_core4.getQueueLattice)(DEFAULT_QUEUE_KEY);
+  return (0, import_core6.getQueueLattice)(DEFAULT_QUEUE_KEY);
 };
 var popAgentTaskFromQueue = async () => {
   const queue = getQueueService();
@@ -1141,10 +1012,10 @@ async function getConfig(request, reply) {
 }
 
 // src/controllers/models.ts
-var import_core5 = require("@axiom-lattice/core");
+var import_core7 = require("@axiom-lattice/core");
 async function getModels(request, reply) {
   try {
-    const allLattices = import_core5.modelLatticeManager.getAllLattices();
+    const allLattices = import_core7.modelLatticeManager.getAllLattices();
     const models = allLattices.map((lattice) => {
       const config = lattice.client.config || {};
       return {
@@ -1194,8 +1065,8 @@ async function updateModels(request, reply) {
         continue;
       }
       try {
-        if (import_core5.modelLatticeManager.hasLattice(modelConfig.key)) {
-          import_core5.modelLatticeManager.removeLattice(modelConfig.key);
+        if (import_core7.modelLatticeManager.hasLattice(modelConfig.key)) {
+          import_core7.modelLatticeManager.removeLattice(modelConfig.key);
         }
         const llmConfig = {
           provider: modelConfig.provider,
@@ -1208,7 +1079,7 @@ async function updateModels(request, reply) {
           timeout: modelConfig.timeout,
           maxRetries: modelConfig.maxRetries
         };
-        (0, import_core5.registerModelLattice)(modelConfig.key, llmConfig);
+        (0, import_core7.registerModelLattice)(modelConfig.key, llmConfig);
         registeredModels.push(modelConfig.key);
       } catch (error) {
         errors.push(
@@ -1242,6 +1113,29 @@ async function updateModels(request, reply) {
   }
 }
 
+// src/controllers/health.ts
+async function getHealth(request, reply) {
+  try {
+    const healthStatus = {
+      status: "healthy",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      uptime: process.uptime(),
+      service: "lattice-gateway",
+      version: process.env.npm_package_version || "1.0.0"
+    };
+    return reply.send({
+      success: true,
+      data: healthStatus
+    });
+  } catch (error) {
+    return reply.status(500).send({
+      success: false,
+      status: "unhealthy",
+      error: error.message || "Health check failed"
+    });
+  }
+}
+
 // src/schemas/index.ts
 var getAllMemoryItemsSchema = {
   description: "Get all memory items for an assistant thread",
@@ -1481,6 +1375,37 @@ var getConfigSchema = {
     }
   }
 };
+var getHealthSchema = {
+  description: "Health check endpoint for monitoring and load balancer checks",
+  tags: ["Health"],
+  summary: "Health Check",
+  response: {
+    200: {
+      type: "object",
+      properties: {
+        success: { type: "boolean" },
+        data: {
+          type: "object",
+          properties: {
+            status: { type: "string", enum: ["healthy", "unhealthy"] },
+            timestamp: { type: "string" },
+            uptime: { type: "number" },
+            service: { type: "string" },
+            version: { type: "string" }
+          }
+        }
+      }
+    },
+    500: {
+      type: "object",
+      properties: {
+        success: { type: "boolean" },
+        status: { type: "string" },
+        error: { type: "string" }
+      }
+    }
+  }
+};
 
 // src/routes/index.ts
 var registerLatticeRoutes = (app2) => {
@@ -1557,6 +1482,11 @@ var registerLatticeRoutes = (app2) => {
   );
   app2.get("/api/models", getModels);
   app2.put("/api/models", updateModels);
+  app2.get(
+    "/health",
+    { schema: getHealthSchema },
+    getHealth
+  );
 };
 
 // src/logger/Logger.ts
@@ -1760,7 +1690,7 @@ var configureSwagger = async (app2, customSwaggerConfig, customSwaggerUiConfig)
 };
 
 // src/services/agent_task_consumer.ts
-var import_core6 = require("@axiom-lattice/core");
+var import_core8 = require("@axiom-lattice/core");
 var handleAgentTask = async (taskRequest, retryCount = 0) => {
   const {
     assistant_id,
@@ -1824,7 +1754,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       }
       if (callback_event) {
         const state = await agent_state({ assistant_id, thread_id });
-        import_core6.eventBus.publish(callback_event, {
+        import_core8.eventBus.publish(callback_event, {
           success: true,
           state,
           config: { assistant_id, thread_id, tenant_id }
@@ -1838,7 +1768,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       await response.text();
       if (callback_event) {
         const state = await agent_state({ assistant_id, thread_id });
-        import_core6.eventBus.publish(callback_event, {
+        import_core8.eventBus.publish(callback_event, {
           success: true,
           state,
           config: { assistant_id, thread_id, tenant_id }
@@ -1865,7 +1795,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       return handleAgentTask(taskRequest, nextRetryCount);
     }
     if (callback_event) {
-      import_core6.eventBus.publish(callback_event, {
+      import_core8.eventBus.publish(callback_event, {
         success: false,
         error: error instanceof Error ? error.message : String(error),
         config: { assistant_id, thread_id, tenant_id }
@@ -1903,7 +1833,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
    * 初始化事件监听和队列轮询
    */
   initialize() {
-    import_core6.eventBus.subscribe(import_core6.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
+    import_core8.eventBus.subscribe(import_core8.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
     this.startPollingQueue();
     console.log("Agent\u4EFB\u52A1\u6D88\u8D39\u8005\u5DF2\u542F\u52A8\u5E76\u76D1\u542C\u4EFB\u52A1\u4E8B\u4EF6\u548C\u961F\u5217");
   }
@@ -2022,7 +1952,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
     handleAgentTask(taskRequest).catch((error) => {
       console.error("\u5904\u7406Agent\u4EFB\u52A1\u65F6\u53D1\u751F\u672A\u6355\u83B7\u7684\u9519\u8BEF:", error);
       if (taskRequest.callback_event) {
-        import_core6.eventBus.publish(taskRequest.callback_event, {
+        import_core8.eventBus.publish(taskRequest.callback_event, {
           success: false,
           error: error instanceof Error ? error.message : String(error),
           config: {
@@ -2050,8 +1980,10 @@ var logger = new Logger({
   name: "fastify-server"
 });
 var app = (0, import_fastify.default)({
-  logger: false
+  logger: false,
   // 禁用内置日志记录器
+  bodyLimit: Number(process.env.BODY_LIMIT) || 50 * 1024 * 1024
+  // Default 50MB, configurable via BODY_LIMIT env var
 });
 app.addHook("onRequest", (request, reply, done) => {
   const context = {