Method getRefusalUsageSnapshot

src/node/services/streamManager.ts:1990–2028

(streamInfo: WorkspaceStreamInfo)

  }

  private async getRefusalUsageSnapshot(streamInfo: WorkspaceStreamInfo): Promise<{
    usage?: LanguageModelV2Usage;
    providerMetadata?: Record<string, unknown>;
  }> {
    let usage = hasTokenUsage(streamInfo.cumulativeUsage) ? streamInfo.cumulativeUsage : undefined;
    let providerMetadata = streamInfo.cumulativeProviderMetadata
      ? { ...streamInfo.cumulativeProviderMetadata }
      : undefined;

    if (!usage) {
      const streamMeta = await this.getStreamMetadata(streamInfo);
      usage = hasTokenUsage(streamMeta.totalUsage) ? streamMeta.totalUsage : undefined;
      providerMetadata = providerMetadata ?? streamMeta.contextProviderMetadata;
    }

    if (!usage) {
      return {};
    }

    // Refused attempts never reach the normal stream-end path, so backfill
    // provider-omitted reasoningTokens before snapshotting usage for the
    // refusing model. reasoningBackfillStartIndex excludes earlier refused hops
    // when a later fallback also refuses after preserved parts.
    await this.backfillReasoningTokensFromParts(streamInfo, usage);

    providerMetadata =
      providerMetadata ??
      (streamInfo.cumulativeProviderMetadata
        ? { ...streamInfo.cumulativeProviderMetadata }
        : await this.getAggregatedProviderMetadata(streamInfo));

    return {
      usage: cloneUsage(usage),
      providerMetadata: markProviderMetadataCostsIncluded(
        providerMetadata ? { ...providerMetadata } : undefined,
        streamInfo.initialMetadata?.costsIncluded
      ),
    };
  }

  /**
   * Attribute a refused fallback attempt's token usage to the refusing model