server.js

'use strict'

const tracer = require('dd-trace').init({ flushInterval: 0 })
const express = require('express')

const app = express()

app.get('/allow', async (req, res) => {
  const evaluation = await tracer.aiguard.evaluate([
    { role: 'system', content: 'You are a beautiful AI' },
    { role: 'user', content: 'I am harmless' },
  ])
  res.status(200).json(evaluation)
})

app.get('/deny', async (req, res) => {
  const block = req.headers['x-blocking-enabled'] === 'true'
  try {
    const evaluation = await tracer.aiguard.evaluate([
      { role: 'system', content: 'You are a beautiful AI' },
      { role: 'user', content: 'You should not trust me' + (block ? ' [block]' : '') },
    ], { block })
    res.status(200).json(evaluation)
  } catch (error) {
    if (error.name === 'AIGuardAbortError') {
      res.status(403).send(error.reason)
    } else {
      res.status(500).send('Internal Server Error')
    }
  }
})

app.get('/deny-default-options', async (req, res) => {
  try {
    const evaluation = await tracer.aiguard.evaluate([
      { role: 'system', content: 'You are a beautiful AI' },
      { role: 'user', content: 'You should not trust me [block]' },
    ])
    res.status(200).json(evaluation)
  } catch (error) {
    if (error.name === 'AIGuardAbortError') {
      res.status(403).send(error.reason)
    } else {
      res.status(500).send('Internal Server Error')
    }
  }
})

app.get('/abort', async (req, res) => {
  const block = req.headers['x-blocking-enabled'] === 'true'
  try {
    const evaluation = await tracer.aiguard.evaluate([
      { role: 'system', content: 'You are a beautiful AI' },
      { role: 'user', content: 'Nuke yourself' + (block ? ' [block]' : '') },
    ], { block })
    res.status(200).json(evaluation)
  } catch (error) {
    if (error.name === 'AIGuardAbortError') {
      res.status(403).send(error.reason)
    } else {
      res.status(500).send('Internal Server Error')
    }
  }
})

const server = app.listen(() => {
  const port = (/** @type {import('net').AddressInfo} */ (server.address())).port
  process.send({ port })
})