sst 3.1.12 → 3.1.13

package/dist/vector/index.js

@@ -1,6 +1,12 @@
-import { LambdaClient, InvokeCommand, } from "@aws-sdk/client-lambda";
+import { AwsClient } from "aws4fetch";
 import { Resource } from "../resource.js";
-const lambda = new LambdaClient();
+const client = new AwsClient({
+    accessKeyId: process.env.AWS_ACCESS_KEY_ID,
+    secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
+    sessionToken: process.env.AWS_SESSION_TOKEN,
+    region: process.env.AWS_REGION,
+});
+const endpoint = `https://lambda.${process.env.AWS_REGION}.amazonaws.com/2015-03-31`;
 /**
  * Create a client to interact with the Vector database.
  * @example
@@ -31,38 +37,76 @@ const lambda = new LambdaClient();
 export function VectorClient(name) {
     return {
         put: async (event) => {
-            const ret = await lambda.send(new InvokeCommand({
-                // @ts-expect-error
-                FunctionName: Resource[name].putFunction,
-                Payload: JSON.stringify(event),
-            }));
-            parsePayload(ret, "Failed to store into the vector db");
+            await invokeFunction(
+            // @ts-expect-error
+            Resource[name].putFunction, JSON.stringify(event), "Failed to store into the vector db");
         },
         query: async (event) => {
-            const ret = await lambda.send(new InvokeCommand({
-                // @ts-expect-error
-                FunctionName: Resource[name].queryFunction,
-                Payload: JSON.stringify(event),
-            }));
-            return parsePayload(ret, "Failed to query the vector db");
+            return await invokeFunction(
+            // @ts-expect-error
+            Resource[name].queryFunction, JSON.stringify(event), "Failed to query the vector db");
         },
         remove: async (event) => {
-            const ret = await lambda.send(new InvokeCommand({
-                // @ts-expect-error
-                FunctionName: Resource[name].removeFunction,
-                Payload: JSON.stringify(event),
-            }));
-            parsePayload(ret, "Failed to remove from the vector db");
+            await invokeFunction(
+            // @ts-expect-error
+            Resource[name].removeFunction, JSON.stringify(event), "Failed to remove from the vector db");
         },
     };
 }
-function parsePayload(output, message) {
-    const payload = JSON.parse(Buffer.from(output.Payload).toString());
-    // Set cause to the payload so that it can be logged in CloudWatch
-    if (output.FunctionError) {
-        const e = new Error(message);
-        e.cause = payload;
-        throw e;
+async function invokeFunction(functionName, body, errorMessage, attempts = 0) {
+    try {
+        const response = await client.fetch(`${endpoint}/functions/${functionName}/invocations`, {
+            method: "POST",
+            headers: { Accept: "application/json" },
+            body,
+        });
+        // success
+        if (response.status === 200 || response.status === 201) {
+            if (response.headers.get("content-length") === "0")
+                return undefined;
+            const text = await response.text();
+            try {
+                return JSON.parse(text);
+            }
+            catch (e) {
+                throw new Error(`Failed to parse JSON response: ${text}`);
+            }
+        }
+        // error
+        const error = new Error();
+        const text = await response.text();
+        try {
+            const json = JSON.parse(text);
+            error.name = json.Error?.Code;
+            error.message = json.Error?.Message ?? json.message ?? text;
+        }
+        catch (e) {
+            error.message = text;
+        }
+        error.name = error.name ?? response.headers.get("x-amzn-ErrorType");
+        // @ts-expect-error
+        error.requestID = response.headers.get("x-amzn-RequestId");
+        // @ts-expect-error
+        error.statusCode = response.status;
+        throw error;
+    }
+    catch (e) {
+        let isRetryable = false;
+        // AWS throttling errors => retry
+        if ([
+            "ThrottlingException",
+            "Throttling",
+            "TooManyRequestsException",
+            "OperationAbortedException",
+            "TimeoutError",
+            "NetworkingError",
+        ].includes(e.name)) {
+            isRetryable = true;
+        }
+        if (!isRetryable)
+            throw e;
+        // retry
+        await new Promise((resolve) => setTimeout(resolve, 1.5 ** attempts * 100 * Math.random()));
+        return await invokeFunction(functionName, body, errorMessage, attempts + 1);
     }
-    return payload;
 }

package/package.json

@@ -3,7 +3,7 @@
   "name": "sst",
   "type": "module",
   "sideEffects": false,
-  "version": "3.1.12",
+  "version": "3.1.13",
   "main": "./dist/index.js",
   "exports": {
     ".": "./dist/index.js",
@@ -47,14 +47,13 @@
     }
   },
   "optionalDependencies": {
-    "sst-linux-x64": "3.1.12",
-    "sst-linux-x86": "3.1.12",
-    "sst-darwin-x64": "3.1.12",
-    "sst-linux-arm64": "3.1.12",
-    "sst-darwin-arm64": "3.1.12"
+    "sst-linux-x64": "3.1.13",
+    "sst-linux-x86": "3.1.13",
+    "sst-linux-arm64": "3.1.13",
+    "sst-darwin-x64": "3.1.13",
+    "sst-darwin-arm64": "3.1.13"
   },
   "dependencies": {
-    "@aws-sdk/client-lambda": "3.478.0",
     "aws4fetch": "^1.0.18",
     "jose": "5.2.3",
     "openid-client": "5.6.4"