Function extractUsage

src/node/services/devToolsMiddleware.ts:195–239

(
  usage: LanguageModelV3Usage | Record<string, unknown> | null | undefined
)

}

export function extractUsage(
  usage: LanguageModelV3Usage | Record<string, unknown> | null | undefined
): DevToolsUsage | null {
  if (!usage || !isRecord(usage)) {
    return null;
  }

  const inputTokens =
    extractInputTokens(usage.inputTokens) ?? extractOptionalTokenCount(usage, "promptTokens");
  const outputTokens =
    extractOutputTokens(usage.outputTokens) ?? extractOptionalTokenCount(usage, "completionTokens");
  const explicitTotalTokens = extractOptionalTokenCount(usage, "totalTokens");
  const inputTokenTotal = extractTokenTotal(inputTokens);
  const outputTokenTotal = extractTokenTotal(outputTokens);

  const totalTokens =
    typeof explicitTotalTokens === "number"
      ? explicitTotalTokens
      : inputTokenTotal !== undefined || outputTokenTotal !== undefined
        ? (inputTokenTotal ?? 0) + (outputTokenTotal ?? 0)
        : undefined;

  const raw = Object.prototype.hasOwnProperty.call(usage, "raw") ? usage.raw : undefined;

  if (
    inputTokens === undefined &&
    outputTokens === undefined &&
    typeof totalTokens !== "number" &&
    raw === undefined
  ) {
    return null;
  }

  const normalizedUsage: DevToolsUsage = {
    inputTokens,
    outputTokens,
    totalTokens,
  };

  if (raw !== undefined) {
    normalizedUsage.raw = raw;
  }

  return normalizedUsage;
}

export function createDevToolsMiddleware(
  workspaceId: string,