dekun
101 lines
2.8 KiB
TypeScript
101 lines
2.8 KiB
TypeScript
"use server";
|
|
import { streamText } from "ai";
|
|
import { createOpenAI } from "@ai-sdk/openai";
|
|
import { createStreamableValue } from "ai/rsc";
|
|
import { ERROR_PREFIX } from "@/lib/constant";
|
|
import {
|
|
extractChangeDetails,
|
|
extractZhangMingRen,
|
|
readGuaMarkdown,
|
|
} from "@/lib/content/zhouyi";
|
|
|
|
const model =
|
|
process.env.OPENAI_MODEL ?? "huihui_ai/gemma-4-abliterated:e4b";
|
|
const openai = createOpenAI({
|
|
baseURL: process.env.OPENAI_BASE_URL ?? "https://op.bz121.com/v1",
|
|
});
|
|
|
|
const STREAM_INTERVAL = 60;
|
|
const MAX_SIZE = 6;
|
|
|
|
export async function getAnswer(
|
|
prompt: string,
|
|
guaMark: string,
|
|
guaTitle: string,
|
|
guaResult: string,
|
|
guaChange: string,
|
|
) {
|
|
console.log(prompt, guaTitle, guaResult, guaChange);
|
|
const stream = createStreamableValue();
|
|
try {
|
|
const guaDetail = await readGuaMarkdown(guaMark);
|
|
const explain = extractZhangMingRen(guaDetail) ?? "";
|
|
const changeList = extractChangeDetails(guaDetail, guaChange, guaTitle);
|
|
|
|
const { fullStream } = streamText({
|
|
temperature: 0.5,
|
|
model: openai(model),
|
|
messages: [
|
|
{
|
|
role: "system",
|
|
content: `你是一位精通《周易》的AI助手，根据用户提供的卦象和问题，提供准确的卦象解读和实用建议
|
|
任务要求：逻辑清晰，语气得当
|
|
1. 解读卦象：分析主卦、变爻及变卦，解读整体趋势和吉凶
|
|
2. 关联问题：针对用户问题，结合卦象信息，提供具体分析
|
|
3. 提供建议：根据卦象启示，给出切实可行的建议，帮助用户解决实际问题`,
|
|
},
|
|
{
|
|
role: "user",
|
|
content: `我摇到的卦象：${guaTitle} ${guaResult} ${guaChange}
|
|
我的问题：${prompt}
|
|
|
|
${explain}
|
|
${changeList.join("\n")}`,
|
|
},
|
|
],
|
|
maxRetries: 0,
|
|
});
|
|
|
|
let buffer = "";
|
|
let done = false;
|
|
const intervalId = setInterval(() => {
|
|
if (done && buffer.length === 0) {
|
|
clearInterval(intervalId);
|
|
stream.done();
|
|
return;
|
|
}
|
|
if (buffer.length <= MAX_SIZE) {
|
|
stream.update(buffer);
|
|
buffer = "";
|
|
} else {
|
|
const chunk = buffer.slice(0, MAX_SIZE);
|
|
buffer = buffer.slice(MAX_SIZE);
|
|
stream.update(chunk);
|
|
}
|
|
}, STREAM_INTERVAL);
|
|
|
|
(async () => {
|
|
for await (const part of fullStream) {
|
|
switch (part.type) {
|
|
case "text-delta":
|
|
buffer += part.textDelta;
|
|
break;
|
|
case "error":
|
|
const err = part.error as any;
|
|
stream.update(ERROR_PREFIX + (err.message ?? err.toString()));
|
|
break;
|
|
}
|
|
}
|
|
})()
|
|
.catch(console.error)
|
|
.finally(() => {
|
|
done = true;
|
|
});
|
|
|
|
return { data: stream.value };
|
|
} catch (err: any) {
|
|
stream.done();
|
|
return { error: err.message ?? err };
|
|
}
|
|
}
|