Skip to main content
Loading navigation
Back to prompt library

Configure AI SDK with Novita AI Endpoint

Linked challenge: Build Consumer AI Agent Session Risk Scoring with AI SDK and Novita AI

Prompt source

implementation19 lines3 sectionsNo variables0 checklist items
import { createOpenAI } from '@ai-sdk/openai';
import { generateObject } from 'ai';
import { z } from 'z WordSchema';

const novita = createOpenAI({
  baseURL: 'https://api.novita.ai/v3/openai',
  apiKey: process.env.NOVITA_API_KEY,
});

export async function evaluateSessionRisk(sessionData: any) {
  const result = await generateObject({
    model: novita('meta-llama/llama-3.1-70b-instruct'),
    schema: z.object({
      threat_score: z.number().min(0).max(1),
      action_required: z.string(),
      is_autonomous_agent: z.boolean(),
    }),
    prompt: `Analyze behavioral session telemetry: ${JSON.stringify(sessionData)}`,
  });
  return result.object;
}