AI Interview Platform

import { NextRequest, NextResponse } from "next/server";
import { getSessionState, updateSessionState } from "@/lib/db";
import { generateNextQuestion } from "@/lib/gemini";

export async function POST(req: NextRequest) {
  const payload = await req.json();
  
  if (payload.message.type === "assistant-request") {
    const { sessionId, transcript } = payload.message;
    const session = await getSessionState(sessionId);
    
    // Retrieve resume context and rubrics from ChromaDB RAG
    const ragContext = await getRAGContext(session.candidateId, transcript);
    
    // Generate next question using Gemini
    const { responseText, nextState } = await generateNextQuestion({
      currentState: session.state,
      history: session.history,
      userResponse: transcript,
      context: ragContext
    });
    
    await updateSessionState(sessionId, {
      state: nextState,
      history: [...session.history, { role: "user", text: transcript }, { role: "assistant", text: responseText }]
    });
    
    return NextResponse.json({
      response: responseText
    });
  }
  
  return NextResponse.json({ status: "ignored" });
}

import { ChromaClient } from "chromadb";
import { getGeminiEmbeddings } from "./gemini";

const chroma = new ChromaClient();

export async function getRAGContext(candidateId: string, transcript: string) {
  const collection = await chroma.getCollection({ name: `candidate-${candidateId}-data` });
  const queryEmbedding = await getGeminiEmbeddings(transcript);
  
  const results = await collection.query({
    queryEmbeddings: [queryEmbedding],
    nResults: 3
  });
  
  return results.documents.flat().join("\n");
}