feat(ts-sdk): add editImage test case with image editing functionality

Author: skeptrunedev

clients/ts-sdk/src/__tests__/constants.ts

@@ -10,8 +10,8 @@ export const TRIEVE = new TrieveSDK({
 
 // export const TRIEVE = new TrieveSDK({
 //   baseUrl: "http://localhost:8090",
-//   organizationId: "310e51a2-4bb4-4317-9661-b5ef3ca85713",
-//   datasetId: "e390d3b2-7c51-4695-a3ec-4487c32220b0",
+//   organizationId: "ad2a06d2-c229-4322-a77c-ed45b2a39d94",
+//   datasetId: "af0893e8-9e1c-4bfd-8ceb-ed9863fc96fb",
 //   apiKey: "admin",
 // });
 

clients/ts-sdk/src/functions/message/message.test.ts

@@ -1,6 +1,6 @@
 import { beforeAll, describe, expect, expectTypeOf } from "vitest";
 import { TrieveSDK } from "../../sdk";
-import { Message } from "../../types.gen";
+import { ImageEditResponse, Message } from "../../types.gen";
 import { TRIEVE } from "../../__tests__/constants";
 import { test } from "../../__tests__/utils";
 
@@ -169,4 +169,29 @@ describe("Message Tests", async () => {
       shirts: false,
     });
   });
+
+  test("editImage", async () => {
+    const prompt = `Create a lovely room design where specifically the countertop is replaced with the Bahia Granite material`;
+    const data = await trieve.editImage({
+      input_images: [
+        {
+          image_src: {
+            url: "https://trieve.b-cdn.net/sdk-tests/102024_KIT_80_PS_4.webp",
+          },
+          file_name: "room_photo",
+        },
+        {
+          image_src: {
+            url: "https://trieve.b-cdn.net/sdk-tests/black-granite-countertop.webp",
+          },
+          file_name: "picture_of_sample",
+        },
+      ],
+      prompt: prompt,
+      quality: "medium",
+      n: 1,
+    });
+
+    expectTypeOf(data).toEqualTypeOf<ImageEditResponse>();
+  });
 });