82f3f64ba1
Co-authored-by: 藍+85CD <50108258+kwaa@users.noreply.github.com> Co-authored-by: junkwarrior87 <115852752+junkwarrior87@users.noreply.github.com>
75 lines
2.1 KiB
TypeScript
75 lines
2.1 KiB
TypeScript
import type { Card } from '../define'
|
|
import type { CharacterCardV3 } from './types/character_card_v3'
|
|
|
|
/**
|
|
* Exports a Card object to CharacterCardV3 format
|
|
* @param data The card data to export
|
|
* @returns A CharacterCardV3 compatible object
|
|
*/
|
|
export function exportToJSON(data: Card): CharacterCardV3 {
|
|
return {
|
|
spec: 'chara_card_v3',
|
|
spec_version: '3.0',
|
|
data: createCardData(data),
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Creates the data portion of a CharacterCardV3 object
|
|
* @param data Source card data
|
|
* @returns The formatted card data
|
|
*/
|
|
function createCardData(data: Card): CharacterCardV3['data'] {
|
|
return {
|
|
name: data.name,
|
|
nickname: data.nickname,
|
|
description: data.description ?? '', // TODO: improve description // FIXME: important
|
|
personality: '', // TODO: add personality
|
|
scenario: '', // TODO: add scenario
|
|
first_mes: data.greetings?.[0] ?? '',
|
|
alternate_greetings: data.greetings?.slice(1) ?? [],
|
|
group_only_greetings: data.greetingsGroupOnly ?? [],
|
|
character_version: data.version,
|
|
creator: data.creator ?? '',
|
|
creator_notes: data.notes ?? '',
|
|
creator_notes_multilingual: data.notesMultilingual,
|
|
system_prompt: '', // TODO: add system_prompt
|
|
post_history_instructions: '', // TODO: add post_history_instructions
|
|
mes_example: formatMessageExample(data.messageExample),
|
|
tags: [], // TODO: add tags
|
|
extensions: createExtensions(data),
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Formats message examples into the required string format
|
|
* @param messageExample The message example array
|
|
* @returns Formatted message example string
|
|
*/
|
|
function formatMessageExample(messageExample: string[][] | undefined): string {
|
|
if (!messageExample)
|
|
return ''
|
|
|
|
return messageExample
|
|
.map(arr => `<START>\n${arr.join('\n')}`)
|
|
.join('\n')
|
|
}
|
|
|
|
/**
|
|
* Creates the extensions object with default values and user extensions
|
|
* @param data Source card data
|
|
* @returns Extensions object
|
|
*/
|
|
function createExtensions(data: Card): Record<string, any> {
|
|
return {
|
|
depth_prompt: {
|
|
depth: 4,
|
|
prompt: '',
|
|
role: 'system',
|
|
},
|
|
fav: false,
|
|
talkativeness: 0.5,
|
|
...data.extensions,
|
|
}
|
|
}
|