@stack-spot/portal-network 0.146.2 → 0.148.0

package/src/api/genAiInference.ts

@@ -328,10 +328,12 @@ export function addSelfHostedModelV1LlmModelsPost({ xAccountId, authorization, c
 /**
  * List Models
  */
-export function listModelsV1LlmModelsGet({ active, resource, $default, orderBy, orderDir, page, pageSize, xAccountId, authorization }: {
+export function listModelsV1LlmModelsGet({ active, resource, $default, displayName, selfHosted, orderBy, orderDir, page, pageSize, xAccountId, authorization }: {
     active?: boolean | null;
     resource?: string | null;
     $default?: boolean | null;
+    displayName?: string | null;
+    selfHosted?: boolean | null;
     orderBy?: OrderBy;
     orderDir?: OrderDir;
     page?: number;
@@ -351,6 +353,8 @@ export function listModelsV1LlmModelsGet({ active, resource, $default, orderBy,
         active,
         resource,
         "default": $default,
+        display_name: displayName,
+        self_hosted: selfHosted,
         order_by: orderBy,
         order_dir: orderDir,
         page,
@@ -545,7 +549,8 @@ export function listLlmProvidersV1LlmProvidersGet({ acceptsSelfHosted, xAccountI
 /**
  * Handle completions requests
  */
-export function createCompletionsV1ChatCompletionsPost({ xAccountId, authorization, chatCompletion }: {
+export function createCompletionsV1ChatCompletionsPost({ xRequestOrigin, xAccountId, authorization, chatCompletion }: {
+    xRequestOrigin?: string | null;
     xAccountId?: string | null;
     authorization: string;
     chatCompletion: ChatCompletion;
@@ -563,6 +568,7 @@ export function createCompletionsV1ChatCompletionsPost({ xAccountId, authorizati
         method: "POST",
         body: chatCompletion,
         headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-request-origin": xRequestOrigin,
             "x-account-id": xAccountId,
             authorization
         })

package/src/client/agent-tools.ts

@@ -1,8 +1,10 @@
 import { HttpError } from '@oazapfts/runtime'
-import { defaults, getPublicToolKitsV1BuiltinToolkitGet, HttpValidationError } from '../api/agent-tools'
+import { createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listToolkitsV1ToolkitsGet, updateToolkitV1ToolkitsToolkitIdPatch } from '../api/agent-tools'
 import apis from '../apis.json'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
+import { removeAuthorizationParam } from '../utils/remove-authorization-param'
+import { AgentToolsOpenAPIPreview } from './types'
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -14,11 +16,66 @@ class AgentToolsClient extends ReactQueryNetworkClient {
       status: error.status,
       headers: error.headers,
       stack: error.stack,
-      message: (error.data as HttpValidationError | undefined)?.detail?.map(d => d.msg)?.join('\n'),
+      // @ts-ignore API documentation is wrong
+      message: (error.data as HttpValidationError | undefined)?.details?.map(d => d.msg)?.join('\n'),
     })
   }
 
   tools = this.query(getPublicToolKitsV1BuiltinToolkitGet)
+
+  /**
+   * Get list of Toolkits
+   */
+  toolkits = this.query(removeAuthorizationParam(listToolkitsV1ToolkitsGet))
+  /**
+   * Get a toolkit by Id
+   */
+  toolkitById = this.query(removeAuthorizationParam(getToolkitV1ToolkitsToolkitIdGet))
+  /**
+   * Create a toolkit
+   */
+  createToolkit = this.mutation(removeAuthorizationParam(createToolkitV1ToolkitsPost))
+  /**
+   * Update a toolkit
+   */
+  updateToolkit = this.mutation(removeAuthorizationParam(updateToolkitV1ToolkitsToolkitIdPatch))
+  /**
+   * Delete a toolkit
+   */
+  deleteToolkit = this.mutation(removeAuthorizationParam(deleteToolkitV1ToolkitsToolkitIdDelete))
+  /**
+   * Get Tool from Toolkit
+   */
+  toolFromToolkit = this.query(removeAuthorizationParam(getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet))
+  /**
+   * Add tool on Toolkit
+   */
+  addToolOnToolkit = this.mutation(removeAuthorizationParam(createToolkitToolsV1ToolkitsToolkitIdToolsPost))
+  /**
+   * Edit tool on Toolkit
+   */
+  editToolOnToolkit = this.mutation(removeAuthorizationParam(editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut))
+  /**
+   * Delete tool from Toolkit
+   */
+  deleteToolFromToolkit = this.mutation(removeAuthorizationParam(deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete))
+  /**
+   * Previews an uploaded open api file.
+   */
+  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
+    },
+  })
 }
 
 export const agentToolsClient = new AgentToolsClient()

package/src/client/data-integration.ts

@@ -9,6 +9,7 @@ import {
   listKsV1KnowledgeSourcesGet,
   saveChunkedFilesDeprecatedV1FileUploadFileUploadIdPost,
   saveChunkedFilesV1FileUploadFileUploadIdKnowledgeObjectsPost,
+  splitUploadedFilePreviewV1FileUploadFileUploadIdSplitPreviewSplitStrategyGet,
   splitUploadedFileV1FileUploadFileUploadIdSplitPost,
 } from '../api/dataIntegration'
 import apis from '../apis.json'
@@ -65,7 +66,12 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
         'Accept': 'text/event-stream',
       },
       signal: options?.signal,
-    })}
+    })
+  }
+  /**
+   * Generate a preview with splitted objects
+   */
+  previewFromSplit = this.query(removeAuthorizationParam(splitUploadedFilePreviewV1FileUploadFileUploadIdSplitPreviewSplitStrategyGet))
   /**
    * List of knowledge sources
    */
@@ -81,24 +87,26 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
 
   uploadFiles = this.mutation({
     name: 'uploadFiles',
-    request: (abortSignal, params: { files: File[], type: FileUploadType }) => Promise.all(params.files.map(async (f) => {
-      try {
-        const { id, url, form } = await getUploadFormV2V2FileUploadFormPost({
-          authorization: '',
-          newFileUploadRequest: { file_name: f.name, target_type: params.type, target_id: '' },
-        }, { signal: abortSignal })
-        const formData = new FormData()
-        Object.entries(form).forEach(([key, value]) => {
-          formData.append(key, value)
-        })
-        formData.append('file', f)
-        const response = await fetch(url, { method: 'post', body: formData, signal: abortSignal })
-        if (!response.ok) throw new StackspotAPIError({ status: response.status, headers: response.headers, message: 'Failed to upload' })
-        return id
-      } catch (e: any) {
-        throw new FileUploadError(f.name, e)
-      } 
-    })),
+    request: (abortSignal, params: { files: File[], type: FileUploadType, targetId?: string }) => (
+      Promise.all(params.files.map(async (f) => {
+        try {
+          const { id, url, form } = await getUploadFormV2V2FileUploadFormPost({
+            authorization: '',
+            newFileUploadRequest: { file_name: f.name, target_type: params.type, target_id: params.targetId ?? '' },
+          }, { signal: abortSignal })
+          const formData = new FormData()
+          Object.entries(form).forEach(([key, value]) => {
+            formData.append(key, value)
+          })
+          formData.append('file', f)
+          const response = await fetch(url, { method: 'post', body: formData, signal: abortSignal })
+          if (!response.ok) throw new StackspotAPIError({ status: response.status, headers: response.headers, message: 'Failed to upload' })
+          return id
+        } catch (e: any) {
+          throw new FileUploadError(f.name, e)
+        } 
+      }))
+    ),
   })
 }
 

package/src/client/types.ts

@@ -1,6 +1,7 @@
 import { RequestOpts } from '@oazapfts/runtime'
 import { AccountScmInfoSaveRequest, AccountScmInfoUpdateRequest, AccountScmStatusResponse, GroupsFromResourceResponse, MembersFromResourceResponse } from '../api/account'
 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'
@@ -363,3 +364,12 @@ export interface AgentResponseWithBuiltIn extends AgentResponse {
 }
 
 export type AgentVisibilityLevel = VisibilityLevel | 'BUILT-IN' | 'ALL'
+
+export interface AgentToolsOpenAPIPreview {
+  name: string,
+  description: string,
+  method: HttpMethod,
+  url: string,
+  parameters?: Record<string, any>,
+  request_body?: Record<string, any>,
+}

package/src/utils/StreamedJson.tsx

@@ -92,8 +92,8 @@ export class StreamedJson<T> {
       else if (typeof source[k] === 'string') target[k] += source[k]
       else if (Array.isArray(source[k])) target[k].push(...source[k])
       else if (typeof source[k] === 'number') parseFloat(target[k] + source[k])
-      else if (typeof source[k] === 'object') this.merge(source[k], target[k])
-      else target[k] = source[k]
+      else if (source[k] && typeof source[k] === 'object') this.merge(source[k], target[k])
+      else if (source[k]) target[k] = source[k]
     })
   }