xueteng/video-flow-b
1
0
Fork 0
forked from 77media/video-flow
Code Pull Requests Activity

update transformMessage

Browse Source
This commit is contained in:
Zixin Zhou 2025-08-28 22:11:20 +08:00
parent 1af5fda76a
commit 3c57131443
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
components/SmartChatBox/api.ts
View File

@ -208,7 +208,7 @@ function transformSystemMessage(
*/
function transformMessage(apiMessage: RealApiMessage): ChatMessage {
try {
const { id, role, content, created_at, function_name, custom_data, status, intent_type, error_message } = apiMessage;
const { id, role, content, created_at, function_name, custom_data, status, intent_type } = apiMessage;
let message: ChatMessage = {
id: id ? id.toString() : Date.now().toString(),
role: role,
@ -218,7 +218,8 @@ function transformMessage(apiMessage: RealApiMessage): ChatMessage {
status: status || 'success',
};
if (error_message && error_message === 'no enough credits') {
const errorMessage = custom_data?.error_message;
if (errorMessage && errorMessage === 'no enough credits') {
message.blocks = NoEnoughCreditsMessageBlocks;
} else {
if (role === 'assistant' || role === 'user') {

9
components/SmartChatBox/types.ts
View File

@ -131,14 +131,19 @@ export interface ApiMessageContent {
url?: string;
}
interface BaseCustomData {
error_message?: string;
rejected_reason?: string;
[key: string]: any;
}
export interface RealApiMessage {
created_at: string;
id: number;
role: Role;
content: string;
function_name?: FunctionName;
custom_data?: ProjectInit | ScriptSummary | CharacterGeneration | SketchGeneration | ShotSketchGeneration | ShotVideoGeneration;
custom_data?: (ProjectInit | ScriptSummary | CharacterGeneration | SketchGeneration | ShotSketchGeneration | ShotVideoGeneration) & BaseCustomData;
status: MessageStatus;
intent_type: IntentType;
error_message?: string;
}