@xopcai/xopc 0.0.8 → 0.0.11

package/dist/src/agent/orchestration/agent-orchestrator.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-orchestrator.js","names":[],"sources":["../../../../src/agent/orchestration/agent-orchestrator.ts"],"sourcesContent":["/**\n * Agent Orchestrator - Coordinates Agent execution flow\n *\n * Manages the complete agent execution pipeline from message processing\n * to response generation.\n */\n\nimport type { Agent, AgentMessage } from '@mariozechner/pi-agent-core';\nimport type { Config } from '../../config/schema.js';\nimport type { InboundMessage } from '../../infra/bus/index.js';\nimport type { SessionConfigStore, SessionStore } from '../../session/index.js';\nimport { resolveEffectiveThinkingLevel } from '../../session/thinking-resolve.js';\nimport type { ThinkLevel } from '../transcript/thinking-types.js';\nimport type { ModelManager } from '../models/index.js';\nimport type { SessionContext } from '../session/session-context.js';\nimport type { AgentEventHandler } from './agent-event-handler.js';\nimport type { FeedbackCoordinator } from '../feedback/feedback-coordinator.js';\nimport type { AgentManager } from '../agent-manager.js';\nimport { sanitizeMessages, cleanTrailingErrors } from '../memory/message-sanitizer.js';\nimport {\n  tryApplySessionTranscriptHygiene,\n  tryApplySessionTranscriptHygieneForPersistence,\n} from '../transcript/transcript-hygiene.js';\nimport { createLogger } from '../../utils/logger.js';\nimport { extractAgentUserPlainText } from '../memory/user-message-text.js';\nimport { runAgentTurnWithModelFallbacks } from './run-agent-turn-with-fallbacks.js';\nimport {\n  persistInboundAttachmentsToWorkspace,\n  formatInboundFileTextBlock,\n} from '../../channels/attachments/inbound-persist.js';\nimport { resolveInboundImageContentParts } from '../image/inbound-image-handling.js';\n\nconst log = createLogger('AgentOrchestrator');\n\nexport interface AgentOrchestratorConfig {\n  agentManager: AgentManager;\n  sessionStore: SessionStore;\n  modelManager: ModelManager;\n  eventHandler: AgentEventHandler;\n  feedbackCoordinator: FeedbackCoordinator;\n  sessionConfigStore: SessionConfigStore;\n  /** Load per-session workspace override and mkdir before creating the agent. */\n  hydrateSessionWorkspaceFromStore?: (sessionKey: string) => Promise<void>;\n  getThinkingDefault: () => ThinkLevel | undefined;\n  /** Per-session default from merged `agents.list` / defaults (optional). */\n  getThinkingDefaultForSession?: (sessionKey: string) => ThinkLevel | undefined;\n  /** Default workspace root when no per-session resolver is set. */\n  workspaceRoot: string;\n  /** Per-agent workspace root for attachments (optional; defaults to `workspaceRoot`). */\n  getWorkspaceRootForSession?: (sessionKey: string) => string;\n  /** Agent home (`…/agents/<id>/`) for inbound/TTS files — keeps internal state out of the markdown workspace. */\n  getAgentInternalStorageRootForSession?: (sessionKey: string) => string;\n  /** Fire-and-forget after full session persist (e.g. LLM session title); not called from mid-turn snapshots. */\n  enqueueAutoTitle?: (sessionKey: string) => void;\n  /** For per-turn timeout via `agents.defaults.maxTaskDurationMs`. */\n  getConfig?: () => Config | undefined;\n}\n\nexport class AgentOrchestrator {\n  private agentManager: AgentManager;\n  private sessionStore: SessionStore;\n  private modelManager: ModelManager;\n  private eventHandler: AgentEventHandler;\n  private feedbackCoordinator: FeedbackCoordinator;\n  private sessionConfigStore: SessionConfigStore;\n  private hydrateSessionWorkspaceFromStore?: (sessionKey: string) => Promise<void>;\n  private getThinkingDefault: () => ThinkLevel | undefined;\n  private getThinkingDefaultForSession?: (sessionKey: string) => ThinkLevel | undefined;\n  private workspaceRoot: string;\n  private getWorkspaceRootForSession?: (sessionKey: string) => string;\n  private getAgentInternalStorageRootForSession: (sessionKey: string) => string;\n  private enqueueAutoTitle?: (sessionKey: string) => void;\n  private getConfig?: () => Config | undefined;\n\n  constructor(config: AgentOrchestratorConfig) {\n    this.agentManager = config.agentManager;\n    this.sessionStore = config.sessionStore;\n    this.modelManager = config.modelManager;\n    this.eventHandler = config.eventHandler;\n    this.feedbackCoordinator = config.feedbackCoordinator;\n    this.sessionConfigStore = config.sessionConfigStore;\n    this.hydrateSessionWorkspaceFromStore = config.hydrateSessionWorkspaceFromStore;\n    this.getThinkingDefault = config.getThinkingDefault;\n    this.getThinkingDefaultForSession = config.getThinkingDefaultForSession;\n    this.workspaceRoot = config.workspaceRoot;\n    this.getWorkspaceRootForSession = config.getWorkspaceRootForSession;\n    this.getAgentInternalStorageRootForSession =\n      config.getAgentInternalStorageRootForSession ??\n      ((sk) => this.getWorkspaceRootForSession?.(sk) ?? this.workspaceRoot);\n    this.enqueueAutoTitle = config.enqueueAutoTitle;\n    this.getConfig = config.getConfig;\n  }\n\n  private async hydrateSessionModelFromStore(sessionKey: string): Promise<void> {\n    const cfg = await this.sessionConfigStore.get(sessionKey);\n    if (cfg?.modelOverride) {\n      await this.modelManager.switchModelForSession(sessionKey, cfg.modelOverride);\n    }\n  }\n\n  /**\n   * Process a message through the agent orchestration pipeline\n   */\n  async process(msg: InboundMessage, context: SessionContext): Promise<void> {\n    const { sessionKey } = context;\n\n    log.debug({ sessionKey }, 'Processing message through agent orchestrator');\n\n    await this.hydrateSessionWorkspaceFromStore?.(sessionKey);\n\n    // Get or create agent for this session\n    const agent = this.agentManager.getOrCreateAgent(sessionKey);\n\n    try {\n      await this.hydrateSessionModelFromStore(sessionKey);\n\n      // 1. Load session history\n      let messages = await this.sessionStore.load(sessionKey);\n\n      // Clean any trailing errors from previous sessions (defensive)\n      messages = cleanTrailingErrors(messages);\n\n      try {\n        const model = this.modelManager.getResolvedModelForSession(sessionKey);\n        messages = tryApplySessionTranscriptHygiene(messages, model);\n      } catch (err) {\n        log.warn({ err, sessionKey }, 'Transcript hygiene skipped (model resolve failed)');\n      }\n\n      agent.state.messages = messages;\n\n      // 2. Apply model configuration for session\n      await this.modelManager.applyModelForSession(agent, sessionKey);\n\n      const thinkingDefault =\n        this.getThinkingDefaultForSession?.(sessionKey) ?? this.getThinkingDefault();\n      const thinkingLevel = await resolveEffectiveThinkingLevel(\n        this.sessionConfigStore,\n        sessionKey,\n        null,\n        thinkingDefault,\n      );\n      this.agentManager.setThinkingLevel(sessionKey, thinkingLevel);\n\n      // 3. Persist inbound files (Telegram, etc.) under agent home, then build user message\n      const storageRoot = this.getAgentInternalStorageRootForSession(sessionKey);\n      const persistedAttachments = await persistInboundAttachmentsToWorkspace(\n        storageRoot,\n        sessionKey,\n        msg.attachments,\n      );\n      const userMessage = await this.buildUserMessage(\n        {\n          ...msg,\n          attachments: persistedAttachments ?? msg.attachments,\n        },\n        sessionKey,\n      );\n      const userPlainForMemory = extractAgentUserPlainText(userMessage);\n      const userMessageForModel = await this.agentManager.applyMemoryPrefetchToUserMessage(\n        userMessage,\n        sessionKey,\n      );\n\n      // 4. Start task feedback\n      this.feedbackCoordinator.startTask();\n\n      // 5. Execute agent\n      await this.executeAgent(agent, userMessageForModel, context);\n\n      this.agentManager.afterAgentTurn(sessionKey, userPlainForMemory);\n      this.agentManager.scheduleBackgroundReviewAfterUserTurn(sessionKey);\n\n      // 6. Sanitize messages before saving (remove error messages, empty content)\n      const rawMessages = agent.state.messages;\n      const { messages: sanitizedMessages, removed } = sanitizeMessages(rawMessages);\n\n      if (removed > 0) {\n        log.info({ sessionKey, removed }, 'Removed problematic messages before saving');\n      }\n\n      // 7. Save session messages (transcript hygiene)\n      await this.saveSessionSnapshot(sessionKey, sanitizedMessages);\n\n      this.enqueueAutoTitle?.(sessionKey);\n\n      // 8. End task feedback\n      this.feedbackCoordinator.endTask();\n\n    } catch (error) {\n      log.error({ err: error, sessionKey }, 'Error in agent orchestration');\n      this.feedbackCoordinator.endTask();\n      throw error;\n    }\n  }\n\n  /**\n   * Transcript hygiene + persist. Expects messages already passed through {@link sanitizeMessages}.\n   * Keeps thinking blocks on disk for UI; agent load path applies full hygiene including dropThinking.\n   */\n  private async saveSessionSnapshot(sessionKey: string, messages: AgentMessage[]): Promise<void> {\n    let toPersist = messages;\n    try {\n      const model = this.modelManager.getResolvedModelForSession(sessionKey);\n      toPersist = tryApplySessionTranscriptHygieneForPersistence(messages, model);\n    } catch (err) {\n      log.warn({ err, sessionKey }, 'Transcript hygiene on save skipped');\n    }\n    await this.sessionStore.save(sessionKey, toPersist);\n  }\n\n  /**\n   * Execute the agent with a user message (primary model, then `agents.defaults.model.fallbacks` on failure).\n   */\n  private async executeAgent(\n    agent: Agent,\n    userMessage: AgentMessage,\n    context: SessionContext\n  ): Promise<void> {\n    const sessionKey = context.sessionKey;\n    await runAgentTurnWithModelFallbacks({\n      agent,\n      sessionKey,\n      modelManager: this.modelManager,\n      userMessage,\n      log,\n      getConfig: this.getConfig,\n      beforeUserPrompt: () => this.agentManager.beginBackgroundReviewUserTurn(sessionKey),\n      afterUserPrompt: async () => {\n        try {\n          const { messages: sanitizedTurn } = sanitizeMessages(agent.state.messages);\n          await this.saveSessionSnapshot(sessionKey, sanitizedTurn);\n          log.debug({ sessionKey }, 'User message saved immediately after prompt');\n        } catch (err) {\n          log.warn({ err, sessionKey }, 'Failed to save user message immediately');\n        }\n      },\n    });\n  }\n\n  /**\n   * Build an agent message from an inbound message\n   */\n  private async buildUserMessage(msg: InboundMessage, sessionKey: string): Promise<AgentMessage> {\n    const storageRootAbs = this.getAgentInternalStorageRootForSession(sessionKey);\n    const textBody = msg.content.trimStart().startsWith('/skill:')\n      ? this.agentManager.expandSkillUserText(msg.content)\n      : msg.content;\n\n    if (!msg.attachments || msg.attachments.length === 0) {\n      return {\n        role: 'user',\n        content: textBody,\n        timestamp: Date.now(),\n      };\n    }\n\n    const modelRef = this.modelManager.getModelForSession(sessionKey);\n    const cfg = this.getConfig?.();\n\n    const messageContent: Array<\n      { type: 'text'; text: string } | { type: 'image'; data: string; mimeType: string }\n    > = [];\n\n    if (msg.content.trim()) {\n      messageContent.push({ type: 'text', text: textBody });\n    }\n\n    const attachments = msg.attachments;\n    let i = 0;\n    while (i < attachments.length) {\n      const att = attachments[i]!;\n      const isImage =\n        att.type === 'image' || att.type === 'photo' || Boolean(att.mimeType?.startsWith('image/'));\n\n      if (isImage) {\n        const group: Array<{ data: string; mimeType: string }> = [];\n        while (i < attachments.length) {\n          const a = attachments[i]!;\n          const img =\n            a.type === 'image' || a.type === 'photo' || Boolean(a.mimeType?.startsWith('image/'));\n          if (!img) {\n            break;\n          }\n          if (!a.data || a.data.length === 0) {\n            log.warn({ type: a.type, name: a.name }, 'Empty image data, skipping');\n            i += 1;\n            continue;\n          }\n          group.push({ data: a.data, mimeType: a.mimeType || 'image/jpeg' });\n          i += 1;\n        }\n        if (group.length > 0) {\n          const parts = await resolveInboundImageContentParts({\n            modelRef,\n            cfg,\n            userTextForContext: msg.content.trim() ? textBody : '',\n            images: group,\n          });\n          messageContent.push(...parts);\n        }\n      } else {\n        const fileBlock = formatInboundFileTextBlock(\n          {\n            type: att.type,\n            mimeType: att.mimeType,\n            name: att.name,\n            size: att.size,\n            workspaceRelativePath: att.workspaceRelativePath,\n          },\n          storageRootAbs,\n        );\n        messageContent.push({ type: 'text', text: fileBlock });\n        i += 1;\n      }\n    }\n\n    const hasText = messageContent.some((item) => item.type === 'text');\n    const hasImage = messageContent.some((item) => item.type === 'image');\n    if (hasImage && !hasText) {\n      messageContent.unshift({ type: 'text', text: 'Please analyze the image(s) I sent.' });\n    }\n\n    if (messageContent.length === 0) {\n      log.warn(\n        { attachmentCount: msg.attachments.length },\n        'All attachments were skipped, falling back to text message',\n      );\n      return {\n        role: 'user',\n        content: textBody || '[Image attachment could not be processed]',\n        timestamp: Date.now(),\n      };\n    }\n\n    return {\n      role: 'user',\n      content: messageContent,\n      timestamp: Date.now(),\n    };\n  }\n\n  /**\n   * Get the current agent model ID\n   */\n  getCurrentModel(): string {\n    return this.modelManager.getCurrentModel();\n  }\n\n  /**\n   * Check if agent is currently processing for a session\n   */\n  isProcessing(sessionKey: string): boolean {\n    const agent = this.agentManager.getAgent(sessionKey);\n    if (!agent) {\n      return false;\n    }\n    return agent.state.messages.length > 0;\n  }\n\n  /**\n   * Abort current agent execution for a session\n   */\n  abort(sessionKey: string): void {\n    const agent = this.agentManager.getAgent(sessionKey);\n    if (agent) {\n      agent.abort();\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;aAuBqD;AASrD,MAAM,MAAM,aAAa,oBAAoB;AA0B7C,IAAa,oBAAb,MAA+B;CAC7B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YAAY,QAAiC;AAC3C,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,sBAAsB,OAAO;AAClC,OAAK,qBAAqB,OAAO;AACjC,OAAK,mCAAmC,OAAO;AAC/C,OAAK,qBAAqB,OAAO;AACjC,OAAK,+BAA+B,OAAO;AAC3C,OAAK,gBAAgB,OAAO;AAC5B,OAAK,6BAA6B,OAAO;AACzC,OAAK,wCACH,OAAO,2CACL,OAAO,KAAK,6BAA6B,GAAG,IAAI,KAAK;AACzD,OAAK,mBAAmB,OAAO;AAC/B,OAAK,YAAY,OAAO;;CAG1B,MAAc,6BAA6B,YAAmC;EAC5E,MAAM,MAAM,MAAM,KAAK,mBAAmB,IAAI,WAAW;AACzD,MAAI,KAAK,cACP,OAAM,KAAK,aAAa,sBAAsB,YAAY,IAAI,cAAc;;;;;CAOhF,MAAM,QAAQ,KAAqB,SAAwC;EACzE,MAAM,EAAE,eAAe;AAEvB,MAAI,MAAM,EAAE,YAAY,EAAE,gDAAgD;AAE1E,QAAM,KAAK,mCAAmC,WAAW;EAGzD,MAAM,QAAQ,KAAK,aAAa,iBAAiB,WAAW;AAE5D,MAAI;AACF,SAAM,KAAK,6BAA6B,WAAW;GAGnD,IAAI,WAAW,MAAM,KAAK,aAAa,KAAK,WAAW;AAGvD,cAAW,oBAAoB,SAAS;AAExC,OAAI;IACF,MAAM,QAAQ,KAAK,aAAa,2BAA2B,WAAW;AACtE,eAAW,iCAAiC,UAAU,MAAM;YACrD,KAAK;AACZ,QAAI,KAAK;KAAE;KAAK;KAAY,EAAE,oDAAoD;;AAGpF,SAAM,MAAM,WAAW;AAGvB,SAAM,KAAK,aAAa,qBAAqB,OAAO,WAAW;GAE/D,MAAM,kBACJ,KAAK,+BAA+B,WAAW,IAAI,KAAK,oBAAoB;GAC9E,MAAM,gBAAgB,MAAM,8BAC1B,KAAK,oBACL,YACA,MACA,gBACD;AACD,QAAK,aAAa,iBAAiB,YAAY,cAAc;GAI7D,MAAM,uBAAuB,MAAM,qCADf,KAAK,sCAAsC,WAAW,EAGxE,YACA,IAAI,YACL;GACD,MAAM,cAAc,MAAM,KAAK,iBAC7B;IACE,GAAG;IACH,aAAa,wBAAwB,IAAI;IAC1C,EACD,WACD;GACD,MAAM,qBAAqB,0BAA0B,YAAY;GACjE,MAAM,sBAAsB,MAAM,KAAK,aAAa,iCAClD,aACA,WACD;AAGD,QAAK,oBAAoB,WAAW;AAGpC,SAAM,KAAK,aAAa,OAAO,qBAAqB,QAAQ;AAE5D,QAAK,aAAa,eAAe,YAAY,mBAAmB;AAChE,QAAK,aAAa,sCAAsC,WAAW;GAGnE,MAAM,cAAc,MAAM,MAAM;GAChC,MAAM,EAAE,UAAU,mBAAmB,YAAY,iBAAiB,YAAY;AAE9E,OAAI,UAAU,EACZ,KAAI,KAAK;IAAE;IAAY;IAAS,EAAE,6CAA6C;AAIjF,SAAM,KAAK,oBAAoB,YAAY,kBAAkB;AAE7D,QAAK,mBAAmB,WAAW;AAGnC,QAAK,oBAAoB,SAAS;WAE3B,OAAO;AACd,OAAI,MAAM;IAAE,KAAK;IAAO;IAAY,EAAE,+BAA+B;AACrE,QAAK,oBAAoB,SAAS;AAClC,SAAM;;;;;;;CAQV,MAAc,oBAAoB,YAAoB,UAAyC;EAC7F,IAAI,YAAY;AAChB,MAAI;AAEF,eAAY,+CAA+C,UAD7C,KAAK,aAAa,2BAA2B,WAAW,CACK;WACpE,KAAK;AACZ,OAAI,KAAK;IAAE;IAAK;IAAY,EAAE,qCAAqC;;AAErE,QAAM,KAAK,aAAa,KAAK,YAAY,UAAU;;;;;CAMrD,MAAc,aACZ,OACA,aACA,SACe;EACf,MAAM,aAAa,QAAQ;AAC3B,QAAM,+BAA+B;GACnC;GACA;GACA,cAAc,KAAK;GACnB;GACA;GACA,WAAW,KAAK;GAChB,wBAAwB,KAAK,aAAa,8BAA8B,WAAW;GACnF,iBAAiB,YAAY;AAC3B,QAAI;KACF,MAAM,EAAE,UAAU,kBAAkB,iBAAiB,MAAM,MAAM,SAAS;AAC1E,WAAM,KAAK,oBAAoB,YAAY,cAAc;AACzD,SAAI,MAAM,EAAE,YAAY,EAAE,8CAA8C;aACjE,KAAK;AACZ,SAAI,KAAK;MAAE;MAAK;MAAY,EAAE,0CAA0C;;;GAG7E,CAAC;;;;;CAMJ,MAAc,iBAAiB,KAAqB,YAA2C;EAC7F,MAAM,iBAAiB,KAAK,sCAAsC,WAAW;EAC7E,MAAM,WAAW,IAAI,QAAQ,WAAW,CAAC,WAAW,UAAU,GAC1D,KAAK,aAAa,oBAAoB,IAAI,QAAQ,GAClD,IAAI;AAER,MAAI,CAAC,IAAI,eAAe,IAAI,YAAY,WAAW,EACjD,QAAO;GACL,MAAM;GACN,SAAS;GACT,WAAW,KAAK,KAAK;GACtB;EAGH,MAAM,WAAW,KAAK,aAAa,mBAAmB,WAAW;EACjE,MAAM,MAAM,KAAK,aAAa;EAE9B,MAAM,iBAEF,EAAE;AAEN,MAAI,IAAI,QAAQ,MAAM,CACpB,gBAAe,KAAK;GAAE,MAAM;GAAQ,MAAM;GAAU,CAAC;EAGvD,MAAM,cAAc,IAAI;EACxB,IAAI,IAAI;AACR,SAAO,IAAI,YAAY,QAAQ;GAC7B,MAAM,MAAM,YAAY;AAIxB,OAFE,IAAI,SAAS,WAAW,IAAI,SAAS,WAAW,QAAQ,IAAI,UAAU,WAAW,SAAS,CAAC,EAEhF;IACX,MAAM,QAAmD,EAAE;AAC3D,WAAO,IAAI,YAAY,QAAQ;KAC7B,MAAM,IAAI,YAAY;AAGtB,SAAI,EADF,EAAE,SAAS,WAAW,EAAE,SAAS,WAAW,QAAQ,EAAE,UAAU,WAAW,SAAS,CAAC,EAErF;AAEF,SAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,WAAW,GAAG;AAClC,UAAI,KAAK;OAAE,MAAM,EAAE;OAAM,MAAM,EAAE;OAAM,EAAE,6BAA6B;AACtE,WAAK;AACL;;AAEF,WAAM,KAAK;MAAE,MAAM,EAAE;MAAM,UAAU,EAAE,YAAY;MAAc,CAAC;AAClE,UAAK;;AAEP,QAAI,MAAM,SAAS,GAAG;KACpB,MAAM,QAAQ,MAAM,gCAAgC;MAClD;MACA;MACA,oBAAoB,IAAI,QAAQ,MAAM,GAAG,WAAW;MACpD,QAAQ;MACT,CAAC;AACF,oBAAe,KAAK,GAAG,MAAM;;UAE1B;IACL,MAAM,YAAY,2BAChB;KACE,MAAM,IAAI;KACV,UAAU,IAAI;KACd,MAAM,IAAI;KACV,MAAM,IAAI;KACV,uBAAuB,IAAI;KAC5B,EACD,eACD;AACD,mBAAe,KAAK;KAAE,MAAM;KAAQ,MAAM;KAAW,CAAC;AACtD,SAAK;;;EAIT,MAAM,UAAU,eAAe,MAAM,SAAS,KAAK,SAAS,OAAO;AAEnE,MADiB,eAAe,MAAM,SAAS,KAAK,SAAS,QAAQ,IACrD,CAAC,QACf,gBAAe,QAAQ;GAAE,MAAM;GAAQ,MAAM;GAAuC,CAAC;AAGvF,MAAI,eAAe,WAAW,GAAG;AAC/B,OAAI,KACF,EAAE,iBAAiB,IAAI,YAAY,QAAQ,EAC3C,6DACD;AACD,UAAO;IACL,MAAM;IACN,SAAS,YAAY;IACrB,WAAW,KAAK,KAAK;IACtB;;AAGH,SAAO;GACL,MAAM;GACN,SAAS;GACT,WAAW,KAAK,KAAK;GACtB;;;;;CAMH,kBAA0B;AACxB,SAAO,KAAK,aAAa,iBAAiB;;;;;CAM5C,aAAa,YAA6B;EACxC,MAAM,QAAQ,KAAK,aAAa,SAAS,WAAW;AACpD,MAAI,CAAC,MACH,QAAO;AAET,SAAO,MAAM,MAAM,SAAS,SAAS;;;;;CAMvC,MAAM,YAA0B;EAC9B,MAAM,QAAQ,KAAK,aAAa,SAAS,WAAW;AACpD,MAAI,MACF,OAAM,OAAO"}
+{"version":3,"file":"agent-orchestrator.js","names":[],"sources":["../../../../src/agent/orchestration/agent-orchestrator.ts"],"sourcesContent":["/**\n * Agent Orchestrator - Coordinates Agent execution flow\n *\n * Manages the complete agent execution pipeline from message processing\n * to response generation.\n */\n\nimport type { Agent, AgentMessage } from '@mariozechner/pi-agent-core';\nimport type { Config } from '../../config/schema.js';\nimport type { InboundMessage } from '../../infra/bus/index.js';\nimport type { SessionConfigStore, SessionStore } from '../../session/index.js';\nimport { resolveEffectiveThinkingLevel } from '../../session/thinking-resolve.js';\nimport type { ThinkLevel } from '../transcript/thinking-types.js';\nimport type { ModelManager } from '../models/index.js';\nimport type { SessionContext } from '../session/session-context.js';\nimport type { AgentEventHandler } from './agent-event-handler.js';\nimport type { FeedbackCoordinator } from '../feedback/feedback-coordinator.js';\nimport type { AgentManager } from '../agent-manager.js';\nimport { sanitizeMessages, cleanTrailingErrors } from '../memory/message-sanitizer.js';\nimport {\n  tryApplySessionTranscriptHygiene,\n  tryApplySessionTranscriptHygieneForPersistence,\n} from '../transcript/transcript-hygiene.js';\nimport { createLogger } from '../../utils/logger.js';\nimport { extractAgentUserPlainText } from '../memory/user-message-text.js';\nimport { runAgentTurnWithModelFallbacks } from './run-agent-turn-with-fallbacks.js';\nimport {\n  persistInboundAttachmentsToWorkspace,\n  formatInboundFileTextBlock,\n} from '../../channels/attachments/inbound-persist.js';\nimport { expandAtFileMentionsInPlainText } from '../context/expand-at-file-mentions.js';\nimport { resolveInboundImageContentParts } from '../image/inbound-image-handling.js';\n\nconst log = createLogger('AgentOrchestrator');\n\nexport interface AgentOrchestratorConfig {\n  agentManager: AgentManager;\n  sessionStore: SessionStore;\n  modelManager: ModelManager;\n  eventHandler: AgentEventHandler;\n  feedbackCoordinator: FeedbackCoordinator;\n  sessionConfigStore: SessionConfigStore;\n  /** Load per-session workspace override and mkdir before creating the agent. */\n  hydrateSessionWorkspaceFromStore?: (sessionKey: string) => Promise<void>;\n  getThinkingDefault: () => ThinkLevel | undefined;\n  /** Per-session default from merged `agents.list` / defaults (optional). */\n  getThinkingDefaultForSession?: (sessionKey: string) => ThinkLevel | undefined;\n  /** Default workspace root when no per-session resolver is set. */\n  workspaceRoot: string;\n  /** Per-agent workspace root for attachments (optional; defaults to `workspaceRoot`). */\n  getWorkspaceRootForSession?: (sessionKey: string) => string;\n  /** Agent home (`…/agents/<id>/`) for inbound/TTS files — keeps internal state out of the markdown workspace. */\n  getAgentInternalStorageRootForSession?: (sessionKey: string) => string;\n  /** Fire-and-forget after full session persist (e.g. LLM session title); not called from mid-turn snapshots. */\n  enqueueAutoTitle?: (sessionKey: string) => void;\n  /** For per-turn timeout via `agents.defaults.maxTaskDurationMs`. */\n  getConfig?: () => Config | undefined;\n}\n\nexport class AgentOrchestrator {\n  private agentManager: AgentManager;\n  private sessionStore: SessionStore;\n  private modelManager: ModelManager;\n  private eventHandler: AgentEventHandler;\n  private feedbackCoordinator: FeedbackCoordinator;\n  private sessionConfigStore: SessionConfigStore;\n  private hydrateSessionWorkspaceFromStore?: (sessionKey: string) => Promise<void>;\n  private getThinkingDefault: () => ThinkLevel | undefined;\n  private getThinkingDefaultForSession?: (sessionKey: string) => ThinkLevel | undefined;\n  private workspaceRoot: string;\n  private getWorkspaceRootForSession?: (sessionKey: string) => string;\n  private getAgentInternalStorageRootForSession: (sessionKey: string) => string;\n  private enqueueAutoTitle?: (sessionKey: string) => void;\n  private getConfig?: () => Config | undefined;\n\n  constructor(config: AgentOrchestratorConfig) {\n    this.agentManager = config.agentManager;\n    this.sessionStore = config.sessionStore;\n    this.modelManager = config.modelManager;\n    this.eventHandler = config.eventHandler;\n    this.feedbackCoordinator = config.feedbackCoordinator;\n    this.sessionConfigStore = config.sessionConfigStore;\n    this.hydrateSessionWorkspaceFromStore = config.hydrateSessionWorkspaceFromStore;\n    this.getThinkingDefault = config.getThinkingDefault;\n    this.getThinkingDefaultForSession = config.getThinkingDefaultForSession;\n    this.workspaceRoot = config.workspaceRoot;\n    this.getWorkspaceRootForSession = config.getWorkspaceRootForSession;\n    this.getAgentInternalStorageRootForSession =\n      config.getAgentInternalStorageRootForSession ??\n      ((sk) => this.getWorkspaceRootForSession?.(sk) ?? this.workspaceRoot);\n    this.enqueueAutoTitle = config.enqueueAutoTitle;\n    this.getConfig = config.getConfig;\n  }\n\n  private async hydrateSessionModelFromStore(sessionKey: string): Promise<void> {\n    const cfg = await this.sessionConfigStore.get(sessionKey);\n    if (cfg?.modelOverride) {\n      await this.modelManager.switchModelForSession(sessionKey, cfg.modelOverride);\n    }\n  }\n\n  /**\n   * Process a message through the agent orchestration pipeline\n   */\n  async process(msg: InboundMessage, context: SessionContext): Promise<void> {\n    const { sessionKey } = context;\n\n    log.debug({ sessionKey }, 'Processing message through agent orchestrator');\n\n    await this.hydrateSessionWorkspaceFromStore?.(sessionKey);\n\n    // Get or create agent for this session\n    const agent = this.agentManager.getOrCreateAgent(sessionKey);\n\n    try {\n      await this.hydrateSessionModelFromStore(sessionKey);\n\n      // 1. Load session history\n      let messages = await this.sessionStore.load(sessionKey);\n\n      // Clean any trailing errors from previous sessions (defensive)\n      messages = cleanTrailingErrors(messages);\n\n      try {\n        const model = this.modelManager.getResolvedModelForSession(sessionKey);\n        messages = tryApplySessionTranscriptHygiene(messages, model);\n      } catch (err) {\n        log.warn({ err, sessionKey }, 'Transcript hygiene skipped (model resolve failed)');\n      }\n\n      agent.state.messages = messages;\n\n      // 2. Apply model configuration for session\n      await this.modelManager.applyModelForSession(agent, sessionKey);\n\n      const thinkingDefault =\n        this.getThinkingDefaultForSession?.(sessionKey) ?? this.getThinkingDefault();\n      const thinkingLevel = await resolveEffectiveThinkingLevel(\n        this.sessionConfigStore,\n        sessionKey,\n        null,\n        thinkingDefault,\n      );\n      this.agentManager.setThinkingLevel(sessionKey, thinkingLevel);\n\n      // 3. Persist inbound files (Telegram, etc.) under agent home, then build user message\n      const storageRoot = this.getAgentInternalStorageRootForSession(sessionKey);\n      const persistedAttachments = await persistInboundAttachmentsToWorkspace(\n        storageRoot,\n        sessionKey,\n        msg.attachments,\n      );\n      const userMessage = await this.buildUserMessage(\n        {\n          ...msg,\n          attachments: persistedAttachments ?? msg.attachments,\n        },\n        sessionKey,\n      );\n      const userPlainForMemory = extractAgentUserPlainText(userMessage);\n      const userMessageForModel = await this.agentManager.applyMemoryPrefetchToUserMessage(\n        userMessage,\n        sessionKey,\n      );\n\n      // 4. Start task feedback\n      this.feedbackCoordinator.startTask();\n\n      // 5. Execute agent\n      await this.executeAgent(agent, userMessageForModel, context);\n\n      this.agentManager.afterAgentTurn(sessionKey, userPlainForMemory);\n      this.agentManager.scheduleBackgroundReviewAfterUserTurn(sessionKey);\n\n      // 6. Sanitize messages before saving (remove error messages, empty content)\n      const rawMessages = agent.state.messages;\n      const { messages: sanitizedMessages, removed } = sanitizeMessages(rawMessages);\n\n      if (removed > 0) {\n        log.info({ sessionKey, removed }, 'Removed problematic messages before saving');\n      }\n\n      // 7. Save session messages (transcript hygiene)\n      await this.saveSessionSnapshot(sessionKey, sanitizedMessages);\n\n      this.enqueueAutoTitle?.(sessionKey);\n\n      // 8. End task feedback\n      this.feedbackCoordinator.endTask();\n\n    } catch (error) {\n      log.error({ err: error, sessionKey }, 'Error in agent orchestration');\n      this.feedbackCoordinator.endTask();\n      throw error;\n    }\n  }\n\n  /**\n   * Transcript hygiene + persist. Expects messages already passed through {@link sanitizeMessages}.\n   * Keeps thinking blocks on disk for UI; agent load path applies full hygiene including dropThinking.\n   */\n  private async saveSessionSnapshot(sessionKey: string, messages: AgentMessage[]): Promise<void> {\n    let toPersist = messages;\n    try {\n      const model = this.modelManager.getResolvedModelForSession(sessionKey);\n      toPersist = tryApplySessionTranscriptHygieneForPersistence(messages, model);\n    } catch (err) {\n      log.warn({ err, sessionKey }, 'Transcript hygiene on save skipped');\n    }\n    await this.sessionStore.save(sessionKey, toPersist);\n  }\n\n  /**\n   * Execute the agent with a user message (primary model, then `agents.defaults.model.fallbacks` on failure).\n   */\n  private async executeAgent(\n    agent: Agent,\n    userMessage: AgentMessage,\n    context: SessionContext\n  ): Promise<void> {\n    const sessionKey = context.sessionKey;\n    await runAgentTurnWithModelFallbacks({\n      agent,\n      sessionKey,\n      modelManager: this.modelManager,\n      userMessage,\n      log,\n      getConfig: this.getConfig,\n      beforeUserPrompt: () => this.agentManager.beginBackgroundReviewUserTurn(sessionKey),\n      afterUserPrompt: async () => {\n        try {\n          const { messages: sanitizedTurn } = sanitizeMessages(agent.state.messages);\n          await this.saveSessionSnapshot(sessionKey, sanitizedTurn);\n          log.debug({ sessionKey }, 'User message saved immediately after prompt');\n        } catch (err) {\n          log.warn({ err, sessionKey }, 'Failed to save user message immediately');\n        }\n      },\n    });\n  }\n\n  /**\n   * Build an agent message from an inbound message\n   */\n  private async buildUserMessage(msg: InboundMessage, sessionKey: string): Promise<AgentMessage> {\n    const storageRootAbs = this.getAgentInternalStorageRootForSession(sessionKey);\n    let textBody = msg.content.trimStart().startsWith('/skill:')\n      ? this.agentManager.expandSkillUserText(msg.content)\n      : msg.content;\n\n    if (/@file:/.test(textBody)) {\n      const root = this.agentManager.getResolvedWorkspaceForSession(sessionKey);\n      textBody = await expandAtFileMentionsInPlainText(textBody, root);\n    }\n\n    if (!msg.attachments || msg.attachments.length === 0) {\n      return {\n        role: 'user',\n        content: textBody,\n        timestamp: Date.now(),\n      };\n    }\n\n    const modelRef = this.modelManager.getModelForSession(sessionKey);\n    const cfg = this.getConfig?.();\n\n    const messageContent: Array<\n      { type: 'text'; text: string } | { type: 'image'; data: string; mimeType: string }\n    > = [];\n\n    if (msg.content.trim()) {\n      messageContent.push({ type: 'text', text: textBody });\n    }\n\n    const attachments = msg.attachments;\n    let i = 0;\n    while (i < attachments.length) {\n      const att = attachments[i]!;\n      const isImage =\n        att.type === 'image' || att.type === 'photo' || Boolean(att.mimeType?.startsWith('image/'));\n\n      if (isImage) {\n        const group: Array<{ data: string; mimeType: string }> = [];\n        while (i < attachments.length) {\n          const a = attachments[i]!;\n          const img =\n            a.type === 'image' || a.type === 'photo' || Boolean(a.mimeType?.startsWith('image/'));\n          if (!img) {\n            break;\n          }\n          if (!a.data || a.data.length === 0) {\n            log.warn({ type: a.type, name: a.name }, 'Empty image data, skipping');\n            i += 1;\n            continue;\n          }\n          group.push({ data: a.data, mimeType: a.mimeType || 'image/jpeg' });\n          i += 1;\n        }\n        if (group.length > 0) {\n          const parts = await resolveInboundImageContentParts({\n            modelRef,\n            cfg,\n            userTextForContext: msg.content.trim() ? textBody : '',\n            images: group,\n          });\n          messageContent.push(...parts);\n        }\n      } else {\n        const fileBlock = formatInboundFileTextBlock(\n          {\n            type: att.type,\n            mimeType: att.mimeType,\n            name: att.name,\n            size: att.size,\n            workspaceRelativePath: att.workspaceRelativePath,\n          },\n          storageRootAbs,\n        );\n        messageContent.push({ type: 'text', text: fileBlock });\n        i += 1;\n      }\n    }\n\n    const hasText = messageContent.some((item) => item.type === 'text');\n    const hasImage = messageContent.some((item) => item.type === 'image');\n    if (hasImage && !hasText) {\n      messageContent.unshift({ type: 'text', text: 'Please analyze the image(s) I sent.' });\n    }\n\n    if (messageContent.length === 0) {\n      log.warn(\n        { attachmentCount: msg.attachments.length },\n        'All attachments were skipped, falling back to text message',\n      );\n      return {\n        role: 'user',\n        content: textBody || '[Image attachment could not be processed]',\n        timestamp: Date.now(),\n      };\n    }\n\n    return {\n      role: 'user',\n      content: messageContent,\n      timestamp: Date.now(),\n    };\n  }\n\n  /**\n   * Get the current agent model ID\n   */\n  getCurrentModel(): string {\n    return this.modelManager.getCurrentModel();\n  }\n\n  /**\n   * Check if agent is currently processing for a session\n   */\n  isProcessing(sessionKey: string): boolean {\n    const agent = this.agentManager.getAgent(sessionKey);\n    if (!agent) {\n      return false;\n    }\n    return agent.state.messages.length > 0;\n  }\n\n  /**\n   * Abort current agent execution for a session\n   */\n  abort(sessionKey: string): void {\n    const agent = this.agentManager.getAgent(sessionKey);\n    if (agent) {\n      agent.abort();\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;aAuBqD;AAUrD,MAAM,MAAM,aAAa,oBAAoB;AA0B7C,IAAa,oBAAb,MAA+B;CAC7B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YAAY,QAAiC;AAC3C,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO;AAC3B,OAAK,sBAAsB,OAAO;AAClC,OAAK,qBAAqB,OAAO;AACjC,OAAK,mCAAmC,OAAO;AAC/C,OAAK,qBAAqB,OAAO;AACjC,OAAK,+BAA+B,OAAO;AAC3C,OAAK,gBAAgB,OAAO;AAC5B,OAAK,6BAA6B,OAAO;AACzC,OAAK,wCACH,OAAO,2CACL,OAAO,KAAK,6BAA6B,GAAG,IAAI,KAAK;AACzD,OAAK,mBAAmB,OAAO;AAC/B,OAAK,YAAY,OAAO;;CAG1B,MAAc,6BAA6B,YAAmC;EAC5E,MAAM,MAAM,MAAM,KAAK,mBAAmB,IAAI,WAAW;AACzD,MAAI,KAAK,cACP,OAAM,KAAK,aAAa,sBAAsB,YAAY,IAAI,cAAc;;;;;CAOhF,MAAM,QAAQ,KAAqB,SAAwC;EACzE,MAAM,EAAE,eAAe;AAEvB,MAAI,MAAM,EAAE,YAAY,EAAE,gDAAgD;AAE1E,QAAM,KAAK,mCAAmC,WAAW;EAGzD,MAAM,QAAQ,KAAK,aAAa,iBAAiB,WAAW;AAE5D,MAAI;AACF,SAAM,KAAK,6BAA6B,WAAW;GAGnD,IAAI,WAAW,MAAM,KAAK,aAAa,KAAK,WAAW;AAGvD,cAAW,oBAAoB,SAAS;AAExC,OAAI;IACF,MAAM,QAAQ,KAAK,aAAa,2BAA2B,WAAW;AACtE,eAAW,iCAAiC,UAAU,MAAM;YACrD,KAAK;AACZ,QAAI,KAAK;KAAE;KAAK;KAAY,EAAE,oDAAoD;;AAGpF,SAAM,MAAM,WAAW;AAGvB,SAAM,KAAK,aAAa,qBAAqB,OAAO,WAAW;GAE/D,MAAM,kBACJ,KAAK,+BAA+B,WAAW,IAAI,KAAK,oBAAoB;GAC9E,MAAM,gBAAgB,MAAM,8BAC1B,KAAK,oBACL,YACA,MACA,gBACD;AACD,QAAK,aAAa,iBAAiB,YAAY,cAAc;GAI7D,MAAM,uBAAuB,MAAM,qCADf,KAAK,sCAAsC,WAAW,EAGxE,YACA,IAAI,YACL;GACD,MAAM,cAAc,MAAM,KAAK,iBAC7B;IACE,GAAG;IACH,aAAa,wBAAwB,IAAI;IAC1C,EACD,WACD;GACD,MAAM,qBAAqB,0BAA0B,YAAY;GACjE,MAAM,sBAAsB,MAAM,KAAK,aAAa,iCAClD,aACA,WACD;AAGD,QAAK,oBAAoB,WAAW;AAGpC,SAAM,KAAK,aAAa,OAAO,qBAAqB,QAAQ;AAE5D,QAAK,aAAa,eAAe,YAAY,mBAAmB;AAChE,QAAK,aAAa,sCAAsC,WAAW;GAGnE,MAAM,cAAc,MAAM,MAAM;GAChC,MAAM,EAAE,UAAU,mBAAmB,YAAY,iBAAiB,YAAY;AAE9E,OAAI,UAAU,EACZ,KAAI,KAAK;IAAE;IAAY;IAAS,EAAE,6CAA6C;AAIjF,SAAM,KAAK,oBAAoB,YAAY,kBAAkB;AAE7D,QAAK,mBAAmB,WAAW;AAGnC,QAAK,oBAAoB,SAAS;WAE3B,OAAO;AACd,OAAI,MAAM;IAAE,KAAK;IAAO;IAAY,EAAE,+BAA+B;AACrE,QAAK,oBAAoB,SAAS;AAClC,SAAM;;;;;;;CAQV,MAAc,oBAAoB,YAAoB,UAAyC;EAC7F,IAAI,YAAY;AAChB,MAAI;AAEF,eAAY,+CAA+C,UAD7C,KAAK,aAAa,2BAA2B,WAAW,CACK;WACpE,KAAK;AACZ,OAAI,KAAK;IAAE;IAAK;IAAY,EAAE,qCAAqC;;AAErE,QAAM,KAAK,aAAa,KAAK,YAAY,UAAU;;;;;CAMrD,MAAc,aACZ,OACA,aACA,SACe;EACf,MAAM,aAAa,QAAQ;AAC3B,QAAM,+BAA+B;GACnC;GACA;GACA,cAAc,KAAK;GACnB;GACA;GACA,WAAW,KAAK;GAChB,wBAAwB,KAAK,aAAa,8BAA8B,WAAW;GACnF,iBAAiB,YAAY;AAC3B,QAAI;KACF,MAAM,EAAE,UAAU,kBAAkB,iBAAiB,MAAM,MAAM,SAAS;AAC1E,WAAM,KAAK,oBAAoB,YAAY,cAAc;AACzD,SAAI,MAAM,EAAE,YAAY,EAAE,8CAA8C;aACjE,KAAK;AACZ,SAAI,KAAK;MAAE;MAAK;MAAY,EAAE,0CAA0C;;;GAG7E,CAAC;;;;;CAMJ,MAAc,iBAAiB,KAAqB,YAA2C;EAC7F,MAAM,iBAAiB,KAAK,sCAAsC,WAAW;EAC7E,IAAI,WAAW,IAAI,QAAQ,WAAW,CAAC,WAAW,UAAU,GACxD,KAAK,aAAa,oBAAoB,IAAI,QAAQ,GAClD,IAAI;AAER,MAAI,SAAS,KAAK,SAAS,EAAE;GAC3B,MAAM,OAAO,KAAK,aAAa,+BAA+B,WAAW;AACzE,cAAW,MAAM,gCAAgC,UAAU,KAAK;;AAGlE,MAAI,CAAC,IAAI,eAAe,IAAI,YAAY,WAAW,EACjD,QAAO;GACL,MAAM;GACN,SAAS;GACT,WAAW,KAAK,KAAK;GACtB;EAGH,MAAM,WAAW,KAAK,aAAa,mBAAmB,WAAW;EACjE,MAAM,MAAM,KAAK,aAAa;EAE9B,MAAM,iBAEF,EAAE;AAEN,MAAI,IAAI,QAAQ,MAAM,CACpB,gBAAe,KAAK;GAAE,MAAM;GAAQ,MAAM;GAAU,CAAC;EAGvD,MAAM,cAAc,IAAI;EACxB,IAAI,IAAI;AACR,SAAO,IAAI,YAAY,QAAQ;GAC7B,MAAM,MAAM,YAAY;AAIxB,OAFE,IAAI,SAAS,WAAW,IAAI,SAAS,WAAW,QAAQ,IAAI,UAAU,WAAW,SAAS,CAAC,EAEhF;IACX,MAAM,QAAmD,EAAE;AAC3D,WAAO,IAAI,YAAY,QAAQ;KAC7B,MAAM,IAAI,YAAY;AAGtB,SAAI,EADF,EAAE,SAAS,WAAW,EAAE,SAAS,WAAW,QAAQ,EAAE,UAAU,WAAW,SAAS,CAAC,EAErF;AAEF,SAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,WAAW,GAAG;AAClC,UAAI,KAAK;OAAE,MAAM,EAAE;OAAM,MAAM,EAAE;OAAM,EAAE,6BAA6B;AACtE,WAAK;AACL;;AAEF,WAAM,KAAK;MAAE,MAAM,EAAE;MAAM,UAAU,EAAE,YAAY;MAAc,CAAC;AAClE,UAAK;;AAEP,QAAI,MAAM,SAAS,GAAG;KACpB,MAAM,QAAQ,MAAM,gCAAgC;MAClD;MACA;MACA,oBAAoB,IAAI,QAAQ,MAAM,GAAG,WAAW;MACpD,QAAQ;MACT,CAAC;AACF,oBAAe,KAAK,GAAG,MAAM;;UAE1B;IACL,MAAM,YAAY,2BAChB;KACE,MAAM,IAAI;KACV,UAAU,IAAI;KACd,MAAM,IAAI;KACV,MAAM,IAAI;KACV,uBAAuB,IAAI;KAC5B,EACD,eACD;AACD,mBAAe,KAAK;KAAE,MAAM;KAAQ,MAAM;KAAW,CAAC;AACtD,SAAK;;;EAIT,MAAM,UAAU,eAAe,MAAM,SAAS,KAAK,SAAS,OAAO;AAEnE,MADiB,eAAe,MAAM,SAAS,KAAK,SAAS,QAAQ,IACrD,CAAC,QACf,gBAAe,QAAQ;GAAE,MAAM;GAAQ,MAAM;GAAuC,CAAC;AAGvF,MAAI,eAAe,WAAW,GAAG;AAC/B,OAAI,KACF,EAAE,iBAAiB,IAAI,YAAY,QAAQ,EAC3C,6DACD;AACD,UAAO;IACL,MAAM;IACN,SAAS,YAAY;IACrB,WAAW,KAAK,KAAK;IACtB;;AAGH,SAAO;GACL,MAAM;GACN,SAAS;GACT,WAAW,KAAK,KAAK;GACtB;;;;;CAMH,kBAA0B;AACxB,SAAO,KAAK,aAAa,iBAAiB;;;;;CAM5C,aAAa,YAA6B;EACxC,MAAM,QAAQ,KAAK,aAAa,SAAS,WAAW;AACpD,MAAI,CAAC,MACH,QAAO;AAET,SAAO,MAAM,MAAM,SAAS,SAAS;;;;;CAMvC,MAAM,YAA0B;EAC9B,MAAM,QAAQ,KAAK,aAAa,SAAS,WAAW;AACpD,MAAI,MACF,OAAM,OAAO"}

package/dist/src/agent/prompt/service-prompt-builder.js

@@ -1,6 +1,6 @@
-import { resolveStateDir } from "../../config/paths-state.js";
 import { createLogger } from "../../utils/logger/index.js";
 import { init_logger } from "../../utils/logger.js";
+import { resolveStateDir } from "../../config/paths-state.js";
 import { init_paths } from "../../config/paths.js";
 import { DEFAULT_USER_FILENAME, toWorkspaceBootstrapFile } from "../context/workspace.js";
 import { createSkillConfigManager } from "../skills/config.js";

package/dist/src/agent/service.d.ts

@@ -5,6 +5,7 @@ import type { ChannelManager } from '../channels/manager.js';
 import { type ReasoningLevel } from './transcript/thinking-types.js';
 import { type SkillCatalogEntry } from './agent-manager.js';
 import type { AgentServiceConfig, StreamHandle } from './service.types.js';
+import type { CompactionResult } from './memory/compaction.js';
 export type { AgentServiceConfig, AgentContext, StreamHandle } from './service.types.js';
 export declare class AgentService {
     private sessionStore;
@@ -42,6 +43,7 @@ export declare class AgentService {
     /** Webchat SSE queue pushers for `clarify_request` and similar mid-turn UI events. */
     private webchatSseEnqueueBySession;
     private sessionUnsubscribers;
+    private effectiveAppConfig;
     constructor(bus: MessageBus, config: AgentServiceConfig);
     private attachmentRootsForSession;
     private createSessionStore;
@@ -108,7 +110,20 @@ export declare class AgentService {
         workspaceRelativePath?: string;
     }> | undefined>;
     private endDirectRequestContext;
-    compactSession(sessionKey: string, instructions?: string): Promise<void>;
+    compactSession(sessionKey: string, options?: {
+        instructions?: string;
+        force?: boolean;
+    }): Promise<CompactionResult>;
+    /**
+     * One-shot LLM answer for /btw: uses transcript as background only; does not persist to session.
+     */
+    btwQuery(sessionKey: string, question: string): Promise<{
+        text: string;
+        error?: string;
+    }>;
+    private formatMessagesForBtw;
+    /** Markdown or JSON summary for /context (prompt assembly is approximated from config + transcript stats). */
+    getSessionContextReport(sessionKey: string, mode: 'list' | 'detail' | 'json'): Promise<string>;
     getSessionStats(sessionKey: string, messages: AgentMessage[]): {
         windowStats: {
             total: number;

package/dist/src/agent/service.js

@@ -1,11 +1,12 @@
-import { init_agent_scope, resolveAgentBootstrapDir, resolveAgentHomeDir, resolveDefaultAgentId } from "./agent-scope.js";
-import { getAgentDefaultModelRef, init_schema } from "../config/schema.js";
-import { init_session_key, parseSessionKey } from "../routing/session-key.js";
-import { extractProfileAgentId } from "../config/agent-profile.js";
 import { runWithLogContext, updateAsyncLogContext } from "../utils/logger/context.js";
 import { createLogger } from "../utils/logger/index.js";
 import { init_logger } from "../utils/logger.js";
-import { getDefaultModelSync, init_providers } from "../providers/index.js";
+import { init_agent_scope, resolveAgentBootstrapDir, resolveAgentHomeDir, resolveDefaultAgentId } from "./agent-scope.js";
+import { getAgentDefaultModelRef, init_schema } from "../config/schema.js";
+import { init_session_key, parseSessionKey } from "../routing/session-key.js";
+import { applyConfigOverrides } from "../config/runtime-overrides.js";
+import { extractProfileAgentId, resolveEffectiveAgentProfileForSession } from "../config/agent-profile.js";
+import { getDefaultModelSync, init_providers, resolveModel } from "../providers/index.js";
 import { extractTextContent, loadBootstrapFiles } from "./context/workspace.js";
 import { isTTSAvailable } from "../voice/tts/factory.js";
 import { mergeTtsConfigFromAppConfig } from "../voice/tts/merge-config.js";
@@ -58,6 +59,7 @@ import { SessionLifecycleManager } from "./session/session-lifecycle.js";
 import "./session/index.js";
 import { tryApplySessionTranscriptHygiene, tryApplySessionTranscriptHygieneForPersistence } from "./transcript/transcript-hygiene.js";
 import { runAgentTurnWithModelFallbacks } from "./orchestration/run-agent-turn-with-fallbacks.js";
+import { expandAtFileMentionsInPlainText } from "./context/expand-at-file-mentions.js";
 import { resolveInboundImageContentParts } from "./image/inbound-image-handling.js";
 import { AgentOrchestrator } from "./orchestration/agent-orchestrator.js";
 import { AgentEventHandler } from "./orchestration/agent-event-handler.js";
@@ -69,6 +71,7 @@ import { runProcessDirectStreaming } from "./service/process-direct-streaming.js
 import { shouldSilence } from "../heartbeat/tokens.js";
 import { createTypingController } from "./lifecycle/typing.js";
 import { persistOutboundTtsAudio } from "../channels/attachments/outbound-tts-persist.js";
+import { complete } from "@mariozechner/pi-ai";
 import { mkdir } from "node:fs/promises";
 //#region src/agent/service.ts
 init_schema();
@@ -113,6 +116,10 @@ var AgentService = class {
 	/** Webchat SSE queue pushers for `clarify_request` and similar mid-turn UI events. */
 	webchatSseEnqueueBySession = /* @__PURE__ */ new Map();
 	sessionUnsubscribers = /* @__PURE__ */ new Map();
+	effectiveAppConfig() {
+		const base = this.config.config;
+		return base ? applyConfigOverrides(base) : void 0;
+	}
 	constructor(bus, config) {
 		this.bus = bus;
 		this.config = config;
@@ -186,8 +193,8 @@ var AgentService = class {
 			feedbackCoordinator: this.feedbackCoordinator,
 			sessionConfigStore: this.sessionConfigStore,
 			hydrateSessionWorkspaceFromStore: (sessionKey) => this.hydrateSessionWorkspaceFromStore(sessionKey),
-			getConfig: () => this.config.config,
-			getThinkingDefault: () => this.config.config?.agents?.defaults?.thinkingDefault,
+			getConfig: () => this.effectiveAppConfig(),
+			getThinkingDefault: () => this.effectiveAppConfig()?.agents?.defaults?.thinkingDefault,
 			getThinkingDefaultForSession: (sessionKey) => this.agentManager.getThinkingDefaultForSession(sessionKey),
 			workspaceRoot: this.workspaceDir,
 			getWorkspaceRootForSession: (sessionKey) => this.agentManager.getResolvedWorkspaceForSession(sessionKey),
@@ -211,7 +218,10 @@ var AgentService = class {
 			abortSessionTurn: async (sessionKey) => {
 				await this.streamManager.abort();
 				this.agentOrchestrator.abort(sessionKey);
-			}
+			},
+			compactSession: (sessionKey, options) => this.compactSession(sessionKey, options),
+			btwQuery: (sessionKey, question) => this.btwQuery(sessionKey, question),
+			getSessionContextReport: (sessionKey, mode) => this.getSessionContextReport(sessionKey, mode)
 		});
 		this.sessionLifecycleManager = new SessionLifecycleManager(this.sessionStore, this.sessionTracker, this.lifecycleManager);
 		if (!!!process.env.ELECTRON_RUN_AS_NODE) {
@@ -525,12 +535,20 @@ var AgentService = class {
 	}
 	async buildMessageContent(content, attachments, sessionKey) {
 		const messageContent = [];
-		if (content.trim()) messageContent.push({
-			type: "text",
-			text: content
-		});
-		if (!attachments?.length) return messageContent;
 		const sk = sessionKey ?? "";
+		if (content.trim()) {
+			let textPart = content;
+			if (/@file:/.test(textPart)) {
+				const wsKey = sk !== "" ? sk : "cli:direct";
+				const root = this.agentManager.getResolvedWorkspaceForSession(wsKey);
+				textPart = await expandAtFileMentionsInPlainText(textPart, root);
+			}
+			messageContent.push({
+				type: "text",
+				text: textPart
+			});
+		}
+		if (!attachments?.length) return messageContent;
 		const modelRef = sk !== "" ? this.modelManager.getModelForSession(sk) : getAgentDefaultModelRef(this.config.config) ?? getDefaultModelSync(this.config.config);
 		const cfg = this.config.config;
 		const storageRoot = sk !== "" ? resolveAgentHomeDir(this.config.config, extractProfileAgentId(sk, this.config.config)) : resolveAgentHomeDir(this.config.config, resolveDefaultAgentId(this.config.config));
@@ -585,15 +603,155 @@ var AgentService = class {
 		this.feedbackCoordinator.clearContext();
 		this.contextMiddleware.onResponse();
 	}
-	async compactSession(sessionKey, instructions) {
+	async compactSession(sessionKey, options) {
 		const messages = await this.sessionStore.load(sessionKey);
 		const contextWindow = this.getContextWindow();
-		const result = await this.sessionStore.compact(sessionKey, messages, contextWindow, instructions);
-		if (result.compacted) await this.sessionStore.save(sessionKey, await this.sessionStore.load(sessionKey));
+		const result = await this.sessionStore.compact(sessionKey, messages, contextWindow, options?.instructions, options?.force ?? true);
+		if (result.compacted) {
+			await this.sessionStore.save(sessionKey, await this.sessionStore.load(sessionKey));
+			this.agentManager.removeAgent(sessionKey);
+		}
 		log.info({
 			sessionKey,
 			result
 		}, "Manual compaction complete");
+		return result;
+	}
+	/**
+	* One-shot LLM answer for /btw: uses transcript as background only; does not persist to session.
+	*/
+	async btwQuery(sessionKey, question) {
+		const q = question.trim();
+		if (!q) return {
+			text: "",
+			error: "Empty question."
+		};
+		const messages = await this.sessionStore.load(sessionKey);
+		const modelRef = this.modelManager.getModelForSession(sessionKey);
+		let model;
+		try {
+			model = resolveModel(modelRef);
+		} catch (err) {
+			const em = err instanceof Error ? err.message : String(err);
+			log.warn({
+				err,
+				modelRef,
+				errorMessage: em
+			}, "btwQuery: model resolve failed");
+			return {
+				text: "",
+				error: `Could not resolve model: ${modelRef}`
+			};
+		}
+		const background = this.formatMessagesForBtw(messages.slice(-40));
+		const userMessage = {
+			role: "user",
+			content: [
+				[
+					"You are answering an ephemeral /btw side question about the current conversation.",
+					"Use the conversation only as background context.",
+					"Answer only the side question. Do not continue or complete any unfinished task from the conversation.",
+					"Do not use tools, shell, or file writes unless the question explicitly requires a tiny code snippet.",
+					"If the question can be answered briefly, answer briefly."
+				].join("\n"),
+				"",
+				"---",
+				"Conversation background (read-only):",
+				background || "(empty)",
+				"",
+				"---",
+				"Side question:",
+				q
+			].join("\n"),
+			timestamp: Date.now()
+		};
+		const controller = new AbortController();
+		const timeoutId = setTimeout(() => controller.abort(), 9e4);
+		try {
+			const out = await complete(model, { messages: [userMessage] }, {
+				maxTokens: 2048,
+				temperature: .4,
+				signal: controller.signal
+			});
+			return { text: (Array.isArray(out.content) ? out.content.filter((c) => c.type === "text" && typeof c.text === "string").map((c) => c.text || "").join("") : "").trim() };
+		} catch (err) {
+			const em = err instanceof Error ? err.message : String(err);
+			log.warn({
+				err,
+				sessionKey,
+				errorMessage: em
+			}, "btwQuery failed");
+			return {
+				text: "",
+				error: em
+			};
+		} finally {
+			clearTimeout(timeoutId);
+		}
+	}
+	formatMessagesForBtw(messages) {
+		return messages.map((m) => {
+			const role = m.role;
+			let body = "";
+			if (typeof m.content === "string") body = m.content;
+			else if (Array.isArray(m.content)) body = m.content.filter((c) => c.type === "text").map((c) => c.text || "").join("\n");
+			const line = `[${role}]: ${body}`;
+			return line.length > 4e3 ? `${line.slice(0, 4e3)}…` : line;
+		}).join("\n\n");
+	}
+	/** Markdown or JSON summary for /context (prompt assembly is approximated from config + transcript stats). */
+	async getSessionContextReport(sessionKey, mode) {
+		const messages = await this.sessionStore.load(sessionKey);
+		const cw = this.getContextWindow();
+		const stats = this.getSessionStats(sessionKey, messages);
+		const cfg = this.effectiveAppConfig() ?? this.config.config;
+		const model = this.modelManager.getModelForSession(sessionKey);
+		const workspace = effectiveWorkspacePathForSession(cfg, sessionKey, await this.sessionConfigStore.get(sessionKey));
+		const estTokens = await this.sessionStore.estimateTokenUsage(sessionKey, messages);
+		const profile = resolveEffectiveAgentProfileForSession(cfg, sessionKey);
+		const defaults = cfg.agents?.defaults;
+		const compaction = defaults?.compaction;
+		const tools = defaults?.tools;
+		const toolsSummary = tools && typeof tools === "object" ? Object.entries(tools).filter(([, v]) => v === true).map(([k]) => k).slice(0, 16).join(", ") || "(none explicitly true)" : "(see agents.defaults.tools in config)";
+		const payload = {
+			sessionKey,
+			model,
+			workspacePath: workspace,
+			agentId: profile.agentId,
+			messageCount: messages.length,
+			contextWindowNominal: cw,
+			estimatedTranscriptTokens: estTokens,
+			approxWindowUsage: cw > 0 ? estTokens / cw : null,
+			thinkingDefault: defaults?.thinkingDefault,
+			reasoningDefault: defaults?.reasoningDefault,
+			verboseDefault: defaults?.verboseDefault,
+			compaction,
+			toolsFlagsOn: toolsSummary,
+			windowStats: stats.windowStats,
+			compactionRunStats: stats.compactionStats
+		};
+		if (mode === "json") return JSON.stringify(payload, null, 2);
+		const lines = [
+			"📎 *Context overview*",
+			"",
+			`• Session: \`${sessionKey}\``,
+			`• Model: \`${model}\``,
+			`• Agent profile: \`${profile.agentId}\``,
+			`• Workspace: \`${workspace}\``,
+			`• Messages: ${messages.length}`,
+			`• Est. transcript tokens (rough): ${estTokens}`,
+			`• Nominal context budget (4× maxTokens): ${cw}`
+		];
+		if (cw > 0) lines.push(`• Approx. usage vs budget: ${(estTokens / cw * 100).toFixed(1)}%`);
+		lines.push(`• Thinking / reasoning / verbose defaults: ${defaults?.thinkingDefault ?? "—"} / ${defaults?.reasoningDefault ?? "—"} / ${defaults?.verboseDefault ?? "—"}`, `• Tools (true flags): ${toolsSummary}`, "", "_Full system prompt, skills, and memory blocks are assembled at agent runtime; use Web settings or logs for deep inspection._");
+		if (mode === "detail") {
+			lines.push("", "*Compaction config (agents.defaults.compaction):*", "```json");
+			lines.push(JSON.stringify(compaction ?? {}, null, 2));
+			lines.push("```", "", "*Window stats:*", "```json");
+			lines.push(JSON.stringify(stats.windowStats ?? {}, null, 2));
+			lines.push("```");
+		}
+		return lines.join("\n");
 	}
 	getSessionStats(sessionKey, messages) {
 		return {
@@ -603,14 +761,14 @@ var AgentService = class {
 		};
 	}
 	async applyResolvedThinkingLevel(sessionKey, requestOverride) {
-		const def = this.config.config?.agents?.defaults?.thinkingDefault;
+		const def = this.effectiveAppConfig()?.agents?.defaults?.thinkingDefault;
 		const level = await resolveEffectiveThinkingLevel(this.sessionConfigStore, sessionKey, requestOverride, def);
 		this.agentManager.setThinkingLevel(sessionKey, level);
 	}
 	/** Resolved thinking level and effective model ref for a session (Web UI). */
 	async getSessionAgentConfig(sessionKey) {
 		await this.hydrateSessionModelFromStore(sessionKey);
-		const cfg = this.config.config;
+		const cfg = this.effectiveAppConfig();
 		const sc = await this.sessionConfigStore.get(sessionKey);
 		const defThink = cfg.agents?.defaults?.thinkingDefault ?? "medium";
 		const level = await resolveEffectiveThinkingLevel(this.sessionConfigStore, sessionKey, null, defThink);
@@ -786,7 +944,7 @@ var AgentService = class {
 			prepareLoadedSessionMessages: (sk, msgs) => this.prepareLoadedSessionMessages(sk, msgs),
 			modelManager: this.modelManager,
 			applyResolvedThinkingLevel: (sk, t) => this.applyResolvedThinkingLevel(sk, t),
-			getConfig: () => this.config.config,
+			getConfig: () => this.effectiveAppConfig(),
 			sessionConfigStore: this.sessionConfigStore,
 			attachmentRootsForSession: (sk) => this.attachmentRootsForSession(sk),
 			agentOrchestrator: this.agentOrchestrator,
@@ -1112,7 +1270,7 @@ var AgentService = class {
 			reason: prep.stats?.reason,
 			usagePercent: prep.stats?.usagePercent
 		}, "Session needs compaction");
-		const result = await this.sessionStore.compact(sessionKey, messages, contextWindow);
+		const result = await this.sessionStore.compact(sessionKey, messages, contextWindow, void 0, false);
 		await this.hookHandler.trigger("after_compaction", {
 			messageCount: messages.length,
 			tokenCount: result.tokensBefore,