@axiom-lattice/gateway 2.1.65 → 2.1.66

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/gateway@2.1.65 build /home/runner/work/agentic/agentic/packages/gateway
+> @axiom-lattice/gateway@2.1.66 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,22 +9,22 @@
 CLI Cleaning output folder
 CJS Build start
 ESM Build start
+ESM dist/index.mjs     199.60 KB
+ESM dist/index.mjs.map 433.06 KB
+ESM ⚡️ Build success in 312ms
 [warn] ▲ [WARNING] "import.meta" is not available with the "cjs" output format and will be empty [empty-import-meta]
 
-    src/index.ts:152:33:
-      152 │ const __filename = fileURLToPath(import.meta.url);
+    src/index.ts:171:33:
+      171 │ const __filename = fileURLToPath(import.meta.url);
           ╵                                  ~~~~~~~~~~~
 
   You need to set the output format to "esm" for "import.meta" to work correctly.
 
 
-ESM dist/index.mjs     199.74 KB
-ESM dist/index.mjs.map 432.91 KB
-ESM ⚡️ Build success in 395ms
-CJS dist/index.js     203.07 KB
-CJS dist/index.js.map 432.28 KB
-CJS ⚡️ Build success in 403ms
+CJS dist/index.js     202.91 KB
+CJS dist/index.js.map 432.44 KB
+CJS ⚡️ Build success in 346ms
 DTS Build start
-DTS ⚡️ Build success in 12510ms
+DTS ⚡️ Build success in 12529ms
 DTS dist/index.d.ts  3.85 KB
 DTS dist/index.d.mts 3.85 KB

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @axiom-lattice/gateway
 
+## 2.1.66
+
+### Patch Changes
+
+- a326dd4: add user id
+
 ## 2.1.65
 
 ### Patch Changes

package/dist/index.js

@@ -236,6 +236,11 @@ var getAgentGraph = async (request, reply) => {
 var import_uuid = require("uuid");
 var import_core3 = require("@axiom-lattice/core");
 var import_protocols = require("@axiom-lattice/protocols");
+function getUserId(request) {
+  const authUser = request.user;
+  if (authUser?.id) return authUser.id;
+  return request.headers["x-user-id"] || void 0;
+}
 var createRun = async (request, reply) => {
   try {
     const {
@@ -253,6 +258,8 @@ var createRun = async (request, reply) => {
     const workspace_id = request.headers["x-workspace-id"];
     const project_id = request.headers["x-project-id"];
     const x_request_id = request.headers["x-request-id"] || (0, import_uuid.v4)();
+    const user_id = getUserId(request);
+    const mergedConfig = user_id ? { ...custom_run_config, user_id } : custom_run_config;
     if (!assistant_id) {
       reply.status(400).send({
         success: false,
@@ -266,7 +273,7 @@ var createRun = async (request, reply) => {
       tenant_id,
       workspace_id,
       project_id,
-      custom_run_config
+      custom_run_config: mergedConfig
     });
     if (streaming) {
       reply.hijack();
@@ -281,7 +288,7 @@ var createRun = async (request, reply) => {
         const result = await agent.addMessage({
           input: messageInput,
           command,
-          custom_run_config
+          custom_run_config: mergedConfig
         }, mode);
         const stream = agent.chunkStream(result.messageId, [import_protocols.MessageChunkTypes.MESSAGE_COMPLETED]);
         for await (const chunk of stream) {
@@ -311,7 +318,7 @@ var createRun = async (request, reply) => {
       const result = await agent.invoke({
         input: { message: msg, ...restInputNonStream },
         command,
-        custom_run_config
+        custom_run_config: mergedConfig
       });
       reply.status(200).send({
         success: true,
@@ -5018,35 +5025,31 @@ var AuthController = class {
     }
   }
 };
+function extractUserFromAuthHeader(authHeader) {
+  if (!authHeader?.startsWith("Bearer ")) return void 0;
+  const token = authHeader.substring(7);
+  try {
+    const payload = JSON.parse(atob(token));
+    if (payload.exp && payload.exp < Date.now()) return void 0;
+    return {
+      id: payload.userId,
+      tenantId: payload.tenantId
+    };
+  } catch {
+    return void 0;
+  }
+}
 function registerAuthRoutes(app2, config) {
   const controller = new AuthController(config);
   const authHook = async (request, reply) => {
-    const authHeader = request.headers.authorization;
-    if (!authHeader || !authHeader.startsWith("Bearer ")) {
-      return reply.status(401).send({
-        success: false,
-        error: "Unauthorized - Missing or invalid token"
-      });
-    }
-    const token = authHeader.substring(7);
-    try {
-      const payload = JSON.parse(atob(token));
-      if (payload.exp && payload.exp < Date.now()) {
-        return reply.status(401).send({
-          success: false,
-          error: "Unauthorized - Token expired"
-        });
-      }
-      request.user = {
-        id: payload.userId,
-        tenantId: payload.tenantId
-      };
-    } catch {
+    const user = extractUserFromAuthHeader(request.headers.authorization);
+    if (!user) {
       return reply.status(401).send({
         success: false,
-        error: "Unauthorized - Invalid token"
+        error: "Unauthorized"
       });
     }
+    request.user = user;
   };
   app2.post("/api/auth/register", controller.register.bind(controller));
   app2.post("/api/auth/login", controller.login.bind(controller));
@@ -6279,16 +6282,33 @@ app.addContentTypeParser("application/json", { parseAs: "string" }, function(req
     done(err, void 0);
   }
 });
+var getHeaderValue = (header) => {
+  if (Array.isArray(header)) {
+    return header[0];
+  }
+  return header;
+};
+var PUBLIC_ROUTES = ["/api/auth/login", "/api/auth/register", "/health"];
+app.addHook("preHandler", async (request, reply) => {
+  const user = extractUserFromAuthHeader(request.headers.authorization);
+  if (user) {
+    request.user = user;
+    return;
+  }
+  const authRequired = process.env.AUTH_REQUIRED === "true";
+  if (!authRequired) return;
+  if (request.method === "OPTIONS") return;
+  if (PUBLIC_ROUTES.some((r) => request.url === r)) return;
+  return reply.status(401).send({
+    success: false,
+    error: "Unauthorized - Missing or invalid token"
+  });
+});
 app.addHook("onRequest", (request, reply, done) => {
-  const getHeaderValue = (header) => {
-    if (Array.isArray(header)) {
-      return header[0];
-    }
-    return header;
-  };
   const context = {
     "x-tenant-id": getHeaderValue(request.headers["x-tenant-id"]),
-    "x-request-id": getHeaderValue(request.headers["x-request-id"])
+    "x-request-id": getHeaderValue(request.headers["x-request-id"]),
+    "x-user-id": getHeaderValue(request.headers["x-user-id"])
   };
   if (loggerLattice.updateContext) {
     loggerLattice.updateContext(context);
@@ -6296,15 +6316,10 @@ app.addHook("onRequest", (request, reply, done) => {
   done();
 });
 app.addHook("onResponse", (request, reply, done) => {
-  const getHeaderValue = (header) => {
-    if (Array.isArray(header)) {
-      return header[0];
-    }
-    return header;
-  };
   const context = {
     "x-tenant-id": getHeaderValue(request.headers["x-tenant-id"]),
-    "x-request-id": getHeaderValue(request.headers["x-request-id"])
+    "x-request-id": getHeaderValue(request.headers["x-request-id"]),
+    "x-user-id": getHeaderValue(request.headers["x-user-id"])
   };
   done();
 });
@@ -6329,15 +6344,10 @@ app.register(import_static.default, {
   prefix: "/"
 });
 app.setErrorHandler((error, request, reply) => {
-  const getHeaderValue = (header) => {
-    if (Array.isArray(header)) {
-      return header[0];
-    }
-    return header;
-  };
   const context = {
     "x-tenant-id": getHeaderValue(request.headers["x-tenant-id"]),
-    "x-request-id": getHeaderValue(request.headers["x-request-id"])
+    "x-request-id": getHeaderValue(request.headers["x-request-id"]),
+    "x-user-id": getHeaderValue(request.headers["x-user-id"])
   };
   logger.error(
     `\u8BF7\u6C42\u9519\u8BEF: ${request.method} ${request.url} error:${error.message}`,
@@ -6395,21 +6405,6 @@ var start = async (config) => {
       import_core29.sandboxLatticeManager.registerLattice("default", getConfiguredSandboxProvider());
       logger.info("Registered sandbox manager from env configuration");
     }
-    if (process.env.DATABASE_URL) {
-      try {
-        const { PostgreSQLWorkflowTrackingStore } = await import("@axiom-lattice/pg-stores");
-        const pgStore = new PostgreSQLWorkflowTrackingStore({
-          poolConfig: process.env.DATABASE_URL
-        });
-        if (import_core29.storeLatticeManager.hasLattice("default", "workflowTracking")) {
-          import_core29.storeLatticeManager.removeLattice("default", "workflowTracking");
-        }
-        import_core29.storeLatticeManager.registerLattice("default", "workflowTracking", pgStore);
-        logger.info("Workflow tracking store switched to PostgreSQL");
-      } catch (error) {
-        logger.warn("Failed to switch workflow tracking to PostgreSQL, keeping in-memory: " + (error instanceof Error ? error.message : String(error)));
-      }
-    }
     const target_port = config?.port || Number(process.env.PORT) || 4001;
     await app.listen({ port: target_port, host: "0.0.0.0" });
     logger.info(`Lattice Gateway is running on port: ${target_port}`);