76 lines
2.5 KiB
TypeScript
76 lines
2.5 KiB
TypeScript
|
import { generateObject, generateText, type LanguageModel } from "ai";
|
|||
|
|
import type { z } from "zod";
|
|||
|
|
import { extractJsonFromText } from "./coerce-research-map";
|
|||
|
|
import {
|
|||
|
|
getOpenCodeGenerationSettings,
|
|||
|
|
prefersOpenCodeTextFirst,
|
|||
|
|
} from "./opencode-go-settings";
|
|||
|
|
|
|||
|
|
export interface GenerateStructuredObjectOptions<S extends z.ZodTypeAny> {
|
|||
|
|
model: LanguageModel;
|
|||
|
|
schema: S;
|
|||
|
|
system?: string;
|
|||
|
|
prompt: string;
|
|||
|
|
provider: string;
|
|||
|
|
modelId: string;
|
|||
|
|
jsonPromptSuffix?: string;
|
|||
|
|
normalize?: (value: unknown) => unknown;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export async function generateStructuredObject<S extends z.ZodTypeAny>(
|
|||
|
|
options: GenerateStructuredObjectOptions<S>
|
|||
|
|
): Promise<z.infer<S>> {
|
|||
|
|
const settings = getOpenCodeGenerationSettings(options.provider, options.modelId);
|
|||
|
|
const preferText = prefersOpenCodeTextFirst(options.provider, options.modelId);
|
|||
|
|
const jsonSuffix =
|
|||
|
|
options.jsonPromptSuffix ?? "\n\n只回傳 JSON，不要 markdown 或額外說明。\n\n強制：所有文字欄位必須使用繁體中文（台灣用語），絕對禁止簡體字。";
|
|||
|
|
|
|||
|
|
async function viaObject() {
|
|||
|
|
const { object } = await generateObject({
|
|||
|
|
model: options.model,
|
|||
|
|
schema: options.schema,
|
|||
|
|
system: `${options.system ?? ""}\n\n強制：所有文字欄位必須使用繁體中文（台灣用語），絕對禁止簡體字。`,
|
|||
|
|
prompt: options.prompt,
|
|||
|
|
...settings,
|
|||
|
|
});
|
|||
|
|
return options.schema.parse(options.normalize ? options.normalize(object) : object);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function viaText(extra = "") {
|
|||
|
|
const { text } = await generateText({
|
|||
|
|
model: options.model,
|
|||
|
|
system: options.system,
|
|||
|
|
prompt: `${options.prompt}${jsonSuffix}${extra}`,
|
|||
|
|
...settings,
|
|||
|
|
});
|
|||
|
|
let parsed = extractJsonFromText(text);
|
|||
|
|
if (Array.isArray(parsed) && parsed.length > 0) {
|
|||
|
|
parsed = parsed[0];
|
|||
|
|
}
|
|||
|
|
return options.schema.parse(options.normalize ? options.normalize(parsed) : parsed);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const attempts: Array<() => Promise<z.infer<S>>> = preferText
|
|||
|
|
? [
|
|||
|
|
() => viaText(),
|
|||
|
|
() => viaObject(),
|
|||
|
|
() => viaText("\n\n上次格式不完整，請務必回傳合法 JSON。"),
|
|||
|
|
]
|
|||
|
|
: [
|
|||
|
|
() => viaObject(),
|
|||
|
|
() => viaText(),
|
|||
|
|
() => viaText("\n\n上次格式不完整，請務必回傳合法 JSON。"),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
let lastError: unknown;
|
|||
|
|
for (const attempt of attempts) {
|
|||
|
|
try {
|
|||
|
|
return await attempt();
|
|||
|
|
} catch (error) {
|
|||
|
|
lastError = error;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
throw lastError instanceof Error ? lastError : new Error("結構化 AI 回傳失敗");
|
|||
|
|
}
|