✨ 实现了增量翻译

Author: Gaubee

package.json

@@ -4,6 +4,8 @@
   "description": "dweb-browser开发者文档",
   "type": "module",
   "devDependencies": {
+    "@ai-sdk/deepseek": "^0.1.5",
+    "@ai-sdk/google": "^1.1.4",
     "@gaubee/util": "^0.21.0",
     "@plaoc/is-dweb": "^0.1.2",
     "@plaoc/plugins": "^1.1.1",
@@ -12,8 +14,11 @@
     "@shikijs/vitepress-twoslash": "^2.1.0",
     "@std/cli": "npm:@jsr/std__cli@^1.0.11",
     "@std/path": "npm:@jsr/std__path@^1.0.8",
+    "@types/diff": "^7.0.0",
     "@types/node": "^22.10.6",
+    "ai": "^4.1.7",
     "autoprefixer": "^10.4.19",
+    "diff": "^7.0.0",
     "dotenv": "^16.4.7",
     "front-matter": "^4.0.2",
     "gray-matter": "^4.0.3",

scripts/ai-model.ts

@@ -0,0 +1,19 @@
+import { z } from "zod";
+
+export const google = async () => {
+  const { google, createGoogleGenerativeAI } = await import("@ai-sdk/google");
+  return createGoogleGenerativeAI({
+    apiKey: z
+      .string({ required_error: "required env.GOOGLE_API_KEY" })
+      .parse(process.env.GOOGLE_API_KEY),
+  })("gemini-1.5-pro");
+};
+
+export const deepseek = async () => {
+  const { createDeepSeek } = await import("@ai-sdk/deepseek");
+  return createDeepSeek({
+    apiKey: z
+      .string({ required_error: "required env.OPENAI_API_KEY" })
+      .parse(process.env.OPENAI_API_KEY),
+  })("deepseek-chat");
+};

scripts/i18n-ai-core.ts

@@ -0,0 +1,67 @@
+import { OpenAI } from "openai";
+import { z } from "zod";
+import fs from "node:fs";
+import path from "node:path";
+const system_prompt = fs.readFileSync(
+  path.resolve(import.meta.dirname, "system-prompt.en.md"),
+  "utf-8"
+);
+export const getOpenaiOutput = async <T>(
+  user_content: string,
+  schema: z.ZodType<T>
+) => {
+  const openai = new OpenAI({
+    baseURL: z
+      .string({ required_error: "required env.AI_GATEWAY_ENDPOINT" })
+      .parse(process.env.AI_GATEWAY_ENDPOINT),
+    apiKey: z
+      .string({ required_error: "required env.OPENAI_API_KEY" })
+      .parse(process.env.OPENAI_API_KEY),
+  });
+
+  const stream = await openai.chat.completions.create({
+    messages: [
+      {
+        role: "system",
+        content: system_prompt,
+      },
+      {
+        role: "user",
+        content: user_content,
+      },
+    ],
+    model: "deepseek-chat",
+    response_format: {
+      type: "json_object",
+    },
+    stream: true,
+  });
+
+  let res = "";
+  for await (const chunk of stream) {
+    const part = chunk.choices[0].delta.content ?? "";
+    res += part;
+    process.stdout.write(part);
+  }
+  return schema.parse(JSON.parse(res));
+};
+import { type LanguageModelV1, streamObject } from "ai";
+
+export const getAiOutput = async <T>(
+  model: LanguageModelV1,
+  schema: z.ZodType<T>,
+  user_content: string
+) => {
+  const stream = streamObject({
+    model,
+    schema,
+    system: system_prompt,
+    prompt: user_content,
+  });
+  for await (const partialObject of stream.partialObjectStream) {
+    console.clear();
+    console.log(partialObject);
+  }
+
+  return await stream.object;
+};