@stack-spot/portal-network 0.218.2 → 0.219.0

package/src/client/workspace-ai.ts

@@ -1,11 +1,11 @@
 import { HttpError } from '@oazapfts/runtime'
 import { getApiAddresses } from '../api-addresses'
-import { AgentResponse } from '../api/agent'
 import { CustomToolkitResponse } from '../api/agent-tools'
 import { GetAiStackResponse, QuickCommandResponse } from '../api/ai'
 import { KnowledgeSourceItemResponse } from '../api/dataIntegration'
 import {
   addContentV1WorkspacesWorkspaceIdContentTypePost,
+  addContentV2WorkspacesWorkspaceIdContentTypePost,
   addFavoriteV1WorkspacesWorkspaceIdFavoritePost,
   addPermissionV1ResourceTypeResourceTypeResourcesResourceIdentifierPatch,
   addPermissionV1WorkspacesWorkspaceIdPermissionsPatch,
@@ -31,7 +31,7 @@ import { wksAiDictionary } from '../error/dictionary/workspace-ai'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
-import { FixedAddResourceToWorkspaceAi, FixedWorkspaceAiPermissions, ReplaceResult } from './types'
+import { AgentResponseWithBuiltIn, FixedAddResourceToWorkspaceAi, FixedWorkspaceAiPermissions, ReplaceResult } from './types'
 
 class WorkspaceAiClient extends ReactQueryNetworkClient {
   constructor() {
@@ -49,6 +49,7 @@ class WorkspaceAiClient extends ReactQueryNetworkClient {
   deleteWorkspaceAi = this.mutation(removeAuthorizationParam(deleteWorkspaceV1WorkspacesWorkspaceIdDelete))
 
   /**
+   * @deprecated
    * Adds a resource to the Spot.
    */
   addResourceTypeToWorkspaceAi = this.mutation({
@@ -62,6 +63,20 @@ class WorkspaceAiClient extends ReactQueryNetworkClient {
     permission: this.createPermissionFunctionFor(addContentV1WorkspacesWorkspaceIdContentTypePost),
   })
 
+  /**
+   * V2 for adding a resource to the Spot.
+   */
+  addResourceTypeV2ToSpot = this.mutation({
+    name: 'addResourceTypeToWorkspaceAi',
+    request: (
+      signal,
+      variables: Omit<Parameters<typeof addContentV2WorkspacesWorkspaceIdContentTypePost>[0], 'authorization'>,
+    ) =>
+      addContentV2WorkspacesWorkspaceIdContentTypePost({ ...variables, authorization: '' },
+        { signal }) as unknown as Promise<FixedAddResourceToWorkspaceAi>,
+    permission: this.createPermissionFunctionFor(addContentV2WorkspacesWorkspaceIdContentTypePost),
+  })
+
   /**
    * Lists all the knowledge sources in a Spot.
    */
@@ -118,7 +133,7 @@ class WorkspaceAiClient extends ReactQueryNetworkClient {
         'authorization' | 'contentType'>,
     ) =>
       listAllByContentTypeV1WorkspacesWorkspaceIdContentTypeGet({ ...variables, contentType: 'agent', authorization: '' },
-        { signal }) as unknown as Promise<AgentResponse[]>,
+        { signal }) as unknown as Promise<AgentResponseWithBuiltIn[]>,
     permission: this.createPermissionFunctionFor(listAllByContentTypeV1WorkspacesWorkspaceIdContentTypeGet),
   })