import OpenAI from "openai";
import { Query } from "objectiveai";

export type Category =
  | "Job Application"
  | "Inquiry"
  | "Meeting Request"
  | "Invoice"
  | "Requires Human Review";

export async function categorizeEmail(emailContent: string): Promise<Category> {
  const openai = new OpenAI({
    apiKey: "<YOUR_OBJECTIVE_AI_API_KEY>",
    baseURL: "https://api.objective-ai.io",
  });
  const request = {
    messages: [
      {
        role: "user",
        content: `Select the correct category for the following email:\n\n${emailContent}`,
      },
    ],
    model: "objectiveai/7GgWgWm7LLuk1LBd5UmYsh",
    response_format: {
      type: "json_schema",
      json_schema: {
        name: "email_categorization",
        strict: true,
        schema: {
          type: "object",
          properties: {
            category: {
              type: "string",
              enum: [
                "Job Application",
                "Inquiry",
                "Meeting Request",
                "Invoice"
              ],
            },
          },
          required: ["category"],
        },
      },
    },
    select_deterministic: true,
  };
  const completion = await Query.Completions.create(openai, request);
  const winner = completion.choices[0];
  const category = JSON.parse(winner.message.content).category as Category;
  if (winner.confidence < 0.8) {
    return "Requires Human Review";
  } else {
    return category;
  }
}