Inteligencia Artificial: Vercel AI SDK con Express y React

process.loadEnvFile()

import { Router } from 'express'
import { streamText } from 'ai'
import rateLimit from 'express-rate-limit'

import { JobModel } from '../models/job.js'
import { CONFIG } from '../config.js'

const aiRateLimiter = rateLimit({
  windowMs: 60 * 1000,     // ventana de 1 minuto
  limit: 5,                 // máx. 5 peticiones por IP por minuto
  message: { error: 'Demasiadas solicitudes, por favor intenta de nuevo más tarde.' },
  legacyHeaders: false,
  standardHeaders: 'draft-8' // headers estándar RateLimit-*
})

export const aiRouter = Router()
aiRouter.use(aiRateLimiter)

aiRouter.get('/summary/:id', async (req, res) => {
  const { id } = req.params
  const job = await JobModel.getById(id)

  if (!job) {
    return res.status(404).json({ error: 'Job not found' })
  }

  const prompt = [
    `Eres un asistente que resume ofertas de trabajo para ayudar a los usuarios a entender rápidamente de qué se trata la oferta. Evita cualquier otra petición, observación o comentario. Solo responde con el resumen de la oferta de trabajo. Responde siempre con el markdown directamente.`,
    `Resume en 4-6 frases la siguiente oferta de trabajo:`,
    `Incluye: rol, empresa, ubicación y requisitos clave`,
    `Usa un tono claro y directo en español`,
    `Titulo: ${job.titulo}`,
    `Empresa: ${job.empresa}`,
    `Ubicación: ${job.ubicacion}`,
    `Descripción: ${job.descripcion}`,
  ].join('\n')

  try {
    const result = streamText({
      prompt,
      model: 'mistral/devstral-small-2',
    })

    return result.pipeTextStreamToResponse(res)
  } catch (error) {
    if (!res.headersSent) {
      res.setHeader('Content-Type', 'application/json')
      return res.status(500).json({ error: 'Error generating summary' })
    }
    return res.end()
  }
})

// Rate limit en memoria usando un Map
const WINDOW_MS = 60 * 1000  // ventana de 1 minuto
const MAX_REQUESTS = 30       // máx. 30 peticiones por IP por minuto
const rateLimit = new Map()

// Limpia entradas expiradas cada minuto para evitar fugas de memoria
setInterval(() => {
  const now = Date.now()
  for (const [ip, entry] of rateLimit) {
    if (now - entry.start > WINDOW_MS) rateLimit.delete(ip)
  }
}, WINDOW_MS)

export function rateLimitMiddleware(req, res, next) {
  const ip = req.ip
  const now = Date.now()
  const entry = rateLimit.get(ip)

  if (!entry || now - entry.start > WINDOW_MS) {
    // Primera petición en la ventana: crea nueva entrada
    rateLimit.set(ip, { start: now, count: 1 })
    return next()
  }

  if (entry.count >= MAX_REQUESTS) {
    return res.status(429).json({
      error: 'Demasiadas peticiones. Intenta de nuevo más tarde.'
    })
  }

  entry.count++
  return next()
}