@stack-spot/portal-network 1.0.0-dev.1769537511491 → 1.0.0-dev.1769793051111

package/src/client/types.ts

@@ -1,7 +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 { ChatRequest, ChatResponse3, ContentDependencyResponse, ConversationHistoryResponse, ConversationResponse, DependencyResponse, QuickCommandResponse, QuickCommandStepResult } from '../api/ai'
+import { ChatRequest, ChatResponse3, ContentDependencyResponse, ConversationHistoryResponse, ConversationResponse, DependencyResponse, QuickCommandResponse, QuickCommandStepResult, SourceKnowledgeSource, SourceProjectFile3, SourceStackAi } from '../api/ai'
 import { ConnectAccountRequestV2, ManagedAccountProvisionRequest } from '../api/cloudAccount'
 import { AllocationCostRequest, AllocationCostResponse, ChargePeriod, getAllocationCostFilters, ManagedService, ServiceResource } from '../api/cloudServices'
 import { Action } from '../api/workspace-ai'
@@ -172,20 +172,11 @@ export interface FixedPaginatedActivityResponse extends Omit<PaginatedActivityRe
 
 export interface FixedChatRequest extends ChatRequest {
   context?: {
-    workspace?: string,
-    conversation_id?: string,
-    stack_id?: string,
-    language?: string,
-    project_recent_files?: string,
-    knowledge_sources?: string[],
-    agent_id?: string,
-    agent_built_in?: boolean,
     platform?: string,
     platform_version?: string,
     stackspot_ai_version?: string,
     os?: string,
-    upload_ids?: string[],
-    selected_model_id?: string,
+    agent_id?: string,
   },
 }
 
@@ -194,6 +185,7 @@ export interface FixedConversationHistoryResponse extends ConversationHistoryRes
   custom_agent?: {
     id: string,
     built_in: boolean,
+    version_number?: number,
   },
 }
 
@@ -366,9 +358,17 @@ export interface ToolCallsAgentInfo extends BaseAgentInfo {
 
 export type AgentInfo = GenericAgentInfo | PlanningAgentInfo | StepAgentInfo | ToolAgentInfo | ToolCallsAgentInfo
 
-export interface FixedChatResponse extends ChatResponse3 {
-  agent_info: AgentInfo,
-  tools?: string[],
+export type ChatResponse = {
+    message: string,
+    cross_account_source: SourceKnowledgeSource[],
+    source: (SourceStackAi | SourceKnowledgeSource | SourceProjectFile3)[],
+    message_id: string | null,
+    stop_reason: 'stop',
+};
+
+export interface FixedChatResponse extends ChatResponse {
+  agent_info: AgentInfo[],
+  tools_id?: string[],
 }
 export interface OpportunitiesPMAgent {
   title: string,

package/src/utils/StreamedJson.tsx

@@ -21,6 +21,11 @@ interface ConstructorParams<T> {
    * @default 50
    */
   minChangeIntervalMS?: number,
+  /**
+   * Allows overriding the minimum interval by event.
+   * Useful when some events must flush immediately (agent_info events).
+   */
+  minChangeIntervalMSFromEvent?: (event: Partial<T>, data: Partial<T>) => number | undefined,
   /**
    * Optional. If set, this function will be called with every streaming event and must transform the current data object according to the
    * message received.
@@ -45,16 +50,18 @@ export class StreamedJson<T> {
   private abortController: AbortController | undefined
   private transform?: (event: Partial<T>, data: Partial<T>) => void | Promise<void>
   private ignoreKeys?: (keyof T)[]
+  private minChangeIntervalMSFromEvent?: (event: Partial<T>, data: Partial<T>) => number | undefined
   private textFromErrorEvent: (data: Partial<T>) => string
 
   /**
    * @param response the fetch response.
    * @param minChangeIntervalMS a stream can be too fast. This sets a minimum interval between running the listeners. The default is 50ms.
    */
-  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, transform, ignoreKeys, textFromErrorEvent }: ConstructorParams<T>) {
+  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, minChangeIntervalMSFromEvent, transform, ignoreKeys, textFromErrorEvent }: ConstructorParams<T>) {
     this.abortController = abortController
     this.transform = transform
     this.ignoreKeys = ignoreKeys
+    this.minChangeIntervalMSFromEvent = minChangeIntervalMSFromEvent
     this.run(eventsPromise, minChangeIntervalMS)
     this.textFromErrorEvent = textFromErrorEvent ?? ((data) => JSON.stringify(data))
   }
@@ -77,7 +84,9 @@ export class StreamedJson<T> {
           }
           await this.transform?.(json, this.data)
           this.merge(json, this.data)
-          if (event.event !== 'error' && new Date().getTime() - lastChangeCall >= minChangeIntervalMS) {
+
+          const intervalMS = this.minChangeIntervalMSFromEvent?.(json, this.data) ?? minChangeIntervalMS
+          if (event.event !== 'error' && new Date().getTime() - lastChangeCall >= intervalMS) {
             this.onChangeListeners.forEach(l => l(this.data))
             lastChangeCall = new Date().getTime()
             flushed = true