Function createReplOnQuery

src/session/replTurnController.ts:69–170

(deps: ReplOnQueryDeps)

}

export function createReplOnQuery(deps: ReplOnQueryDeps): ReplOnQueryFn {
  return async function onQuery(
    newMessages,
    abortController,
    shouldQuery,
    additionalAllowedTools,
    mainLoopModel,
    onBeforeQueryCallback,
    input,
    effort,
  ): Promise<void> {
    if (isAgentSwarmsEnabled()) {
      const teamName = getTeamName()
      const agentName = getAgentName()
      if (teamName && agentName) {
        void setMemberActive(teamName, agentName, true)
      }
    }

    const thisGeneration = deps.queryGuard.tryStart()
    if (thisGeneration === null) {
      logEvent('ncode_concurrent_onquery_detected', {})

      newMessages
        .filter((message): message is UserMessage => {
          return message.type === 'user' && !message.isMeta
        })
        .map(message => getContentText(message.message.content))
        .filter((message): message is string => message !== null)
        .forEach((message, index) => {
          deps.enqueuePrompt(message)
          if (index === 0) {
            logEvent('ncode_concurrent_onquery_enqueued', {})
          }
        })
      return
    }

    let caughtError: unknown
    try {
      deps.resetTimingRefs()
      deps.setMessages(oldMessages => [...oldMessages, ...newMessages])
      deps.responseLengthRef.current = 0
      if (feature('TOKEN_BUDGET')) {
        const parsedBudget = input ? parseTokenBudget(input) : null
        snapshotOutputTokensForTurn(
          parsedBudget ?? getCurrentTurnTokenBudget(),
        )
      }
      deps.apiMetricsRef.current = []
      deps.setStreamingToolUses([])
      deps.setStreamingText(null)

      const latestMessages = deps.messagesRef.current
      if (input) {
        await deps.mrOnBeforeQuery(input, latestMessages, newMessages.length)
      }