import sqlite3
from openai import OpenAI

client = OpenAI()
PROMPT_RULE = 'Extract a VIN. If no VIN is present or you are not confident, reply UNKNOWN. Reply with only the VIN or UNKNOWN.'
VALID_VINS = {'1HGCM82633A004352', '5YJ3E1EA7HF000337'}

db = sqlite3.connect(':memory:')
db.execute('CREATE TABLE vins (vin TEXT PRIMARY KEY)')
db.executemany('INSERT INTO vins VALUES (?)', ((vin,) for vin in VALID_VINS))

def propose_vin(text):
    response = client.chat.completions.create(model='gpt-4o-mini', messages=[{'role': 'system', 'content': PROMPT_RULE}, {'role': 'user', 'content': text}], temperature=0)
    return response.choices[0].message.content.strip().upper()

def cross_check(vin):
    if vin == 'UNKNOWN':
        return False
    row = db.execute('SELECT 1 FROM vins WHERE vin = ?', (vin,)).fetchone()
    return row is not None

text = input('Paste text with a VIN: ')
vin = propose_vin(text)
print('AI proposes:', vin)
if not cross_check(vin):
    print('Rejected: missing or unknown VIN.')
elif input('Approve this VIN? [y/n]: ').strip().lower() == 'y':
    print('Verified and approved.')
else:
    print('Rejected by human.')

import { createOpenAI } from '@ai-sdk/openai'
import { createWorkersAI } from '@ai-sdk/cloudflare-workers-ai'
import { generateText } from 'ai'
import { defineEventHandler, readBody } from 'h3'

const gateway = createOpenAI({
  apiKey: process.env.GATEWAY_API_KEY,
  baseURL: process.env.GATEWAY_URL,
})

export default defineEventHandler(async (event) => {
  const workersAI = createWorkersAI({
    binding: event.context.cloudflare?.env.AI,
  })

  const model = workersAI('@cf/meta/llama-3.1-8b-instruct')
  // Switch back: const model = gateway('gpt-4o-mini')

  const body = await readBody(event)
  const { text } = await generateText({
    model,
    prompt: body.prompt,
  })

  return { text }
})