chore: creatingh react context for a2a

tomkis · jezekra1 · commit e07f8ea30179 · 2026-01-06T14:29:31.000+01:00
Signed-off-by: Tomas Weiss &lt;tomas.weiss2@gmail.com&gt;
diff --git a/apps/agentstack-sdk-ts/src/client/a2a/create-authenticated-fetch.ts b/apps/agentstack-sdk-ts/src/client/a2a/create-authenticated-fetch.ts
@@ -3,10 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-export function createAuthenticatedFetch(
-  token: string,
-  baseFetch?: typeof fetch,
-): typeof fetch {
+export function createAuthenticatedFetch(token: string, baseFetch?: typeof fetch): typeof fetch {
   const fetchImpl = baseFetch ?? (typeof globalThis.fetch !== 'undefined' ? globalThis.fetch : undefined);
 
   if (!fetchImpl) {
diff --git a/apps/agentstack-sdk-ts/src/client/a2a/extensions/handle-agent-card.ts b/apps/agentstack-sdk-ts/src/client/a2a/extensions/handle-agent-card.ts
@@ -33,6 +33,9 @@ export interface Fulfillments {
   secrets: (demand: SecretDemands) => Promise<SecretFulfillments>;
   form: (demand: FormDemands) => Promise<FormFulfillments>;
   oauthRedirectUri: () => string | null;
+  /**
+   * @deprecated - keeping this for backwards compatibility, context token is now passed via A2A client headers
+   */
   getContextToken: () => ContextToken;
 }
 
diff --git a/apps/agentstack-ui/src/modules/platform-context/api/index.ts b/apps/agentstack-ui/src/modules/platform-context/api/index.ts
@@ -54,7 +54,6 @@ export async function matchProviders(matchProvidersParams: MatchProvidersParams)
 }
 
 export async function createContextToken(createContextTokenParams: CreateContextTokenParams) {
-  console.log(createContextTokenParams);
   const result = await agentstackClient.createContextToken(createContextTokenParams);
   return result.token;
 }
diff --git a/apps/agentstack-ui/src/modules/platform-context/api/keys.ts b/apps/agentstack-ui/src/modules/platform-context/api/keys.ts
@@ -12,4 +12,6 @@ export const contextKeys = {
   histories: () => [...contextKeys.all(), 'history'] as const,
   history: ({ contextId, query = {} }: ListContextHistoryParams) =>
     [...contextKeys.histories(), contextId, query] as const,
+  tokens: () => [...contextKeys.all(), 'token'] as const,
+  token: (contextId: string, providerId: string) => [...contextKeys.tokens(), contextId, providerId] as const,
 };
diff --git a/apps/agentstack-ui/src/modules/platform-context/api/queries/useContextToken.ts b/apps/agentstack-ui/src/modules/platform-context/api/queries/useContextToken.ts
@@ -0,0 +1,44 @@
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { useQuery } from '@tanstack/react-query';
+import type { ContextToken } from 'agentstack-sdk';
+
+import { useApp } from '#contexts/App/index.ts';
+import type { Agent } from '#modules/agents/api/types.ts';
+
+import { usePlatformContext } from '../../contexts';
+import { createContextToken } from '..';
+import { contextKeys } from '../keys';
+
+export function useContextToken(agent: Agent) {
+  const {
+    config: { contextTokenPermissions },
+  } = useApp();
+  const { contextId } = usePlatformContext();
+
+  return useQuery<ContextToken>({
+    queryKey: contextKeys.token(contextId ?? '', agent.provider.id),
+    queryFn: async () => {
+      if (!contextId) {
+        throw new Error('Context ID is not set.');
+      }
+
+      const token = await createContextToken({
+        contextId,
+        contextPermissions: contextTokenPermissions.grant_context_permissions ?? {},
+        globalPermissions: contextTokenPermissions.grant_global_permissions ?? {},
+      });
+
+      if (!token) {
+        throw new Error('Could not generate context token');
+      }
+
+      return token;
+    },
+    enabled: !!contextId,
+    staleTime: Infinity,
+  });
+}
diff --git a/apps/agentstack-ui/src/modules/platform-context/constants.ts b/apps/agentstack-ui/src/modules/platform-context/constants.ts
@@ -15,7 +15,7 @@ export const contextTokenPermissionsDefaults: DeepRequired<ContextTokenPermissio
     llm: ['*'],
     embeddings: ['*'],
     model_providers: [],
-    a2a_proxy: [],
+    a2a_proxy: ['*'],
     providers: [],
     provider_variables: [],
     contexts: [],
diff --git a/apps/agentstack-ui/src/modules/runs/api/queries/useBuildA2AClient.ts b/apps/agentstack-ui/src/modules/runs/api/queries/useBuildA2AClient.ts
@@ -4,6 +4,7 @@
  */
 
 import { useQuery } from '@tanstack/react-query';
+
 import { buildA2AClient, type CreateA2AClientParams } from '#api/a2a/client.ts';
 
 import { runKeys } from '../keys';
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/A2AClientProvider.tsx b/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/A2AClientProvider.tsx
@@ -0,0 +1,43 @@
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+'use client';
+import type { PropsWithChildren } from 'react';
+import { useMemo } from 'react';
+
+import type { Agent } from '#modules/agents/api/types.ts';
+import { useContextToken } from '#modules/platform-context/api/queries/useContextToken.ts';
+import { useBuildA2AClient } from '#modules/runs/api/queries/useBuildA2AClient.ts';
+
+import { A2AClientContext } from './a2a-client-context';
+
+interface Props {
+  agent: Agent;
+}
+
+export function A2AClientProvider({ agent, children }: PropsWithChildren<Props>) {
+  const { data: contextToken } = useContextToken(agent);
+  const { agentClient } = useBuildA2AClient({
+    providerId: agent.provider.id,
+    authToken: contextToken,
+  });
+
+  const contextValue = useMemo(() => {
+    if (!contextToken || !agentClient) {
+      return null;
+    }
+
+    return {
+      contextToken,
+      agentClient,
+    };
+  }, [contextToken, agentClient]);
+
+  if (!contextValue) {
+    return null;
+  }
+
+  return <A2AClientContext.Provider value={contextValue}>{children}</A2AClientContext.Provider>;
+}
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/a2a-client-context.ts b/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/a2a-client-context.ts
@@ -0,0 +1,16 @@
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+'use client';
+import type { ContextToken } from 'agentstack-sdk';
+import { createContext } from 'react';
+
+import type { AgentA2AClient } from '#api/a2a/types.ts';
+
+export const A2AClientContext = createContext<A2AClientContextValue | null>(null);
+
+export interface A2AClientContextValue {
+  contextToken: ContextToken;
+  agentClient: AgentA2AClient;
+}
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/index.ts b/apps/agentstack-ui/src/modules/runs/contexts/a2a-client/index.ts
@@ -0,0 +1,20 @@
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { use } from 'react';
+
+import { A2AClientContext } from './a2a-client-context';
+
+export function useA2AClient() {
+  const context = use(A2AClientContext);
+
+  if (!context) {
+    throw new Error('useA2AClient must be used within A2AClientProvider');
+  }
+
+  return context;
+}
+
+export { A2AClientProvider } from './A2AClientProvider';
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/agent-demands/AgentDemandsProvider.tsx b/apps/agentstack-ui/src/modules/runs/contexts/agent-demands/AgentDemandsProvider.tsx
@@ -3,32 +3,22 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import type { ContextToken } from 'agentstack-sdk';
 import { type AgentSettings, type FormFulfillments, ModelCapability } from 'agentstack-sdk';
 import { type PropsWithChildren, useCallback, useRef, useState } from 'react';
 
-import type { AgentA2AClient } from '#api/a2a/types.ts';
 import { useListConnectors } from '#modules/connectors/api/queries/useListConnectors.ts';
 import type { RunFormValues } from '#modules/form/types.ts';
 import { useMatchProviders } from '#modules/platform-context/api/mutations/useMatchProviders.ts';
 import { getSettingsDemandsDefaultValues } from '#modules/runs/settings/utils.ts';
 
+import { useA2AClient } from '../a2a-client';
 import { useAgentSecrets } from '../agent-secrets';
 import type { FulfillmentsContext } from './agent-demands-context';
 import { AgentDemandsContext } from './agent-demands-context';
 import { buildFulfillments } from './build-fulfillments';
 
-interface Props<UIGenericPart> {
-  agentClient: AgentA2AClient<UIGenericPart>;
-  /** @deprecated Context token should be passed via A2A client, not fulfillments */
-  contextToken: ContextToken;
-}
-
-export function AgentDemandsProvider<UIGenericPart>({
-  agentClient,
-  contextToken,
-  children,
-}: PropsWithChildren<Props<UIGenericPart>>) {
+export function AgentDemandsProvider({ children }: PropsWithChildren) {
+  const { agentClient, contextToken } = useA2AClient();
   const { demandedSecrets } = useAgentSecrets();
 
   const [selectedEmbeddingProviders, setSelectedEmbeddingProviders] = useState<Record<string, string>>({});
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/agent-run/AgentRunProvider.tsx b/apps/agentstack-ui/src/modules/runs/contexts/agent-run/AgentRunProvider.tsx
@@ -5,16 +5,14 @@
 
 'use client';
 import { useQueryClient } from '@tanstack/react-query';
-import type { ContextToken } from 'agentstack-sdk';
 import { TaskStatusUpdateType } from 'agentstack-sdk';
 import type { PropsWithChildren } from 'react';
-import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+import { useCallback, useMemo, useRef, useState } from 'react';
 import { v4 as uuid } from 'uuid';
 
-import type { AgentA2AClient, ChatRun } from '#api/a2a/types.ts';
+import type { ChatRun } from '#api/a2a/types.ts';
 import { createTextPart } from '#api/a2a/utils.ts';
 import { getErrorCode } from '#api/utils.ts';
-import { useApp } from '#contexts/App/index.ts';
 import { useHandleError } from '#hooks/useHandleError.ts';
 import type { Agent } from '#modules/agents/api/types.ts';
 import { CanvasProvider } from '#modules/canvas/contexts/CanvasProvider.tsx';
@@ -30,17 +28,16 @@ import type { UIAgentMessage, UIMessageForm, UIUserMessage } from '#modules/mess
 import { UIMessagePartKind, UIMessageStatus } from '#modules/messages/types.ts';
 import { addMessagePart, isAgentMessage } from '#modules/messages/utils.ts';
 import { contextKeys } from '#modules/platform-context/api/keys.ts';
-import { useCreateContextToken } from '#modules/platform-context/api/mutations/useCreateContextToken.ts';
 import { usePlatformContext } from '#modules/platform-context/contexts/index.ts';
 import { useEnsurePlatformContext } from '#modules/platform-context/hooks/useEnsurePlatformContext.ts';
-import { useBuildA2AClient } from '#modules/runs/api/queries/useBuildA2AClient.ts';
 import { useStartOAuth } from '#modules/runs/hooks/useStartOAuth.ts';
 import type { RunStats } from '#modules/runs/types.ts';
 import { SourcesProvider } from '#modules/sources/contexts/SourcesProvider.tsx';
 import { getMessagesSourcesMap } from '#modules/sources/utils.ts';
 import type { TaskId } from '#modules/tasks/api/types.ts';
 import { isNotNull } from '#utils/helpers.ts';
 
+import { A2AClientProvider, useA2AClient } from '../a2a-client';
 import { useAgentDemands } from '../agent-demands';
 import type { FulfillmentsContext } from '../agent-demands/agent-demands-context';
 import { AgentDemandsProvider } from '../agent-demands/AgentDemandsProvider';
@@ -54,70 +51,28 @@ interface Props {
 
 export function AgentRunProviders({ agent, children }: PropsWithChildren<Props>) {
   useEnsurePlatformContext(agent);
-  const {
-    config: { contextTokenPermissions },
-  } = useApp();
-
-  const { contextId } = usePlatformContext();
-  const { mutateAsync: createContextToken } = useCreateContextToken();
-  const [contextToken, setContextToken] = useState<ContextToken | null>(null);
-
-  useEffect(() => {
-    const createToken = async () => {
-      if (contextId === null) {
-        // todo uncaught error
-        throw new Error('Illegal State - Context ID is not set.');
-      }
-
-      const token = await createContextToken({
-        contextId,
-        contextPermissions: contextTokenPermissions.grant_context_permissions ?? {},
-        globalPermissions: {
-          ...(contextTokenPermissions.grant_global_permissions ?? {}),
-          a2a_proxy: [...(contextTokenPermissions.grant_global_permissions?.a2a_proxy ?? []), agent.provider.id],
-        },
-      });
-      if (!token) {
-        throw new Error('Could not generate context token');
-      }
-      setContextToken(token);
-    };
-    createToken();
-  }, [contextId, contextTokenPermissions, createContextToken, agent.provider.id]);
-
-  const { agentClient } = useBuildA2AClient({
-    providerId: agent.provider.id,
-    authToken: contextToken,
-  });
-
-  if (!agentClient || !contextToken) {
-    return null;
-  }
 
   return (
-    <AgentSecretsProvider agent={agent} agentClient={agentClient}>
-      <AgentDemandsProvider agentClient={agentClient} contextToken={contextToken}>
-        <FileUploadProvider allowedContentTypes={agent.defaultInputModes}>
-          <MessagesProvider>
-            <CanvasProvider>
-              <AgentRunProvider agent={agent} agentClient={agentClient}>
-                {children}
-              </AgentRunProvider>
-            </CanvasProvider>
-          </MessagesProvider>
-        </FileUploadProvider>
-      </AgentDemandsProvider>
-    </AgentSecretsProvider>
+    <A2AClientProvider agent={agent}>
+      <AgentSecretsProvider agent={agent}>
+        <AgentDemandsProvider>
+          <FileUploadProvider allowedContentTypes={agent.defaultInputModes}>
+            <MessagesProvider>
+              <CanvasProvider>
+                <AgentRunProvider agent={agent}>{children}</AgentRunProvider>
+              </CanvasProvider>
+            </MessagesProvider>
+          </FileUploadProvider>
+        </AgentDemandsProvider>
+      </AgentSecretsProvider>
+    </A2AClientProvider>
   );
 }
 
-interface AgentRunProviderProps extends Props {
-  agentClient?: AgentA2AClient;
-}
-
-function AgentRunProvider({ agent, agentClient, children }: PropsWithChildren<AgentRunProviderProps>) {
+function AgentRunProvider({ agent, children }: PropsWithChildren<Props>) {
   const queryClient = useQueryClient();
   const errorHandler = useHandleError();
+  const { agentClient } = useA2AClient();
 
   const { messages, getMessages, setMessages } = useMessages();
 
diff --git a/apps/agentstack-ui/src/modules/runs/contexts/agent-secrets/AgentSecretsProvider.tsx b/apps/agentstack-ui/src/modules/runs/contexts/agent-secrets/AgentSecretsProvider.tsx
@@ -8,17 +8,16 @@ import { useCallback, useMemo } from 'react';
 import { useLocalStorage } from 'usehooks-ts';
 import z from 'zod';
 
-import type { AgentA2AClient } from '#api/a2a/types.ts';
 import type { Agent } from '#modules/agents/api/types.ts';
 import { useListVariables } from '#modules/variables/api/queries/useListVariables.ts';
 import { AGENT_SECRETS_SETTINGS_STORAGE_KEY } from '#utils/constants.ts';
 
+import { useA2AClient } from '../a2a-client';
 import { AgentSecretsContext } from './agent-secrets-context';
 import type { NonReadySecretDemand, ReadySecretDemand } from './types';
 
 interface Props {
   agent: Agent;
-  agentClient?: AgentA2AClient;
 }
 
 const secretsSchema = z.record(
@@ -41,7 +40,8 @@ const secretsLocalStorageOptions = {
   },
 };
 
-export function AgentSecretsProvider({ agent, agentClient, children }: PropsWithChildren<Props>) {
+export function AgentSecretsProvider({ agent, children }: PropsWithChildren<Props>) {
+  const { agentClient } = useA2AClient();
   const [agentSecrets, setAgentSecrets] = useLocalStorage<Secrets>(
     AGENT_SECRETS_SETTINGS_STORAGE_KEY,
     {},
@@ -56,7 +56,7 @@ export function AgentSecretsProvider({ agent, agentClient, children }: PropsWith
   }, [agentSecrets, agent.provider.id]);
 
   const secretDemands = useMemo(() => {
-    return agentClient?.demands.secretDemands ?? null;
+    return agentClient.demands.secretDemands ?? null;
   }, [agentClient]);
 
   const markModalAsSeen = useCallback(() => {
diff --git a/helm/values.yaml b/helm/values.yaml
@@ -70,7 +70,7 @@ contextTokenPermissions:
     llm: ["*"]
     embeddings: ["*"]
     model_providers: []
-    a2a_proxy: []
+    a2a_proxy: ["*"]
     providers: []
     provider_variables: []
     contexts: []

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,9 @@ export interface Fulfillments {`
`33`	`33`	`secrets: (demand: SecretDemands) => Promise<SecretFulfillments>;`
`34`	`34`	`form: (demand: FormDemands) => Promise<FormFulfillments>;`
`35`	`35`	`oauthRedirectUri: () => string \| null;`
	`36`	`+ /**`
	`37`	`+ * @deprecated - keeping this for backwards compatibility, context token is now passed via A2A client headers`
	`38`	`+ */`
`36`	`39`	`getContextToken: () => ContextToken;`
`37`	`40`	`}`
`38`	`41`
Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,6 @@ export async function matchProviders(matchProvidersParams: MatchProvidersParams)`
`54`	`54`	`}`
`55`	`55`
`56`	`56`	`export async function createContextToken(createContextTokenParams: CreateContextTokenParams) {`
`57`		`- console.log(createContextTokenParams);`
`58`	`57`	`const result = await agentstackClient.createContextToken(createContextTokenParams);`
`59`	`58`	`return result.token;`
`60`	`59`	`}`