feat: support latest langgraph context (ag-ui-protocol#375)

Author: ranst91

typescript-sdk/integrations/langgraph/package.json

@@ -24,7 +24,7 @@
   },
   "dependencies": {
     "@langchain/core": "^0.3.66",
-    "@langchain/langgraph-sdk": "^0.0.105",
+    "@langchain/langgraph-sdk": "^0.1.2",
     "partial-json": "^0.1.7",
     "rxjs": "7.8.1"
   },

typescript-sdk/integrations/langgraph/src/agent.ts

@@ -333,6 +333,10 @@ export class LangGraphAgent extends AbstractAgent {
       streamMode,
       input: payloadInput,
       config: payloadConfig,
+      context: {
+        ...context,
+        ...(payloadConfig?.configurable ?? {}),
+      }
     };
 
     // If there are still outstanding unresolved interrupts, we must force resolution of them before moving forward
@@ -947,11 +951,15 @@ export class LangGraphAgent extends AbstractAgent {
     try {
       const graphSchema = await this.client.assistants.getSchemas(this.assistant!.assistant_id);
       let configSchema = null;
+      let contextSchema: string[] = []
+      if ('context_schema' in graphSchema && graphSchema.context_schema?.properties) {
+        contextSchema = Object.keys(graphSchema.context_schema.properties);
+      }
       if (graphSchema.config_schema?.properties) {
         configSchema = Object.keys(graphSchema.config_schema.properties);
       }
       if (!graphSchema.input_schema?.properties || !graphSchema.output_schema?.properties) {
-        return { config: [], input: null, output: null };
+        return { config: [], input: null, output: null, context: contextSchema };
       }
       const inputSchema = Object.keys(graphSchema.input_schema.properties);
       const outputSchema = Object.keys(graphSchema.output_schema.properties);
@@ -963,10 +971,11 @@ export class LangGraphAgent extends AbstractAgent {
           outputSchema && outputSchema.length
             ? [...outputSchema, ...this.constantSchemaKeys]
             : null,
+        context: contextSchema,
         config: configSchema,
       };
     } catch (e) {
-      return { config: [], input: this.constantSchemaKeys, output: this.constantSchemaKeys };
+      return { config: [], input: this.constantSchemaKeys, output: this.constantSchemaKeys, context: [] };
     }
   }
 

typescript-sdk/integrations/langgraph/src/types.ts

@@ -39,6 +39,7 @@ export interface StateEnrichment {
 export type SchemaKeys = {
   input: string[] | null;
   output: string[] | null;
+  context: string[] | null;
   config: string[] | null;
 } | null;