Function handleReportUsage

server/src/training-agent/task-handlers.ts:3375–3468

(args: ToolArgs, ctx: TrainingContext)

}

export async function handleReportUsage(args: ToolArgs, ctx: TrainingContext) {
  const req = args as unknown as ReportUsageArgs;
  const session = await getSession(sessionKeyFromArgs(req, ctx.mode, ctx.userId, ctx.moduleId));

  if (!req.reporting_period || !req.usage?.length) {
    return { errors: [{ code: 'INVALID_USAGE_DATA', message: 'reporting_period and at least one usage record are required.' }] };
  }

  if (session.usageRecords.length + req.usage.length > MAX_USAGE_RECORDS_PER_SESSION) {
    return { errors: [{ code: 'LIMIT_EXCEEDED', message: `Usage record limit (${MAX_USAGE_RECORDS_PER_SESSION}) would be exceeded.` }] };
  }

  let accepted = 0;
  const errors: Array<{ code: string; message: string; field?: string }> = [];

  for (let i = 0; i < req.usage.length; i++) {
    const record = req.usage[i];

    // Validate required fields
    if (record.vendor_cost === undefined || record.vendor_cost === null) {
      errors.push({ code: 'INVALID_USAGE_DATA', message: 'vendor_cost is required.', field: `usage[${i}].vendor_cost` });
      continue;
    }
    if (record.vendor_cost < 0) {
      errors.push({ code: 'INVALID_USAGE_DATA', message: 'vendor_cost must be non-negative.', field: `usage[${i}].vendor_cost` });
      continue;
    }
    if (!record.currency) {
      errors.push({ code: 'INVALID_USAGE_DATA', message: 'currency is required.', field: `usage[${i}].currency` });
      continue;
    }
    if (!record.account) {
      errors.push({ code: 'INVALID_USAGE_DATA', message: 'account is required.', field: `usage[${i}].account` });
      continue;
    }
    if (record.impressions !== undefined && record.impressions < 0) {
      errors.push({ code: 'INVALID_USAGE_DATA', message: 'impressions must be non-negative.', field: `usage[${i}].impressions` });
      continue;
    }

    // Validate creative_id exists if provided
    if (record.creative_id) {
      const creative = session.creatives.get(record.creative_id) ?? getComplianceCreative(record.creative_id);
      if (!creative) {
        errors.push({ code: 'NOT_FOUND', message: `Creative "${record.creative_id}" not found in session.`, field: `usage[${i}].creative_id` });
        continue;
      }

      // Validate pricing_option_id matches if provided
      if (record.pricing_option_id && creative.pricingOptionId && record.pricing_option_id !== creative.pricingOptionId) {
        errors.push({
          code: 'INVALID_PRICING_OPTION',
          message: `pricing_option_id mismatch: expected ${creative.pricingOptionId}, received ${record.pricing_option_id}`,
          field: `usage[${i}].pricing_option_id`,
        });
        continue;
      }
    }