...

src/lib/actions/ai.ts

@@ -0,0 +1,62 @@
+"use server";
+
+import { format } from "util";
+
+async function callZhipuAPI(
+  messages: { role: string; content: string }[],
+  model = process.env.ZHIPU_MODEL_NAME,
+) {
+  const url = "https://open.bigmodel.cn/api/paas/v4/chat/completions";
+
+  const response = await fetch(url, {
+    method: "POST",
+    headers: {
+      Authorization: "Bearer " + process.env.ZHIPU_API_KEY,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model: model,
+      messages: messages,
+      temperature: 0.2,
+      thinking: {
+        type: "disabled",
+      },
+    }),
+  });
+
+  if (!response.ok) {
+    throw new Error(`API 调用失败: ${response.status} ${response.statusText}`);
+  }
+
+  return await response.json();
+}
+
+export async function getLLMAnswer(prompt: string) {
+  return (
+    await callZhipuAPI([
+      {
+        role: "user",
+        content: prompt,
+      },
+    ])
+  ).choices[0].message.content.trim() as string;
+}
+
+export async function simpleGetLLMAnswer(
+  prompt: string,
+  searchParams: URLSearchParams,
+  args: string[],
+) {
+  if (args.some((arg) => typeof searchParams.get(arg) !== "string")) {
+    return Response.json({
+      status: "error",
+      message: "Missing required parameters",
+    });
+  }
+  return Response.json({
+    status: "success",
+    message: await getLLMAnswer(
+      format(prompt, ...args.map((v) => searchParams.get(v))),
+    ),
+  });
+}

src/lib/actions/services/folderService.ts

@@ -0,0 +1,68 @@
+"use server";
+
+import {
+  folderCreateInput,
+  folderUpdateInput,
+} from "../../../../generated/prisma/models";
+import prisma from "../../db";
+
+export async function getFoldersByOwner(owner: string) {
+  const folders = await prisma.folder.findMany({
+    where: {
+      owner: owner,
+    },
+  });
+  return folders;
+}
+
+export async function getFoldersWithTotalPairsByOwner(owner: string) {
+  const folders = await prisma.folder.findMany({
+    where: {
+      owner: owner
+    },
+    include: {
+      text_pair: {
+        select: {
+          id: true
+        }
+      }
+    }
+  });
+
+  return folders.map(folder => ({
+    ...folder,
+    total_pairs: folder.text_pair.length
+  }));
+}
+
+export async function createFolder(folder: folderCreateInput) {
+  await prisma.folder.create({
+    data: folder,
+  });
+}
+
+export async function deleteFolderById(id: number) {
+  await prisma.folder.delete({
+    where: {
+      id: id,
+    },
+  });
+}
+
+export async function updateFolderById(id: number, data: folderUpdateInput) {
+  await prisma.folder.update({
+    where: {
+      id: id,
+    },
+    data: data,
+  });
+}
+
+export async function getOwnerByFolderId(id: number) {
+  const folder = await prisma.folder.findUnique({
+    where: {
+      id: id,
+    },
+  });
+  return folder?.owner;
+}

src/lib/actions/services/textPairService.ts

@@ -0,0 +1,51 @@
+"use server";
+
+import {
+  text_pairCreateInput,
+  text_pairUpdateInput,
+} from "../../../../generated/prisma/models";
+import prisma from "../../db";
+
+export async function createTextPair(data: text_pairCreateInput) {
+  await prisma.text_pair.create({
+    data: data,
+  });
+}
+
+export async function deleteTextPairById(id: number) {
+  await prisma.text_pair.delete({
+    where: {
+      id: id,
+    },
+  });
+}
+
+export async function updateTextPairById(
+  id: number,
+  data: text_pairUpdateInput,
+) {
+  await prisma.text_pair.update({
+    where: {
+      id: id,
+    },
+    data: data,
+  });
+}
+
+export async function getTextPairCountByFolderId(folderId: number) {
+  const count = await prisma.text_pair.count({
+    where: {
+      folder_id: folderId,
+    },
+  });
+  return count;
+}
+
+export async function getTextPairsByFolderId(folderId: number) {
+  const textPairs = await prisma.text_pair.findMany({
+    where: {
+      folder_id: folderId,
+    },
+  });
+  return textPairs;
+}

src/lib/actions/translatorActions.ts

@@ -1,6 +1,6 @@
 "use server";
 
-import { getLLMAnswer } from "../ai";
+import { getLLMAnswer } from "./ai";
 
 export const genIPA = async (text: string) => {
   return (