Function postChatCompletions

web/src/app/api/v1/chat/completions/_post.ts:194–1107

(params: {
  req: NextRequest
  getUserInfoFromApiKey: GetUserInfoFromApiKeyFn
  logger: Logger
  loggerWithContext: LoggerWithContextFn
  trackEvent: TrackEventFn
  getUserUsageData: GetUserUsageDataFn
  getAgentRunFromId: GetAgentRunFromIdFn
  fetch: typeof globalThis.fetch
  insertMessageBigquery: InsertMessageBigqueryFn
  insertChatCompletionTraceBigquery?: InsertChatCompletionTraceBigqueryFn
  ensureSubscriberBlockGrant?: (params: {
    userId: string
    logger: Logger
  }) => Promise<BlockGrantResult | null>
  getUserPreferences?: GetUserPreferencesFn
  /** Optional override for the freebuff waiting-room gate. Defaults to the
   *  real check backed by Postgres; tests inject a no-op. */
  checkSessionAdmissible?: CheckSessionAdmissibleFn
  /** Optional override for the free-mode rate limiter. Tests inject this to
   *  avoid coupling to process-global limiter state. */
  checkFreeModeRateLimit?: CheckFreeModeRateLimitFn
  /** Optional override for country/cache checks. Tests inject this to avoid
   *  coupling to Postgres-backed cache state. */
  resolveFreeModeCountryAccess?: ResolveFreeModeCountryAccessFn
  /** Optional override for releasing stale waiting-room rows on hard blocks. */
  endFreebuffSession?: EndUserSessionFn
})

}

export async function postChatCompletions(params: {
  req: NextRequest
  getUserInfoFromApiKey: GetUserInfoFromApiKeyFn
  logger: Logger
  loggerWithContext: LoggerWithContextFn
  trackEvent: TrackEventFn
  getUserUsageData: GetUserUsageDataFn
  getAgentRunFromId: GetAgentRunFromIdFn
  fetch: typeof globalThis.fetch
  insertMessageBigquery: InsertMessageBigqueryFn
  insertChatCompletionTraceBigquery?: InsertChatCompletionTraceBigqueryFn
  ensureSubscriberBlockGrant?: (params: {
    userId: string
    logger: Logger
  }) => Promise<BlockGrantResult | null>
  getUserPreferences?: GetUserPreferencesFn
  /** Optional override for the freebuff waiting-room gate. Defaults to the
   *  real check backed by Postgres; tests inject a no-op. */
  checkSessionAdmissible?: CheckSessionAdmissibleFn
  /** Optional override for the free-mode rate limiter. Tests inject this to
   *  avoid coupling to process-global limiter state. */
  checkFreeModeRateLimit?: CheckFreeModeRateLimitFn
  /** Optional override for country/cache checks. Tests inject this to avoid
   *  coupling to Postgres-backed cache state. */
  resolveFreeModeCountryAccess?: ResolveFreeModeCountryAccessFn
  /** Optional override for releasing stale waiting-room rows on hard blocks. */
  endFreebuffSession?: EndUserSessionFn
}) {
  const {
    req,
    getUserInfoFromApiKey,
    loggerWithContext,
    getUserUsageData,
    getAgentRunFromId,
    fetch,
    insertMessageBigquery,
    insertChatCompletionTraceBigquery,
    ensureSubscriberBlockGrant,
    getUserPreferences,
    checkSessionAdmissible: checkSession = checkSessionAdmissible,
    checkFreeModeRateLimit = defaultCheckFreeModeRateLimit,
    resolveFreeModeCountryAccess,
    endFreebuffSession = endUserSession,
  } = params
  let { logger } = params
  let { trackEvent } = params
  const resolveCountryAccess: ResolveFreeModeCountryAccessFn =
    resolveFreeModeCountryAccess ??
    ((userId, req, options) =>
      getCachedFreeModeCountryAccess({ userId, req, options, logger }))

  try {
    // Parse request body
    let body: Record<string, unknown>
    try {
      body = await req.json()
    } catch (error) {
      trackEvent({