supermemory/apps/web/app/api/chat/route.ts

import { type NextRequest } from "next/server";
import {
  ChatHistory,
  ChatHistoryZod,
  convertChatHistoryList,
  SourcesFromApi,
} from "@repo/shared-types";
import { ensureAuth } from "../ensureAuth";
import { z } from "zod";

export const runtime = "edge";

export async function POST(req: NextRequest) {
  const session = await ensureAuth(req);

  if (!session) {
    return new Response("Unauthorized", { status: 401 });
  }

  if (!process.env.BACKEND_SECURITY_KEY) {
    return new Response("Missing BACKEND_SECURITY_KEY", { status: 500 });
  }

  const url = new URL(req.url);

  const query = url.searchParams.get("q");
  const spaces = url.searchParams.get("spaces");

  const sourcesOnly = url.searchParams.get("sourcesOnly") ?? "false";

  const jsonRequest = (await req.json()) as {
    chatHistory: ChatHistory[];
    sources: SourcesFromApi[] | undefined;
  };
  const { chatHistory, sources } = jsonRequest;

  if (!query || query.trim.length < 0) {
    return new Response(JSON.stringify({ message: "Invalid query" }), {
      status: 400,
    });
  }

  const validated = z.array(ChatHistoryZod).safeParse(chatHistory ?? []);

  if (!validated.success) {
    return new Response(
      JSON.stringify({
        message: "Invalid chat history",
        error: validated.error,
      }),
      { status: 400 },
    );
  }

  const modelCompatible = await convertChatHistoryList(validated.data);

  const resp = await fetch(
    `${process.env.BACKEND_BASE_URL}/api/chat?query=${query}&user=${session.user.id}&sourcesOnly=${sourcesOnly}&spaces=${spaces}`,
    {
      headers: {
        Authorization: `Bearer ${process.env.BACKEND_SECURITY_KEY}`,
        "Content-Type": "application/json",
      },
      method: "POST",
      body: JSON.stringify({
        chatHistory: modelCompatible,
        sources,
      }),
    },
  );

  if (sourcesOnly == "true") {
    const data = (await resp.json()) as SourcesFromApi;
    return new Response(JSON.stringify(data), { status: 200 });
  }

  if (resp.status !== 200 || !resp.ok) {
    const errorData = await resp.text();
    console.log(errorData);
    return new Response(
      JSON.stringify({ message: "Error in CF function", error: errorData }),
      { status: resp.status },
    );
  }

  return new Response(resp.body, { status: 200 });
}