59 lines
1.7 KiB
TypeScript
59 lines
1.7 KiB
TypeScript
import {AiProvider} from "../model/ai-provider";
|
|
import type {StoredAttachment} from "../model/stored-attachment";
|
|
import {persistInternalJsonArtifactAttachment} from "./internal-artifact-store";
|
|
|
|
export async function persistFinalTextArtifactAttachment(params: {
|
|
provider: AiProvider;
|
|
model: string;
|
|
text: string;
|
|
chatId: number;
|
|
messageId: number;
|
|
}): Promise<StoredAttachment | undefined> {
|
|
const text = params.text.trim();
|
|
if (!text) return Promise.resolve(undefined);
|
|
|
|
return await persistInternalJsonArtifactAttachment({
|
|
artifactKind: "final_text",
|
|
fileNamePrefix: "final-text",
|
|
chatId: params.chatId,
|
|
messageId: params.messageId,
|
|
payload: {
|
|
provider: params.provider,
|
|
model: params.model,
|
|
text,
|
|
},
|
|
metadata: {
|
|
provider: params.provider,
|
|
model: params.model,
|
|
textChars: text.length,
|
|
},
|
|
});
|
|
}
|
|
|
|
export async function persistErrorArtifactAttachment(params: {
|
|
provider: AiProvider;
|
|
model: string;
|
|
message: string;
|
|
recoverable: boolean;
|
|
chatId: number;
|
|
messageId: number;
|
|
}): Promise<StoredAttachment> {
|
|
return await persistInternalJsonArtifactAttachment({
|
|
artifactKind: "error",
|
|
fileNamePrefix: "error",
|
|
chatId: params.chatId,
|
|
messageId: params.messageId,
|
|
payload: {
|
|
provider: params.provider,
|
|
model: params.model,
|
|
message: params.message,
|
|
recoverable: params.recoverable,
|
|
},
|
|
metadata: {
|
|
provider: params.provider,
|
|
model: params.model,
|
|
recoverable: params.recoverable,
|
|
},
|
|
});
|
|
}
|