fix: use class property for default schema keys

Author: ranst91

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

@@ -122,7 +122,7 @@ export class LangGraphAgent extends AbstractAgent {
   activeRun?: RunMetadata;
   // @ts-expect-error no need to initialize subscriber right now
   subscriber: Subscriber<ProcessedEvents>;
-  constantSchemaKeys: string[] = ["messages", "tools"];
+  constantSchemaKeys: string[] = DEFAULT_SCHEMA_KEYS;
 
   constructor(config: LangGraphAgentConfig) {
     super(config);
@@ -766,7 +766,7 @@ export class LangGraphAgent extends AbstractAgent {
     const schemaKeys = this.activeRun!.schemaKeys!;
     // Do not emit state keys that are not part of the output schema
     if (schemaKeys?.output) {
-      state = filterObjectBySchemaKeys(state, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.output]);
+      state = filterObjectBySchemaKeys(state, [...this.constantSchemaKeys, ...schemaKeys.output]);
     }
     // return state
     return state;
@@ -813,7 +813,7 @@ export class LangGraphAgent extends AbstractAgent {
       if (cfg.configurable) {
         filteredConfigurable = schemaKeys?.config
           ? filterObjectBySchemaKeys(cfg?.configurable, [
-            ...DEFAULT_SCHEMA_KEYS,
+            ...this.constantSchemaKeys,
             ...(schemaKeys?.config ?? []),
           ])
           : cfg?.configurable;

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

@@ -2,7 +2,7 @@ import { Message as LangGraphMessage } from "@langchain/langgraph-sdk";
 import { State, SchemaKeys, LangGraphReasoning } from "./types";
 import { Message, ToolCall } from "@ag-ui/client";
 
-export const DEFAULT_SCHEMA_KEYS = ["tools"];
+export const DEFAULT_SCHEMA_KEYS = ["messages", "tools"];
 
 export function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {
   return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));