DGAI: Create embedding api route - ENG-291

apps/website/app/api/embeddings/openai/small/route.ts

@@ -0,0 +1,86 @@
+import { NextRequest, NextResponse } from "next/server";
+import OpenAI from "openai";
+import cors from "~/utils/llm/cors";
+
+const apiKey = process.env.OPENAI_API_KEY;
+
+if (!apiKey) {
+  console.error(
+    "Missing OPENAI_API_KEY environment variable. The embeddings API will not function.",
+  );
+}
+
+const openai = apiKey ? new OpenAI({ apiKey }) : null;
+
+type RequestBody = {
+  input: string | string[];
+};
+
+const OPENAI_REQUEST_TIMEOUT_MS = 30000;
+
+export async function POST(req: NextRequest): Promise<NextResponse> {
+  let response: NextResponse;
+
+  if (!apiKey) {
+    response = NextResponse.json(
+      {
+        error: "Server configuration error.",
+        details: "Embeddings service is not configured.",
+      },
+      { status: 500 },
+    );
+    return cors(req, response) as NextResponse;
+  }
+
+  try {
+    const body: RequestBody = await req.json();
+    const { input } = body;
+
+    if (!input || (Array.isArray(input) && input.length === 0)) {
+      response = NextResponse.json(
+        { error: "Input text cannot be empty." },
+        { status: 400 },
+      );
+      return cors(req, response) as NextResponse;
+    }
+
+    const options: OpenAI.EmbeddingCreateParams = {
+      model: "text-embedding-3-small",
+      input: input,
+      encoding_format: "float",
+      dimensions: 1536,
+    };
+
+    const embeddingsPromise = openai!.embeddings.create(options);
+    const timeoutPromise = new Promise((_, reject) =>
+      setTimeout(
+        () => reject(new Error("OpenAI API request timeout")),
+        OPENAI_REQUEST_TIMEOUT_MS,
+      ),
+    );
+
+    const openAIResponse = await Promise.race([
+      embeddingsPromise,
+      timeoutPromise,
+    ]);
+
+    response = NextResponse.json(openAIResponse, { status: 200 });
+  } catch (error: unknown) {
+    console.error("Error calling OpenAI Embeddings API:", error);
+    const errorMessage =
+      error instanceof Error ? error.message : "Unknown error";
+    response = NextResponse.json(
+      {
+        error: "Failed to generate embeddings.",
+        details: errorMessage,
+      },
+      { status: 500 },
+    );
+  }
+
+  return cors(req, response) as NextResponse;
+}
+
+export async function OPTIONS(req: NextRequest): Promise<NextResponse> {
+  return cors(req, new NextResponse(null, { status: 204 })) as NextResponse;
+}

apps/website/package.json

@@ -15,6 +15,7 @@
     "@sindresorhus/slugify": "^2.2.1",
     "gray-matter": "^4.0.3",
     "next": "^15.0.3",
+    "openai": "^4.97.0",
     "react": "19.0.0-rc-66855b96-20241106",
     "react-dom": "19.0.0-rc-66855b96-20241106",
     "rehype-parse": "^9.0.1",