fix(ui): created context metadata (#1302)

Author: PetrBulanek

apps/beeai-ui/src/modules/platform-context/api/index.ts

@@ -13,6 +13,7 @@ import type {
   ListContextHistoryParams,
   ListContextsParams,
   MatchModelProvidersParams,
+  UpdateContextParams,
 } from './types';
 
 export async function createContext(body: CreateContextParams) {
@@ -27,8 +28,17 @@ export async function listContexts({ query }: ListContextsParams) {
   return ensureData(response);
 }
 
-export async function deleteContext({ context_id }: DeleteContextParams) {
-  const response = await api.DELETE('/api/v1/contexts/{context_id}', { params: { path: { context_id } } });
+export async function updateContext({ context_id, metadata }: UpdateContextParams) {
+  const response = await api.PUT('/api/v1/contexts/{context_id}', {
+    body: { metadata },
+    params: { path: { context_id } },
+  });
+
+  return ensureData(response);
+}
+
+export async function deleteContext(path: DeleteContextParams) {
+  const response = await api.DELETE('/api/v1/contexts/{context_id}', { params: { path } });
 
   return ensureData(response);
 }

apps/beeai-ui/src/modules/platform-context/api/mutations/useUpdateContext.ts

@@ -0,0 +1,20 @@
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { useMutation } from '@tanstack/react-query';
+
+import { updateContext } from '..';
+import { contextKeys } from '../keys';
+
+export function useUpdateContext() {
+  const mutation = useMutation({
+    mutationFn: updateContext,
+    meta: {
+      invalidates: [contextKeys.lists()],
+    },
+  });
+
+  return mutation;
+}

apps/beeai-ui/src/modules/platform-context/api/types.ts

@@ -7,17 +7,21 @@ import type { ApiPath, ApiQuery, ApiRequest, ApiResponse } from '#@types/utils.t
 
 import type { ContextMetadata, ModelCapability } from '../types';
 
-type CreateContextRequest = ApiRequest<'/api/v1/contexts'>;
-export type CreateContextParams = CreateContextRequest & {
-  metadata: Required<Pick<ContextMetadata, 'agent_name' | 'provider_id'>>;
-};
+export type CreateContextRequest = ApiRequest<'/api/v1/contexts'>;
+export type CreateContextParams = CreateContextRequest;
 export type CreateContextResponse = ApiResponse<'/api/v1/contexts', 'post', 'application/json', 201>;
 
 export type ListContextsQuery = ApiQuery<'/api/v1/contexts'>;
 export type ListContextsResponse = ApiResponse<'/api/v1/contexts'>;
 export type ListContextsParams = { query?: ListContextsQuery };
 
-type DeleteContextPath = ApiPath<'/api/v1/contexts/{context_id}'>;
+export type UpdateContextPath = ApiPath<'/api/v1/contexts/{context_id}', 'put'>;
+export type UpdateContextRequest = ApiRequest<'/api/v1/contexts/{context_id}', 'put'> & {
+  metadata: Pick<ContextMetadata, 'agent_name' | 'provider_id'>;
+};
+export type UpdateContextParams = UpdateContextPath & UpdateContextRequest;
+
+export type DeleteContextPath = ApiPath<'/api/v1/contexts/{context_id}'>;
 export type DeleteContextParams = DeleteContextPath;
 
 export type ListContextHistoryQuery = ApiQuery<'/api/v1/contexts/{context_id}/history'>;

apps/beeai-ui/src/modules/platform-context/contexts/PlatformContextProvider.tsx

@@ -9,6 +9,7 @@ import { useParamsFromUrl } from '#hooks/useParamsFromUrl.ts';
 import type { Agent } from '#modules/agents/api/types.ts';
 
 import { useCreateContext } from '../api/mutations/useCreateContext';
+import { useUpdateContext } from '../api/mutations/useUpdateContext';
 import type { ListContextHistoryResponse } from '../api/types';
 import { PlatformContext } from './platform-context';
 
@@ -24,7 +25,7 @@ export function PlatformContextProvider({ history, children }: PropsWithChildren
     setContextId(urlContextId ?? null);
   }, [urlContextId]);
 
-  const { mutateAsync: createContextMutate } = useCreateContext({
+  const { mutateAsync: createContext } = useCreateContext({
     onSuccess: (context) => {
       if (!context) {
         throw new Error(`Context has not been created`);
@@ -34,20 +35,27 @@ export function PlatformContextProvider({ history, children }: PropsWithChildren
     },
   });
 
+  const { mutateAsync: updateContext } = useUpdateContext();
+
   const resetContext = useCallback(() => {
     setContextId(null);
   }, []);
 
-  const createContext = useCallback(
+  const updateContextWithAgentMetadata = useCallback(
     async (agent: Agent) => {
-      await createContextMutate({
+      if (!contextId) {
+        return;
+      }
+
+      await updateContext({
+        context_id: contextId,
         metadata: {
           agent_name: agent.name ?? '',
           provider_id: agent.provider.id ?? '',
         },
       });
     },
-    [createContextMutate],
+    [contextId, updateContext],
   );
 
   const getContextId = useCallback(() => {
@@ -66,6 +74,7 @@ export function PlatformContextProvider({ history, children }: PropsWithChildren
         createContext,
         getContextId,
         resetContext,
+        updateContextWithAgentMetadata,
       }}
     >
       {children}

apps/beeai-ui/src/modules/platform-context/contexts/platform-context.ts

@@ -3,12 +3,13 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
+import type { UseMutateAsyncFunction } from '@tanstack/react-query';
 import { createContext } from 'react';
 
 import type { Agent } from '#modules/agents/api/types.ts';
 import type { ContextId } from '#modules/tasks/api/types.ts';
 
-import type { ListContextHistoryResponse } from '../api/types';
+import type { CreateContextParams, CreateContextResponse, ListContextHistoryResponse } from '../api/types';
 
 export type ContextToken = {
   token: string;
@@ -21,7 +22,8 @@ interface PlatformContextValue {
 
   getContextId: () => ContextId;
   resetContext: () => void;
-  createContext: (agent: Agent) => Promise<void>;
+  createContext: UseMutateAsyncFunction<CreateContextResponse | undefined, Error, CreateContextParams>;
+  updateContextWithAgentMetadata: (agent: Agent) => Promise<void>;
 }
 
 export const PlatformContext = createContext<PlatformContextValue | null>(null);

apps/beeai-ui/src/modules/platform-context/hooks/useEnsurePlatformContext.ts

@@ -22,7 +22,7 @@ export function useEnsurePlatformContext(agent?: Agent) {
 
     const ensureContext = async () => {
       if (!contextId) {
-        await createContext(agent);
+        await createContext({});
       }
     };
 

apps/beeai-ui/src/modules/runs/contexts/agent-run/AgentRunProvider.tsx

@@ -106,7 +106,7 @@ function AgentRunProvider({
   const pendingSubscription = useRef<() => void>(undefined);
   const pendingRun = useRef<ChatRun>(undefined);
 
-  const { contextId, getContextId } = usePlatformContext();
+  const { contextId, getContextId, updateContextWithAgentMetadata } = usePlatformContext();
   const { getFullfilments } = useAgentDemands();
   const { files, clearFiles } = useFileUpload();
 
@@ -203,6 +203,7 @@ function AgentRunProvider({
       }
 
       checkPendingRun();
+      updateContextWithAgentMetadata(agent);
       setIsPending(true);
       setStats({ startTime: Date.now() });
 
@@ -295,6 +296,8 @@ function AgentRunProvider({
       getFullfilments,
       setMessages,
       agentClient,
+      agent,
+      updateContextWithAgentMetadata,
       updateCurrentAgentMessage,
       handleError,
     ],