sst 3.0.1-7 → 3.0.1-8

package/dist/index.d.ts

@@ -1 +1,2 @@
 export * from "./resource.js";
+export * from "./vector-client.js";

package/dist/index.js

@@ -1 +1,2 @@
 export * from "./resource.js";
+export * from "./vector-client.js";

package/dist/vector-client.d.ts

@@ -0,0 +1,126 @@
+export type IngestEvent = {
+    /**
+     * The external ID of the event.
+     * If the external ID already exists, the embedding and metadata will be updated.
+     * @example
+     * ```js
+     * {
+     *   externalId: "37043bc3-2166-437d-bbf8-a2238d7a5796"
+     * }
+     */
+    externalId: string;
+    /**
+     * The text used to generate the embedding vector.
+     * @example
+     * ```js
+     * {
+     *   text: "This is an example text.",
+     * }
+     * ```
+     */
+    text: string;
+    /**
+     * The base64 representation of the image used to generate the embedding vector.
+     * @example
+     * ```js
+     * {
+     *   image: await fs.readFile("./file.jpg").toString("base64"),
+     * }
+     * ```
+     */
+    image?: string;
+    /**
+     * Additional metadata for the event in JSON format.
+     * This metadata will be used to filter the retrieval of events.
+     * @example
+     * ```js
+     * {
+     *   metadata: {
+     *     key1: "value1",
+     *     key2: "value2"
+     *   }
+     * }
+     * ```
+     */
+    metadata: any;
+};
+export type RetrieveEvent = {
+    /**
+     * The prompt used to retrieve events.
+     * @example
+     * ```js
+     * {
+     *   prompt: "This is an example text.",
+     * }
+     * ```
+     */
+    prompt: string;
+    /**
+     * The metadata used to filter the retrieval of events.
+     * Only events with metadata that match the provided fields will be returned.
+     * @example
+     * ```js
+     * {
+     *   metadata: {
+     *     type: "movie",
+     *     name: "Spiderman",
+     *   }
+     * }
+     * ```
+     * This will match event with metadata:
+     *  {
+     *    type: "movie",
+     *    name: "Spiderman",
+     *    release: "2001",
+     *  }
+     * But this will not match event with metadata:
+     *  {
+     *    type: "book",
+     *    name: "Spiderman",
+     *    release: "1962",
+     *  }
+     */
+    metadata: any;
+    /**
+     * The threshold of similarity between the prompt and the retrieved events.
+     * Only events with a similarity score higher than the threshold will be returned.
+     * Expected value is between 0 and 1.
+     * - 0 means the prompt and the retrieved events are completely different.
+     * - 1 means the prompt and the retrieved events are identical.
+     * @default 0
+     * @example
+     * ```js
+     * {
+     *   threshold: 0.5,
+     * }
+     * ```
+     */
+    threshold?: number;
+    /**
+     * The number of results to return.
+     * @default 10
+     * @example
+     * ```js
+     * {
+     *   count: 10,
+     * }
+     * ```
+     */
+    count?: number;
+};
+export type RemoveEvent = {
+    /**
+     * The external ID of the event to remove.
+     * @example
+     * ```js
+     * {
+     *   externalId: "37043bc3-2166-437d-bbf8-a2238d7a5796"
+     * }
+     */
+    externalId: string;
+};
+export declare const VectorClient: (name: string) => {
+    ingest: (event: IngestEvent) => Promise<any>;
+    retrieve: (event: RetrieveEvent) => Promise<any>;
+    delete: (event: RemoveEvent) => Promise<any>;
+};

package/dist/vector-client.js

@@ -0,0 +1,38 @@
+import { LambdaClient, InvokeCommand, } from "@aws-sdk/client-lambda";
+import { Resource } from "./resource.js";
+const lambda = new LambdaClient();
+export const VectorClient = (name) => {
+    return {
+        ingest: async (event) => {
+            const ret = await lambda.send(new InvokeCommand({
+                FunctionName: Resource[name].ingestorFunctionName,
+                Payload: JSON.stringify(event),
+            }));
+            return parsePayload(ret, "Failed to ingest into the vector db");
+        },
+        retrieve: async (event) => {
+            const ret = await lambda.send(new InvokeCommand({
+                FunctionName: Resource[name].retrieverFunctionName,
+                Payload: JSON.stringify(event),
+            }));
+            return parsePayload(ret, "Failed to retrieve from the vector db");
+        },
+        delete: async (event) => {
+            const ret = await lambda.send(new InvokeCommand({
+                FunctionName: Resource[name].removerFunctionName,
+                Payload: JSON.stringify(event),
+            }));
+            return parsePayload(ret, "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;
+    }
+    return payload;
+}

package/package.json

@@ -2,7 +2,7 @@
   "$schema": "https://json.schemastore.org/package.json",
   "name": "sst",
   "type": "module",
-  "version": "3.0.1-7",
+  "version": "3.0.1-8",
   "main": "./dist/index.js",
   "exports": {
     ".": "./dist/index.js",
@@ -16,8 +16,12 @@
   "files": [
     "dist"
   ],
+  "dependencies": {
+    "@aws-sdk/client-lambda": "3.478.0"
+  },
   "scripts": {
     "build": "tsc",
+    "dev": "tsc -w",
     "release": "pnpm build && pnpm version prerelease && pnpm publish --no-git-checks --tag=ion --access=public"
   }
 }