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

package/src/client/agent.ts

@@ -1,8 +1,12 @@
 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, postV1AgentsTrial, 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 +32,56 @@ 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' }))
+        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 +92,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
    */

package/src/client/ai.ts

@@ -33,7 +33,6 @@ import {
   quickCommandsRunV2V2QuickCommandsSlugStepsStepSlugRunPost,
   resetKnowledgeObjectsV1KnowledgeSourcesSlugObjectsDelete,
   runFetchStepV1QuickCommandsSlugStepsStepSlugFetchRunPost,
-  searchKnowledgeSourcesV1KnowledgeSourcesSearchPost,
   updateQuickCommandV1QuickCommandsSlugPatch,
   updateTitleV1ConversationsConversationIdPatch,
   vectorizeCustomKnowledgeSourceV1KnowledgeSourcesSlugCustomPost,
@@ -135,10 +134,6 @@ class AIClient extends ReactQueryNetworkClient {
    * Gets a knowledge source document by the slug of the parent knowledge source and the document id.
    */
   knowledgeSourceDocument = this.query(removeAuthorizationParam(findKnowledgeObjectByCustomIdV1KnowledgeSourcesSlugObjectsCustomIdGet))
-  /**
-   * Lists knowledge sources matching the provided IDs.
-   */
-  searchKnowledgeSources = this.query(removeAuthorizationParam(searchKnowledgeSourcesV1KnowledgeSourcesSearchPost))
   /**
    * Gets the chat history. This is a paginated resource.
    */

package/src/client/discovery.ts

@@ -1,5 +1,5 @@
 import { HttpError } from '@oazapfts/runtime'
-import { attach, createHypothesis, defaults, deleteHypothesisById, deleteV1DocumentsByDocumentIdHypothesesAndHypothesisId, download, findHypothesisById, getAll, getById, list, list1, listAllDocumentsWithHypotheses, listHypothesis, upload } from '../api/discovery'
+import { attach, createHypothesis, defaults, deleteHypothesisById, deleteV1DocumentsByDocumentIdHypothesesAndHypothesisId, download, findHypothesisById, getAll, getById, list, list1, listAllDocumentsWithHypotheses, listHypothesis, listVersions, upload } from '../api/discovery'
 import apis from '../apis.json'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { baseDictionary } from '../error/dictionary/base'
@@ -68,6 +68,10 @@ class DiscoveryClient extends ReactQueryNetworkClient {
    * Download an document
    */
   downloadDocument = this.query(removeAuthorizationParam(download))
+  /**
+   * List versions of a document
+   */
+  listDocumentVersions = this.query(removeAuthorizationParam(listVersions))
 }
 
 export const discoveryClient = new DiscoveryClient()

package/src/client/types.ts

@@ -1,6 +1,7 @@
 import { RequestOpts } from '@oazapfts/runtime'
 import { AccountScmInfoSaveRequest, AccountScmInfoUpdateRequest, AccountScmStatusResponse, GroupsFromResourceResponse, MembersFromResourceResponse } from '../api/account'
-import { AgentVisibilityLevelEnum, HttpMethod, ListAgentResponse, VisibilityLevelEnum } from '../api/agent-tools'
+import { AgentResponse, VisibilityLevel } from '../api/agent'
+import { HttpMethod } from '../api/agent-tools'
 import { ChatRequest, ChatResponse3, ContentDependencyResponse, ConversationHistoryResponse, ConversationResponse, DependencyResponse } from '../api/ai'
 import { ConnectAccountRequestV2, ManagedAccountProvisionRequest } from '../api/cloudAccount'
 import { AllocationCostRequest, AllocationCostResponse, ChargePeriod, getAllocationCostFilters, ManagedService, ServiceResource } from '../api/cloudServices'
@@ -219,7 +220,6 @@ export interface WorkspaceAiMembersPermissions {
   totalPages: number,
 }
 
-
 export interface WorkspaceAiGroupsPermissions extends GroupsFromResourceResponse {
   actions: Action[],
 }
@@ -358,14 +358,12 @@ export type FixVariables<
 
 export type ReplaceResult<T extends (...args: any[]) => Promise<any>, Fix> = (...args: Parameters<T>) => Promise<Fix>
 
-export interface AgentResponseWithBuiltIn extends Omit<ListAgentResponse, 'conversation_starter' | 'avatar'>   {
+export interface AgentResponseWithBuiltIn extends AgentResponse {
   builtIn?: boolean,
   spaceName?: string,
-  conversation_starter?: string[] | null,
-  avatar?: string | null | undefined,
-} 
+}
 
-export type AgentVisibilityLevel = AgentVisibilityLevelEnum | VisibilityLevelEnum
+export type AgentVisibilityLevel = VisibilityLevel | 'BUILT-IN' | 'ALL'
 
 export interface AgentToolsOpenAPIPreview {
   name: string,