Function createAIRouterAttemptRequest

src/server/model/aiRouter.ts:1520–1544

(args: {
  req: Request;
  node: AIRouterAttemptNode;
  payload: Record<string, unknown>;
})

}

export function createAIRouterAttemptRequest(args: {
  req: Request;
  node: AIRouterAttemptNode;
  payload: Record<string, unknown>;
}): AIRouterAttemptRequestContext {
  const attemptReq = Object.create(args.req) as Request &
    AIGatewayLogAwareRequest;
  let capturedGatewayLogId: string | undefined;

  attemptReq.params = {
    ...args.req.params,
    gatewayId: args.node.gatewayId,
  };
  attemptReq.body = args.payload;
  attemptReq.__aiGatewayLogPromise = undefined;
  attemptReq.__onAIGatewayLogCreated = (log) => {
    capturedGatewayLogId = log.id;
  };

  return {
    attemptReq,
    getGatewayLogId: () =>
      getBufferedAIGatewayLogId(attemptReq, capturedGatewayLogId),
  };
}

export function buildBufferedAIGatewayAttemptResult(args: {
  protocol?: AIRouterProtocol;