update

inferablehq · Nov 13, 2024 · 193e77b · 193e77b
1 parent c313b7c
commit 193e77b
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 20 deletions.
diff --git a/sdk-node/package-lock.json b/sdk-node/package-lock.json
diff --git a/sdk-node/package.json b/sdk-node/package.json
@@ -27,7 +27,7 @@
   "devDependencies": {
     "@babel/preset-env": "^7.26.0",
     "@babel/preset-typescript": "^7.22.11",
-    "@types/jest": "^29.5.4",
+    "@types/jest": "^29.5.14",
     "@types/node-os-utils": "^1.3.4",
     "@typescript-eslint/eslint-plugin": "^8.12.2",
     "@typescript-eslint/parser": "^8.12.2",

diff --git a/sdk-node/src/Inferable.test.ts b/sdk-node/src/Inferable.test.ts
@@ -294,25 +294,22 @@ describe("Inferable SDK End to End Test", () => {
     it("should be able to call the api directly", async () => {
       const client = inferableInstance();
 
-      expect(
-        await client.api.createStructuredOutput({
-          params: {
-            clusterId: TEST_CLUSTER_ID,
-          },
-          body: {
-            prompt: "What is the capital of France?",
-            modelId: "claude-3-5-sonnet",
-            resultSchema: {
-              type: "object",
-              properties: {
-                capital: { type: "string" },
-              },
-            },
+      const result = await client.api.createStructuredOutput<{
+        capital: string;
+      }>({
+        prompt: "What is the capital of France?",
+        modelId: "claude-3-5-sonnet",
+        resultSchema: {
+          type: "object",
+          properties: {
+            capital: { type: "string" },
           },
-        }),
-      ).toEqual({
+        },
+      });
+
+      expect(result).toMatchObject({
         success: true,
-        result: {
+        data: {
           capital: "Paris",
         },
       });

diff --git a/sdk-node/src/Inferable.ts b/sdk-node/src/Inferable.ts
@@ -491,7 +491,22 @@ export class Inferable {
   }
 
   public get api() {
-    return this.client;
+    return {
+      createStructuredOutput: async <T>(
+        input: Parameters<typeof this.client.createStructuredOutput>[0]["body"],
+      ) =>
+        this.client
+          .createStructuredOutput({
+            params: {
+              clusterId: await this.getClusterId(),
+            },
+            body: input,
+          })
+          .then((r) => r.body) as Promise<{
+          success: boolean;
+          data?: T;
+        }>,
+    };
   }
 
   private async getClusterId() {