@axiom-lattice/gateway 2.1.17 → 2.1.19

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/gateway@2.1.17 build /home/runner/work/agentic/agentic/packages/gateway
+> @axiom-lattice/gateway@2.1.19 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     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
+ESM dist/index.mjs     61.30 KB
+ESM dist/index.mjs.map 132.19 KB
+ESM ⚡️ Build success in 147ms
+CJS dist/index.js     63.84 KB
+CJS dist/index.js.map 132.29 KB
+CJS ⚡️ Build success in 148ms
 DTS Build start
-DTS ⚡️ Build success in 7551ms
+DTS ⚡️ Build success in 7807ms
 DTS dist/index.d.ts  3.32 KB
 DTS dist/index.d.mts 3.32 KB

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @axiom-lattice/gateway
 
+## 2.1.19
+
+### Patch Changes
+
+- d43ea0b: add schedule task
+- Updated dependencies [d43ea0b]
+  - @axiom-lattice/protocols@2.1.8
+  - @axiom-lattice/core@2.1.14
+  - @axiom-lattice/queue-redis@1.0.7
+
+## 2.1.18
+
+### Patch Changes
+
+- Updated dependencies [e26be9a]
+  - @axiom-lattice/protocols@2.1.7
+  - @axiom-lattice/core@2.1.13
+  - @axiom-lattice/queue-redis@1.0.6
+
 ## 2.1.17
 
 ### Patch Changes

package/dist/index.js

@@ -62,13 +62,15 @@ async function agent_invoke({
   command,
   run_id
 }) {
-  const runnable_agent = (0, import_core.getAgentClient)(assistant_id);
+  const agentLattice = (0, import_core.getAgentLattice)(assistant_id);
+  const runnable_agent = agentLattice?.client;
   const { message, ...rest } = input;
   const humanMessage = new import_messages.HumanMessage(message || "");
   const messages = [humanMessage];
   if (!runnable_agent) {
     throw new Error(`Agent ${assistant_id} not found`);
   }
+  const runConfig = agentLattice?.config?.runConfig || {};
   const result = await runnable_agent.invoke(
     command ? new import_langgraph.Command(command) : { ...rest, messages, "x-tenant-id": tenant_id },
     {
@@ -77,7 +79,9 @@ async function agent_invoke({
         run_id: run_id || (0, import_uuid.v4)(),
         "x-tenant-id": tenant_id,
         "x-request-id": run_id,
-        "x-thread-id": thread_id
+        "x-thread-id": thread_id,
+        runConfig
+        // Inject runConfig for tools to access
       },
       recursionLimit: 200
     }
@@ -100,6 +104,7 @@ async function agent_stream({
   run_id
 }) {
   const runnable_agent = (0, import_core.getAgentClient)(assistant_id);
+  const agentLattice = (0, import_core.getAgentLattice)(assistant_id);
   const { message, ...rest } = input;
   let messages = [];
   if (!command) {
@@ -107,6 +112,7 @@ async function agent_stream({
     messages = [humanMessage];
   }
   const chunkBuffer = getOrCreateChunkBuffer();
+  const runConfig = agentLattice?.config?.runConfig || {};
   try {
     if (!runnable_agent) {
       throw new Error(`Agent ${assistant_id} not found`);
@@ -123,7 +129,9 @@ async function agent_stream({
           run_id: run_id || (0, import_uuid.v4)(),
           "x-tenant-id": tenant_id,
           "x-request-id": run_id,
-          "x-thread-id": thread_id
+          "x-thread-id": thread_id,
+          runConfig
+          // Inject runConfig for tools to access
         },
         streamMode: ["updates", "messages"],
         subgraphs: false,
@@ -823,6 +831,201 @@ async function deleteThread(request, reply) {
   };
 }
 
+// src/controllers/schedules.ts
+var import_core6 = require("@axiom-lattice/core");
+function getScheduleLattice() {
+  const keys = import_core6.scheduleLatticeManager.getLatticeKeys();
+  if (keys.length === 0) {
+    return null;
+  }
+  if (import_core6.scheduleLatticeManager.hasLattice("default")) {
+    return import_core6.scheduleLatticeManager.getScheduleLattice("default");
+  }
+  return import_core6.scheduleLatticeManager.getScheduleLattice(keys[0]);
+}
+async function getThreadSchedules(request, reply) {
+  const { assistantId, threadId } = request.params;
+  const { status, limit, offset } = request.query;
+  try {
+    const scheduleLattice = getScheduleLattice();
+    if (!scheduleLattice) {
+      return {
+        success: true,
+        message: "No schedule lattice configured",
+        data: {
+          records: [],
+          total: 0
+        }
+      };
+    }
+    const storage = scheduleLattice.client.getStorage();
+    if (!storage) {
+      return {
+        success: true,
+        message: "No schedule storage configured",
+        data: {
+          records: [],
+          total: 0
+        }
+      };
+    }
+    const filters = {
+      threadId,
+      assistantId
+    };
+    if (status) {
+      filters.status = status;
+    }
+    if (limit) {
+      filters.limit = parseInt(limit, 10);
+    }
+    if (offset) {
+      filters.offset = parseInt(offset, 10);
+    }
+    const tasks = await storage.getAllTasks(filters);
+    const total = await storage.countTasks({
+      threadId,
+      assistantId,
+      status
+    });
+    return {
+      success: true,
+      message: "Successfully retrieved scheduled tasks",
+      data: {
+        records: tasks,
+        total
+      }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get thread schedules");
+    return reply.status(500).send({
+      success: false,
+      message: "Failed to retrieve scheduled tasks",
+      data: {
+        records: [],
+        total: 0
+      }
+    });
+  }
+}
+async function getScheduledTask(request, reply) {
+  const { taskId } = request.params;
+  try {
+    const scheduleLattice = getScheduleLattice();
+    if (!scheduleLattice) {
+      return reply.status(404).send({
+        success: false,
+        message: "No schedule lattice configured"
+      });
+    }
+    const task = await scheduleLattice.client.getTask(taskId);
+    if (!task) {
+      return reply.status(404).send({
+        success: false,
+        message: "Scheduled task not found"
+      });
+    }
+    return {
+      success: true,
+      message: "Successfully retrieved scheduled task",
+      data: task
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get scheduled task");
+    return reply.status(500).send({
+      success: false,
+      message: "Failed to retrieve scheduled task"
+    });
+  }
+}
+async function cancelScheduledTask(request, reply) {
+  const { taskId } = request.params;
+  try {
+    const scheduleLattice = getScheduleLattice();
+    if (!scheduleLattice) {
+      return reply.status(404).send({
+        success: false,
+        message: "No schedule lattice configured"
+      });
+    }
+    const result = await scheduleLattice.client.cancel(taskId);
+    if (!result) {
+      return reply.status(404).send({
+        success: false,
+        message: "Scheduled task not found or already cancelled"
+      });
+    }
+    return {
+      success: true,
+      message: "Successfully cancelled scheduled task"
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to cancel scheduled task");
+    return reply.status(500).send({
+      success: false,
+      message: "Failed to cancel scheduled task"
+    });
+  }
+}
+async function pauseScheduledTask(request, reply) {
+  const { taskId } = request.params;
+  try {
+    const scheduleLattice = getScheduleLattice();
+    if (!scheduleLattice) {
+      return reply.status(404).send({
+        success: false,
+        message: "No schedule lattice configured"
+      });
+    }
+    const result = await scheduleLattice.client.pause(taskId);
+    if (!result) {
+      return reply.status(404).send({
+        success: false,
+        message: "Scheduled task not found or cannot be paused"
+      });
+    }
+    return {
+      success: true,
+      message: "Successfully paused scheduled task"
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to pause scheduled task");
+    return reply.status(500).send({
+      success: false,
+      message: "Failed to pause scheduled task"
+    });
+  }
+}
+async function resumeScheduledTask(request, reply) {
+  const { taskId } = request.params;
+  try {
+    const scheduleLattice = getScheduleLattice();
+    if (!scheduleLattice) {
+      return reply.status(404).send({
+        success: false,
+        message: "No schedule lattice configured"
+      });
+    }
+    const result = await scheduleLattice.client.resume(taskId);
+    if (!result) {
+      return reply.status(404).send({
+        success: false,
+        message: "Scheduled task not found or cannot be resumed"
+      });
+    }
+    return {
+      success: true,
+      message: "Successfully resumed scheduled task"
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to resume scheduled task");
+    return reply.status(500).send({
+      success: false,
+      message: "Failed to resume scheduled task"
+    });
+  }
+}
+
 // src/config.ts
 var ConfigService = class {
   constructor() {
@@ -896,7 +1099,7 @@ var ConfigService = class {
 var configService = new ConfigService();
 
 // src/services/queue_service.ts
-var import_core6 = require("@axiom-lattice/core");
+var import_core7 = 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";
@@ -915,8 +1118,8 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     } : void 0
   };
-  if (import_core6.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
-    import_core6.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
+  if (import_core7.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+    import_core7.queueLatticeManager.removeLattice(DEFAULT_QUEUE_KEY);
   }
   let client;
   if (type === "redis") {
@@ -925,13 +1128,13 @@ var setQueueServiceType = (type) => {
       redisPassword: process.env.REDIS_PASSWORD
     });
   }
-  (0, import_core6.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
+  (0, import_core7.registerQueueLattice)(DEFAULT_QUEUE_KEY, config, client);
 };
 var getQueueService = () => {
-  if (!import_core6.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
+  if (!import_core7.queueLatticeManager.hasLattice(DEFAULT_QUEUE_KEY)) {
     setQueueServiceType(queueServiceType);
   }
-  return (0, import_core6.getQueueLattice)(DEFAULT_QUEUE_KEY);
+  return (0, import_core7.getQueueLattice)(DEFAULT_QUEUE_KEY);
 };
 var popAgentTaskFromQueue = async () => {
   const queue = getQueueService();
@@ -1012,10 +1215,10 @@ async function getConfig(request, reply) {
 }
 
 // src/controllers/models.ts
-var import_core7 = require("@axiom-lattice/core");
+var import_core8 = require("@axiom-lattice/core");
 async function getModels(request, reply) {
   try {
-    const allLattices = import_core7.modelLatticeManager.getAllLattices();
+    const allLattices = import_core8.modelLatticeManager.getAllLattices();
     const models = allLattices.map((lattice) => {
       const config = lattice.client.config || {};
       return {
@@ -1065,8 +1268,8 @@ async function updateModels(request, reply) {
         continue;
       }
       try {
-        if (import_core7.modelLatticeManager.hasLattice(modelConfig.key)) {
-          import_core7.modelLatticeManager.removeLattice(modelConfig.key);
+        if (import_core8.modelLatticeManager.hasLattice(modelConfig.key)) {
+          import_core8.modelLatticeManager.removeLattice(modelConfig.key);
         }
         const llmConfig = {
           provider: modelConfig.provider,
@@ -1079,7 +1282,7 @@ async function updateModels(request, reply) {
           timeout: modelConfig.timeout,
           maxRetries: modelConfig.maxRetries
         };
-        (0, import_core7.registerModelLattice)(modelConfig.key, llmConfig);
+        (0, import_core8.registerModelLattice)(modelConfig.key, llmConfig);
         registeredModels.push(modelConfig.key);
       } catch (error) {
         errors.push(
@@ -1487,6 +1690,14 @@ var registerLatticeRoutes = (app2) => {
     { schema: getHealthSchema },
     getHealth
   );
+  app2.get(
+    "/api/assistants/:assistantId/threads/:threadId/schedules",
+    getThreadSchedules
+  );
+  app2.get("/api/schedules/:taskId", getScheduledTask);
+  app2.post("/api/schedules/:taskId/cancel", cancelScheduledTask);
+  app2.post("/api/schedules/:taskId/pause", pauseScheduledTask);
+  app2.post("/api/schedules/:taskId/resume", resumeScheduledTask);
 };
 
 // src/logger/Logger.ts
@@ -1690,7 +1901,7 @@ var configureSwagger = async (app2, customSwaggerConfig, customSwaggerUiConfig)
 };
 
 // src/services/agent_task_consumer.ts
-var import_core8 = require("@axiom-lattice/core");
+var import_core9 = require("@axiom-lattice/core");
 var handleAgentTask = async (taskRequest, retryCount = 0) => {
   const {
     assistant_id,
@@ -1754,7 +1965,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       }
       if (callback_event) {
         const state = await agent_state({ assistant_id, thread_id });
-        import_core8.eventBus.publish(callback_event, {
+        import_core9.eventBus.publish(callback_event, {
           success: true,
           state,
           config: { assistant_id, thread_id, tenant_id }
@@ -1768,7 +1979,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       await response.text();
       if (callback_event) {
         const state = await agent_state({ assistant_id, thread_id });
-        import_core8.eventBus.publish(callback_event, {
+        import_core9.eventBus.publish(callback_event, {
           success: true,
           state,
           config: { assistant_id, thread_id, tenant_id }
@@ -1795,7 +2006,7 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
       return handleAgentTask(taskRequest, nextRetryCount);
     }
     if (callback_event) {
-      import_core8.eventBus.publish(callback_event, {
+      import_core9.eventBus.publish(callback_event, {
         success: false,
         error: error instanceof Error ? error.message : String(error),
         config: { assistant_id, thread_id, tenant_id }
@@ -1833,7 +2044,7 @@ var _AgentTaskConsumer = class _AgentTaskConsumer {
    * 初始化事件监听和队列轮询
    */
   initialize() {
-    import_core8.eventBus.subscribe(import_core8.AGENT_TASK_EVENT, this.trigger_agent_task.bind(this));
+    import_core9.eventBus.subscribe(import_core9.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");
   }
@@ -1952,7 +2163,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_core8.eventBus.publish(taskRequest.callback_event, {
+        import_core9.eventBus.publish(taskRequest.callback_event, {
           success: false,
           error: error instanceof Error ? error.message : String(error),
           config: {