Function decodeUsageQueryInput

packages/runtime-host/src/protocol/usage-pricing.ts:327–373

(value: unknown)

} as const;

export function decodeUsageQueryInput(value: unknown): UsageQueryInput {
  const input = requireRecord(value, 'usage query input');
  if (input.kind === 'summary') {
    const exact = requireExactRecord(input, 'usage summary input', ['kind', 'query']);
    return { kind: 'summary', query: decodeLlmUsageQuery(exact.query) };
  }
  if (input.kind === 'buckets') {
    assertOptionalExactKeys(
      input,
      'usage buckets input',
      ['kind', 'query', 'groupBy'],
      ['offset', 'limit'],
    );
    const groupBy = decodeUsageGroupBy(input.groupBy);
    const page = { offset: decodeOffset(input.offset), limit: decodeLimit(input.limit) };
    return groupBy === 'tool'
      ? { kind: 'buckets', query: decodeToolUsageQuery(input.query), groupBy, ...page }
      : { kind: 'buckets', query: decodeLlmUsageQuery(input.query), groupBy, ...page };
  }
  if (input.kind === 'logs') {
    assertOptionalExactKeys(
      input,
      'usage logs input',
      ['kind', 'source', 'query'],
      ['offset', 'limit'],
    );
    const page = { offset: decodeOffset(input.offset), limit: decodeLimit(input.limit) };
    if (input.source === 'llm') {
      return {
        kind: 'logs',
        source: 'llm',
        query: decodeLlmUsageQuery(input.query),
        ...page,
      };
    }
    if (input.source === 'tool') {
      return {
        kind: 'logs',
        source: 'tool',
        query: decodeToolUsageQuery(input.query),
        ...page,
      };
    }
    throw invalidProtocolFrame('Invalid usage log source');
  }
  throw invalidProtocolFrame('Invalid usage query kind');
}

export function decodeUsageQueryResult(value: unknown): UsageQueryResult {
  const result = requireRecord(value, 'usage query result');