import { GoogleCustomSearch } from "openai-function-calling-tools";
import { LLMError, LLMStream } from './stream';

const handler = async (req, res) => {
  try {
    const googleCustomSearch = new GoogleCustomSearch({
        apiKey: process.env.API_KEY,
        googleCSEId: process.env.CONTEXT_KEY
    });
    const messages = [
        {
          role: "user",
          content: "What are some fun things to do in Seattle, WA this weekend?"
        },
      ];
    
    const functions = {
        googleCustomSearch
    };

    let promptToSend = "You are a helpful assistant";
    const stream = await LLMStream({ id: "gpt-3.5-turbo-0613" }, promptToSend, 0.8, messages, functions);
    return new Response(stream);
  } catch (error) {
    console.error(error);
    if (error instanceof LLMError) {
      res.status(500).send({ error: error.message });
    } else {
      res.status(500).send({ error: 'An error occurred' });
    }
  }
};

export default handler;