AI Code Companion & Chat Assistant

import { GoogleGenAI } from '@google/genai';
import { NextRequest } from 'next/server';

const ai = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });

export async function POST(req: NextRequest) {
  const { messages, model } = await req.json();
  
  // Create a text stream chunk response
  const responseStream = await ai.models.generateContentStream({
    model: model || 'gemini-2.5-flash',
    contents: messages,
    config: {
      systemInstruction: "You are CodeCompanion, an elite software architect. Output clear code block tags."
    }
  });

  const encoder = new TextEncoder();
  const readable = new ReadableStream({
    async start(controller) {
      for await (const chunk of responseStream) {
        const text = chunk.text;
        controller.enqueue(encoder.encode(text));
      }
      controller.close();
    }
  });

  return new Response(readable, {
    headers: { 'Content-Type': 'text/plain; charset=utf-8' }
  });
}