sequenceDiagram
  actor User
  participant Controls as ChatInputControls
  participant Store as ChatStore
  participant Runtime as AgentRuntimeService
  participant Exec as StreamingExecutor
  participant Agent as GeneralChatAgent

  User->>Controls: Toggle enableContextCompression
  Controls->>Store: Update chatConfig.enableContextCompression

  User->>Controls: Send message
  Controls->>Store: Dispatch streamingExecutor

  Store->>Exec: streamingExecutor(agentConfigData, modelRuntimeConfig)
  Exec->>Agent: new GeneralChatAgent({
  Exec->>Agent:  compressionConfig.enabled = agentConfigData.chatConfig.enableContextCompression ?? true
  Exec->>Agent: })

  User->>Runtime: Start server-side chat (if applicable)
  Runtime->>Agent: new GeneralChatAgent({
  Runtime->>Agent:  compressionConfig.enabled = metadata.agentConfig.chatConfig.enableContextCompression ?? true
  Runtime->>Agent: })

  Agent->>Agent: phase init/user_input
  Agent->>Agent: compressionEnabled = config.compressionConfig.enabled ?? true
  Agent->>Agent: if compressionEnabled then shouldCompress(messages)
  Agent-->>Store: AgentInstruction compress_context (only if needsCompression)
  Agent-->>Store: Normal LLM call if no compression or disabled

classDiagram
  class GeneralChatAgent {
    - GeneralAgentConfig config
    + constructor(config: GeneralAgentConfig)
    + handlePhaseInit(state: any, context: any): any
    + findExistingSummary(messages: any[]): any
  }

  class GeneralAgentConfig {
    + any agentConfig
    + CompressionConfig compressionConfig
    + ModelRuntimeConfig modelRuntimeConfig
    + string operationId
    + string userId
  }

  class CompressionConfig {
    + boolean enabled
    + number maxWindowToken
  }

  class ModelRuntimeConfig {
    + string model
    + string provider
    + CompressionModelConfig compressionModel
  }

  class CompressionModelConfig {
    + string model
    + string provider
  }

  class AgentRuntimeService {
    - Map~string, StepLifecycleCallbacks~ stepCallbacks
    + createAgent(operationId: string, metadata: any): GeneralChatAgent
    + baseURL(): string
  }

  class StreamingExecutor {
    + executeStreaming(agentConfigData: any, modelRuntimeConfig: ModelRuntimeConfig, messageKey: string, parentMessageId: string): void
  }

  GeneralChatAgent --> GeneralAgentConfig : uses
  GeneralAgentConfig --> CompressionConfig : has
  GeneralAgentConfig --> ModelRuntimeConfig : has
  ModelRuntimeConfig --> CompressionModelConfig : has
  AgentRuntimeService --> GeneralChatAgent : instantiates
  StreamingExecutor --> GeneralChatAgent : instantiates

flowchart TD
  A[Start findLastMessageIdRecursive] --> B{node is undefined?}
  B -->|yes| Z[Return undefined]
  B -->|no| C{node has children?}
  C -->|yes| D[Get last child]
  D --> E[Recurse on last child]
  E --> Z
  C -->|no| F{node has tools?}
  F -->|yes| G[Get last tool]
  G --> H[Return last tool result_msg_id]
  H --> Z
  F -->|no| I{node.role is compressedGroup and node.lastMessageId exists?}
  I -->|yes| J[Return node.lastMessageId]
  J --> Z
  I -->|no| K[Return node.id]
  K --> Z