feat(sdk-node): Run Context (#97)

inferablehq · Nov 13, 2024 · 86affdb · 86affdb
1 parent 08075f1
commit 86affdb
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/sdk-node/src/contract.ts b/sdk-node/src/contract.ts
@@ -429,6 +429,9 @@ export const definition = {
             .optional(),
         })
         .optional(),
+      context: anyObject
+        .optional()
+        .describe("Additional context to propogate to all calls in the run"),
       template: z
         .object({
           id: z.string().describe("DEPRECATED"),
@@ -1400,7 +1403,8 @@ export const definition = {
           id: z.string(),
           function: z.string(),
           input: z.any(),
-          customerAuthContext: z.any().nullable(),
+          authContext: z.any().nullable(),
+          runContext: z.any().nullable(),
         }),
       ),
     },

diff --git a/sdk-node/src/service.ts b/sdk-node/src/service.ts
@@ -16,7 +16,8 @@ type CallMessage = {
   id: string;
   function: string;
   input?: unknown;
-  customerAuthContext?: unknown;
+  authContext?: unknown;
+  runContext?: string;
 };
 
 export class Service {
@@ -270,7 +271,8 @@ export class Service {
     const result = await executeFn(
       registration.func,
       [args, {
-        customerAuthContext: call.customerAuthContext,
+        authContext: call.authContext,
+        runContext: call.runContext,
       }],
     );
 

diff --git a/sdk-node/src/types.ts b/sdk-node/src/types.ts
@@ -5,7 +5,8 @@ import { FunctionConfigSchema } from "./contract";
  * Context object which is passed to function calls
  */
 export type ContextInput = {
-  customerAuthContext?: unknown;
+  authContext?: unknown;
+  runContext?: unknown;
 }
 
 export type FunctionConfig = z.infer<typeof FunctionConfigSchema>;