diff --git a/dev/src/server/adk_api_server.ts b/dev/src/server/adk_api_server.ts index 80a84465..f2eefa96 100644 --- a/dev/src/server/adk_api_server.ts +++ b/dev/src/server/adk_api_server.ts @@ -56,6 +56,17 @@ interface ServerOptions { registerProcessors?: (tracerProvider: TracerProvider) => void; } +function toSessionScopedDelta( + delta: Record | undefined, +): Record | undefined { + if (!delta) return undefined; + return Object.fromEntries( + Object.entries(delta).filter( + ([k]) => !k.startsWith('app:') && !k.startsWith('user:'), + ), + ); +} + export class AdkApiServer { private readonly host: string; private readonly port: number; @@ -727,7 +738,7 @@ export class AdkApiServer { userId, sessionId, newMessage, - stateDelta, + stateDelta: toSessionScopedDelta(stateDelta), })) { events.push(e); } @@ -776,7 +787,7 @@ export class AdkApiServer { runConfig: { streamingMode: streaming ? StreamingMode.SSE : StreamingMode.NONE, }, - stateDelta, + stateDelta: toSessionScopedDelta(stateDelta), })) { res.write(`data: ${JSON.stringify(event)}\n\n`); }