@stack-spot/portal-network 0.149.0-beta.2 → 0.150.0

package/src/api/discovery.ts

@@ -15,7 +15,7 @@ export const servers = {
     generatedServerUrl: "https://discover-discover-core.dev.stackspot.com"
 };
 export type OpportunityResponse = {
-    id?: string;
+    id: string;
     name?: string;
     "number"?: string;
     createdAt?: string;
@@ -25,14 +25,14 @@ export type OpportunityResponse = {
     };
 };
 export type HypothesisResponse = {
-    id?: string;
-    opportunity?: OpportunityResponse;
-    name?: string;
+    id: string;
+    opportunity: OpportunityResponse;
+    name: string;
     description?: string;
-    "number"?: string;
-    status?: "TO_DO" | "PRIORITIZED" | "IN_EXPERIMENTATION" | "VALIDATED" | "DISCARDED" | "CANCELED";
+    "number": string;
+    status: "TO_DO" | "PRIORITIZED" | "IN_EXPERIMENTATION" | "VALIDATED" | "DISCARDED" | "CANCELED";
     createdAt?: string;
-    createdBy?: string;
+    createdBy: string;
     updatedAt?: string;
     updatedBy?: string;
 };
@@ -43,144 +43,188 @@ export type HypothesisUpdateRequest = {
     status?: string;
 };
 export type SortResponse = {
-    direction?: string;
-    nullHandling?: string;
-    ascending?: boolean;
-    property?: string;
-    ignoreCase?: boolean;
+    direction: string;
+    nullHandling: string;
+    ascending: boolean;
+    property: string;
+    ignoreCase: boolean;
 };
 export type PageableResponse = {
-    offset?: number;
-    sort?: SortResponse[];
-    paged?: boolean;
-    pageSize?: number;
-    pageNumber?: number;
-    unpaged?: boolean;
+    offset: number;
+    sort: SortResponse[];
+    paged: boolean;
+    pageSize: number;
+    pageNumber: number;
+    unpaged: boolean;
 };
 export type PageResponseHypothesisResponse = {
-    totalElements?: number;
-    totalPages?: number;
-    first?: boolean;
-    last?: boolean;
-    size?: number;
-    content?: HypothesisResponse[];
-    "number"?: number;
-    sort?: SortResponse[];
-    numberOfElements?: number;
-    pageable?: PageableResponse;
-    empty?: boolean;
+    totalElements: number;
+    totalPages: number;
+    first: boolean;
+    last: boolean;
+    size: number;
+    content: HypothesisResponse[];
+    "number": number;
+    sort: SortResponse[];
+    numberOfElements: number;
+    pageable: PageableResponse;
+    empty: boolean;
 };
 export type HypothesisCreateRequest = {
-    name?: string;
-    description?: string;
-    opportunityId?: string;
+    name: string;
+    description: string;
+    opportunityId: string;
 };
 export type DocumentUploadRequest = {
-    file?: Blob;
-    description?: string;
-    name?: string;
-    typeId?: string;
-    origin?: "AI" | "UPLOAD";
-};
-export type DocumentAttachRequest = {
-    documentIds?: string[];
+    file: Blob;
+    description: string;
+    name: string;
+    typeId: string;
+    origin: "AI" | "UPLOAD";
 };
 export type AgentResponse = {
-    id?: string;
-    label?: string;
+    id: string;
+    label: string;
     image?: string;
-    builtIn?: boolean;
-    slug?: string;
+    builtIn: boolean;
+    slug: string;
     createdAt?: string;
 };
 export type DocumentTypeResponse = {
-    id?: string;
-    typeName?: "PRESS_RELEASE" | "PRODUCT_VISION" | "PRODUCT_REQUIREMENTS" | "MARKET_COMPARISON";
+    id: string;
+    typeName: "PRESS_RELEASE" | "PRODUCT_VISION" | "PRODUCT_REQUIREMENTS" | "MARKET_COMPARISON";
     agent?: AgentResponse;
-    description?: string;
+    description: string;
+    createdAt?: string;
+};
+export type DocumentVersionSummaryResponse = {
+    id: string;
+    name: string;
     createdAt?: string;
+    createdBy?: string;
+};
+export type DocumentUploadResponse = {
+    id: string;
+    name: string;
+    description: string;
+    "type": DocumentTypeResponse;
+    origin: "AI" | "UPLOAD";
+    createdAt?: string;
+    createdBy: string;
+    version: DocumentVersionSummaryResponse;
+};
+export type DocumentAttachRequest = {
+    documentIds: string[];
 };
 export type DocumentDetailContent = {
-    hasPreview?: boolean;
+    hasPreview: boolean;
     content?: string;
 };
-export type DocumentDetailResponse = {
-    id?: string;
-    name?: string;
-    description?: string;
-    "type"?: DocumentTypeResponse;
-    origin?: "AI" | "UPLOAD";
+export type DocumentVersion = {
+    id: string;
+    name: string;
     createdAt?: string;
     createdBy?: string;
-    details?: DocumentDetailContent;
+};
+export type DocumentDetailResponse = {
+    id: string;
+    name: string;
+    description: string;
+    "type": DocumentTypeResponse;
+    origin: "AI" | "UPLOAD";
+    createdAt?: string;
+    createdBy: string;
+    details: DocumentDetailContent;
+    version: DocumentVersion;
 };
 export type DocumentPatchRequest = {
-    file?: Blob;
+    file: Blob;
     description?: string;
     name?: string;
 };
 export type PageResponseOpportunityResponse = {
-    totalElements?: number;
-    totalPages?: number;
-    first?: boolean;
-    last?: boolean;
-    size?: number;
-    content?: OpportunityResponse[];
-    "number"?: number;
-    sort?: SortResponse[];
-    numberOfElements?: number;
-    pageable?: PageableResponse;
-    empty?: boolean;
+    totalElements: number;
+    totalPages: number;
+    first: boolean;
+    last: boolean;
+    size: number;
+    content: OpportunityResponse[];
+    "number": number;
+    sort: SortResponse[];
+    numberOfElements: number;
+    pageable: PageableResponse;
+    empty: boolean;
 };
 export type HypothesisSummaryResponse = {
-    id?: string;
-    name?: string;
-    "number"?: string;
-    status?: "TO_DO" | "PRIORITIZED" | "IN_EXPERIMENTATION" | "VALIDATED" | "DISCARDED" | "CANCELED";
+    id: string;
+    name: string;
+    "number": string;
+    status: "TO_DO" | "PRIORITIZED" | "IN_EXPERIMENTATION" | "VALIDATED" | "DISCARDED" | "CANCELED";
 };
 export type DocumentWithHypothesesResponse = {
-    id?: string;
-    name?: string;
+    id: string;
+    name: string;
     description?: string;
-    origin?: "AI" | "UPLOAD";
-    hypotheses?: HypothesisSummaryResponse[];
+    origin: "AI" | "UPLOAD";
+    hypotheses: HypothesisSummaryResponse[];
     createdAt?: string;
-    createdBy?: string;
+    createdBy: string;
 };
 export type PageResponseDocumentWithHypothesesResponse = {
-    totalElements?: number;
-    totalPages?: number;
-    first?: boolean;
-    last?: boolean;
-    size?: number;
-    content?: DocumentWithHypothesesResponse[];
-    "number"?: number;
-    sort?: SortResponse[];
-    numberOfElements?: number;
-    pageable?: PageableResponse;
-    empty?: boolean;
+    totalElements: number;
+    totalPages: number;
+    first: boolean;
+    last: boolean;
+    size: number;
+    content: DocumentWithHypothesesResponse[];
+    "number": number;
+    sort: SortResponse[];
+    numberOfElements: number;
+    pageable: PageableResponse;
+    empty: boolean;
 };
-export type DocumentSummaryResponse = {
-    id?: string;
-    name?: string;
-    description?: string;
-    origin?: "AI" | "UPLOAD";
-    hasMultipleHypotheses?: boolean;
+export type DocumentVersionResponse = {
+    id: string;
+    documentId: string;
+    versionName: string;
+    s3VersionId: string;
     createdAt?: string;
     createdBy?: string;
 };
+export type PageResponseDocumentVersionResponse = {
+    totalElements: number;
+    totalPages: number;
+    first: boolean;
+    last: boolean;
+    size: number;
+    content: DocumentVersionResponse[];
+    "number": number;
+    sort: SortResponse[];
+    numberOfElements: number;
+    pageable: PageableResponse;
+    empty: boolean;
+};
+export type DocumentSummaryResponse = {
+    id: string;
+    name: string;
+    description: string;
+    origin: "AI" | "UPLOAD";
+    hasMultipleHypotheses: boolean;
+    createdAt?: string;
+    createdBy: string;
+};
 export type PageResponseDocumentSummaryResponse = {
-    totalElements?: number;
-    totalPages?: number;
-    first?: boolean;
-    last?: boolean;
-    size?: number;
-    content?: DocumentSummaryResponse[];
-    "number"?: number;
-    sort?: SortResponse[];
-    numberOfElements?: number;
-    pageable?: PageableResponse;
-    empty?: boolean;
+    totalElements: number;
+    totalPages: number;
+    first: boolean;
+    last: boolean;
+    size: number;
+    content: DocumentSummaryResponse[];
+    "number": number;
+    sort: SortResponse[];
+    numberOfElements: number;
+    pageable: PageableResponse;
+    empty: boolean;
 };
 export function findHypothesisById({ hypothesisId }: {
     hypothesisId: string;
@@ -248,7 +292,10 @@ export function upload({ hypothesisId, documentUploadRequest }: {
     hypothesisId: string;
     documentUploadRequest?: DocumentUploadRequest;
 }, opts?: Oazapfts.RequestOpts) {
-    return oazapfts.ok(oazapfts.fetchText(`/v1/documents/hypotheses/${encodeURIComponent(hypothesisId)}/upload`, oazapfts.multipart({
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: DocumentUploadResponse;
+    }>(`/v1/documents/hypotheses/${encodeURIComponent(hypothesisId)}/upload`, oazapfts.multipart({
         ...opts,
         method: "POST",
         body: documentUploadRequest
@@ -264,13 +311,16 @@ export function attach({ hypothesisId, documentAttachRequest }: {
         body: documentAttachRequest
     })));
 }
-export function getById({ documentId }: {
+export function getById({ documentId, versionId }: {
     documentId: string;
+    versionId?: string;
 }, opts?: Oazapfts.RequestOpts) {
     return oazapfts.ok(oazapfts.fetchJson<{
         status: 200;
         data: DocumentDetailResponse;
-    }>(`/v1/documents/${encodeURIComponent(documentId)}`, {
+    }>(`/v1/documents/${encodeURIComponent(documentId)}${QS.query(QS.explode({
+        versionId
+    }))}`, {
         ...opts
     }));
 }
@@ -334,6 +384,21 @@ export function listAllDocumentsWithHypotheses({ filter, page, size, sort, direc
         ...opts
     }));
 }
+export function listVersions({ documentId, page, size }: {
+    documentId: string;
+    page?: number;
+    size?: number;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: PageResponseDocumentVersionResponse;
+    }>(`/v1/documents/${encodeURIComponent(documentId)}/versions${QS.query(QS.explode({
+        page,
+        size
+    }))}`, {
+        ...opts
+    }));
+}
 export function download({ documentId }: {
     documentId: string;
 }, opts?: Oazapfts.RequestOpts) {
@@ -373,6 +438,26 @@ export function getAll(opts?: Oazapfts.RequestOpts) {
         ...opts
     }));
 }
+export function systemHealth(opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: {
+            [key: string]: object;
+        };
+    }>("/system/health", {
+        ...opts
+    }));
+}
+export function health(opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: {
+            [key: string]: string;
+        };
+    }>("/orchestrator/health", {
+        ...opts
+    }));
+}
 export function deleteV1DocumentsByDocumentIdHypothesesAndHypothesisId({ hypothesisId, documentId }: {
     hypothesisId: string;
     documentId: string;

package/src/client/agent-tools.ts

@@ -1,13 +1,12 @@
 import { HttpError } from '@oazapfts/runtime'
-import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteAgentV1AgentsAgentIdDelete, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listAgentsV1AgentsGet, listToolkitsV1ToolkitsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, updateAgentV1AgentsAgentIdPut, updateToolkitV1ToolkitsToolkitIdPatch, VisibilityLevelEnum } from '../api/agent-tools'
+import { createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listAgentsUsingToolsV1ToolkitsToolkitIdToolsAgentsPost, listToolkitsV1ToolkitsGet, updateToolkitV1ToolkitsToolkitIdPatch } from '../api/agent-tools'
 import apis from '../apis.json'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
+import { FetchEventStream } from '../network/types'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
-import { AgentResponseWithBuiltIn, AgentToolsOpenAPIPreview, AgentVisibilityLevel } from './types'
-import { workspaceAiClient } from './workspace-ai'
-
-const AGENT_DEFAULT_SLUG = 'stk_code_buddy'
+import { StreamedArray } from '../utils/StreamedArray'
+import { AgentToolsOpenAPIPreview } from './types'
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -27,104 +26,8 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   tools = this.query(removeAuthorizationParam(getPublicToolKitsV1BuiltinToolkitGet))
 
   /**
-   * Create agent
-   */
-  createAgent = this.mutation(removeAuthorizationParam(createAgentV1AgentsPost))
-
-  /**
-   * Delete agent
-   */
-  deleteAgent = this.mutation(deleteAgentV1AgentsAgentIdDelete)
-
-  /**
-   * Updates an agent
-   */
-  updateAgent = this.mutation(updateAgentV1AgentsAgentIdPut)
-
-  /**
-  * Favorite an agent
-  */
-  favoriteAgent = this.mutation(addFavoriteV1AgentsAgentIdFavoritePost)
-
-  /**
-  * Publish an agent
-  */
-  publishAgent = this.mutation(publishAgentV1AgentsAgentIdPublishPost)
-
-  /** 
-   * List agents
-   */
-  agents = this.infiniteQuery(removeAuthorizationParam(listAgentsV1AgentsGet))
-
-  /** 
-   * Gets agent by id
-   */
-  agent = this.query(removeAuthorizationParam(getAgentV1AgentsAgentIdGet))
-
-  /** 
-   * Gets agents by ids
-   */
-  agentsByIds = this.query(removeAuthorizationParam(searchAgentsV1AgentsSearchPost))
-
-  /** 
-   * Gets the default agent slug
-   */
-  agentDefaultSlug = AGENT_DEFAULT_SLUG
-
-  /** 
-   * Gets the default agent
+   * Get list of Toolkits
    */
-  agentDefault = this.query({
-    name: 'agentDefault',
-    request: async (signal) => {
-      const agentDefault = await listAgentsV1AgentsGet(
-        { visibility: 'built_in', slug: this.agentDefaultSlug, authorization: '' }, { signal },
-      )
-
-      const agentId = agentDefault.at(0)?.id
-      const agent = agentId ? await this.agent.query({ agentId }) : undefined
-      return agent
-    },
-  })
-
-  /** 
-   * List agents filtered by visibility.
-   */
-  allAgents = this.query({
-    name: 'allAgents',
-    request: async (signal, variables: { visibilities: (AgentVisibilityLevel | 'all')[] }) => {
-      const allVisibilities = ['account', 'built_in', 'favorite', 'personal', 'shared', 'workspace'] as const
-      const visibilities = variables.visibilities.includes('all')
-        ? allVisibilities
-        : variables.visibilities as Array<AgentVisibilityLevelEnum | VisibilityLevelEnum>
-
-      const shouldFetchWorkspaceAgents = visibilities.includes('workspace')
-
-      const workspaceAgentsPromise = shouldFetchWorkspaceAgents
-        ? workspaceAiClient.workspacesContentsByType.query({ contentType: 'agent' })
-        : Promise.resolve([])
-
-      const [workspaceAgents, ...agentsByVisibility] = await Promise.all([
-        workspaceAgentsPromise,
-        ...visibilities.map((visibility) => listAgentsV1AgentsGet({ visibility, authorization: '' }, { signal })),
-      ])
-
-      const workspaceAgentsWithSpaceName = workspaceAgents.flatMap(({ agents, space_name }) =>
-        agents?.map((agent) => ({ ...agent, spaceName: space_name, builtIn: false }))) as AgentResponseWithBuiltIn[]
-
-      const allAgents: AgentResponseWithBuiltIn[] = workspaceAgentsWithSpaceName ?? []
-
-      agentsByVisibility.forEach(agents => allAgents.push(...agents.map(agent => ({
-        ...agent,
-        builtIn: agent?.visibility_level === 'built_in',
-      }))))
-
-      return allAgents
-    },
-  })
-
-  /* Get list of Toolkits
-  */
   toolkits = this.query(removeAuthorizationParam(listToolkitsV1ToolkitsGet))
   /**
    * Get a toolkit by Id
@@ -159,22 +62,19 @@ class AgentToolsClient extends ReactQueryNetworkClient {
    */
   deleteToolFromToolkit = this.mutation(removeAuthorizationParam(deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete))
   /**
-   * Previews an uploaded open api file.
+   * Previews an uploaded open api file as a stream where each array item is an OpenAPI operation.
+   */
+  streamUploadedAPIs = ({ fileUploadIds, signal }: { fileUploadIds: string[], signal?: AbortSignal }) => {
+    const promises: Promise<FetchEventStream>[] = fileUploadIds.map((id) => {
+      const url = `/v1/toolkits/tools/preview/${encodeURIComponent(id)}`
+      return this.stream(url, { signal })
+    })
+    return new StreamedArray<AgentToolsOpenAPIPreview>({ eventsPromises: promises, signal })
+  }
+  /**
+   * Lists the agents using the tools passed as parameter.
    */
-  previewUploadedAPI = this.query({
-    name: 'previewUploadedAPI',
-    request: async (signal, params: { fileUploadId: string }) => {
-      const url = `/v1/toolkits/tools/preview/${encodeURIComponent(params.fileUploadId)}`
-      const events = await this.stream(url, { signal })
-      const result: AgentToolsOpenAPIPreview[] = []
-      for await (const event of events) {
-        if (event.data) {
-          result.push(JSON.parse(event.data))
-        }
-      }
-      return result
-    },
-  })
+  agentsUsingTools = this.query(removeAuthorizationParam(listAgentsUsingToolsV1ToolkitsToolkitIdToolsAgentsPost))
 }
 
 export const agentToolsClient = new AgentToolsClient()

package/src/client/agent.ts

@@ -1,8 +1,16 @@
 import { HttpError } from '@oazapfts/runtime'
-import { defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents, postV1AgentByAgentIdFavorite, postV1AgentsTrial, putV1AgentByAgentId } from '../api/agent'
+import {
+  defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents,
+  postV1AgentByAgentIdFavorite, putV1AgentByAgentId,
+  VisibilityLevel,
+} from '../api/agent'
 import apis from '../apis.json'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
+import { AgentResponseWithBuiltIn, AgentVisibilityLevel } from './types'
+import { workspaceAiClient } from './workspace-ai'
+
+export const isAgentDefault = (agentSlug?: string) => agentSlug === 'stk_code_buddy'
 
 interface AgentError {
   code?: string,
@@ -28,6 +36,57 @@ class AgentClient extends ReactQueryNetworkClient {
     })
   }
 
+  /** 
+   * List agents filtered by visibility.
+   */
+  allAgents = this.query({
+    name: 'allAgents',
+    request: async (signal, variables: { visibilities: AgentVisibilityLevel[] }) => {
+      const visibilities: VisibilityLevel[] = variables.visibilities.includes('ALL')
+        ? ['PERSONAL', 'SHARED', 'WORKSPACE', 'ACCOUNT', 'FAVORITE']
+        : variables.visibilities.filter((visibility) => visibility !== 'BUILT-IN' && visibility !== 'ALL') as VisibilityLevel[]
+
+      const shouldIncludeBuiltInAgent = variables.visibilities.includes('ALL') || variables.visibilities.includes('BUILT-IN')
+      const shouldFetchBuiltInAgent = shouldIncludeBuiltInAgent || variables.visibilities.includes('FAVORITE')
+      const shouldFetchWorkspaceAgents = variables.visibilities.includes('ALL') || variables.visibilities.includes('WORKSPACE')
+
+      const publicAgentsPromise = shouldFetchBuiltInAgent ? getV1PublicAgents({}, { signal }) : Promise.resolve([])
+      const workspaceAgentsPromise = shouldFetchWorkspaceAgents
+        ? workspaceAiClient.workspacesContentsByType.query({ contentType: 'agent' })
+        : Promise.resolve([])
+
+      const [publicAgents, workspaceAgents, ...agentsByVisibility] = await Promise.all([
+        publicAgentsPromise,
+        workspaceAgentsPromise,
+        ...visibilities.map((visibility) => getV1Agents({ visibility }, { signal })),
+      ])
+
+      const workspaceAgentsWithSpaceName = workspaceAgents?.flatMap(({ agents, space_name }) =>
+        agents?.map((agent) => ({
+          ...agent,
+          spaceName: space_name,
+          builtIn: publicAgents?.some(publicAgent => publicAgent.id === agent.id),
+        }))) as AgentResponseWithBuiltIn[]
+
+      const allAgents: AgentResponseWithBuiltIn[] = workspaceAgentsWithSpaceName || []
+
+      if (shouldIncludeBuiltInAgent) {
+        const builtInsAgents = publicAgents?.map((agent) => ({ ...agent, builtIn: true, visibility_level: 'BUILT-IN' }))
+        // @ts-ignore fixme: above, BUILT-IN is not a valid value for the enum VisibilityLevel.
+        allAgents.push(...builtInsAgents)
+      }
+
+      agentsByVisibility.forEach(agents => {
+        allAgents.push(...agents.map(agent => ({
+          ...agent,
+          builtIn: publicAgents?.some(publicAgent => publicAgent.id === agent.id),
+        })))
+      })
+
+      return allAgents
+    },
+  })
+
   /** 
    * Gets an agent by id
    */
@@ -38,6 +97,18 @@ class AgentClient extends ReactQueryNetworkClient {
       : getV1AgentByAgentId({ ...variables }, { signal }),
   })
 
+  /** 
+   * Gets the default agent
+   */
+  agentDefault = this.query({
+    name: 'agentDefault',
+    request: async (signal) => {
+      const publicAgents = await getV1PublicAgents({}, { signal })
+      const agentDefault = publicAgents.find((agent) => isAgentDefault(agent.slug))
+      return agentDefault ? { ...agentDefault, builtIn: true } : undefined
+    },
+  })
+
   /** 
    * List commons agents
    */
@@ -67,11 +138,6 @@ class AgentClient extends ReactQueryNetworkClient {
    * Removes the resource of type Agent from the list of favorites.
    */
   removeFavoriteAgent = this.mutation(deleteV1AgentByAgentIdFavorite)
-
-  /**
-   * Create a test agent if it does not exist
-   */
-  createTrialAgents = this.mutation(postV1AgentsTrial)
   /**
    * Updates an agent
    */