Fix #4312 mostly functional chub import

979fe60ebca7ff529f8b5b0a6e2649ab72b7c9f0

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: 0e5b509dc321661489907e2c188230b9e67cbe8b

src/endpoints/content-manager.js+38 -15 Viewed

 async function downloadChubCharacter(id) {
     const [creatorName, projectName] = id.split('/');
     const result = await fetch(`https://api.chub.ai/api/characters/${creatorName}/${projectName}?full=true`, {
         method: 'GET',
         headers: { 'Accept': 'application/json', 'User-Agent': USER_AGENT },
     });

     /** @type {any} */
     const metadata = await result.json();
     const downloadUrl{ definition, topics } = metadata.node?.max_res_url;
++
 +    /** @type {TavernCardV2} */
 +    const characterCard = {
 +        data: {
 +            name: definition.name,
 +            description: definition.personality,
 +            personality: definition.tavern_personality,
 +            scenario: definition.scenario,
 +            first_mes: definition.first_message,
 +            mes_example: definition.example_dialogs,
 +            creator_notes: definition.description,
 +            system_prompt: definition.system_prompt,
 +            post_history_instructions: definition.post_history_instructions,
 +            alternate_greetings: definition.alternate_greetings,
 +            tags: topics,
 +            creator: creatorName,
 +            character_version: '',
 +            character_book: definition.embedded_lorebook,
 +            extensions: definition.extensions,
 +        },
 +        spec: 'chara_card_v2',
 +        spec_version: '2.0',
 +    };
 -    if (!downloadUrl) {
 +    const defaultAvatarPath = path.join(serverDirectory, DEFAULT_AVATAR_PATH);
 -        throw new Error('Download URL not found in character metadata');
 +    const defaultAvatarBuffer = fs.readFileSync(defaultAvatarPath);
 -    }
     constlet downloadResultimageBuffer = await fetch(downloadUrl)defaultAvatarBuffer;
 -    if (!downloadResult.ok) {
 +    const imageUrl = metadata.node?.max_res_url;
 -        const text = await downloadResult.text();
++
 -        console.error('Chub returned error', downloadResult.statusText, text);
 +    if (imageUrl) {
 -        throw new Error('Failed to download character');
 +        const downloadResult = await fetch(imageUrl);
 +        if (downloadResult.ok) {
 +            imageBuffer = Buffer.from(await downloadResult.arrayBuffer());
 +        }
+    }
     const buffer = Buffer.fromwrite(awaitimageBuffer, downloadResultJSON.arrayBufferstringify(characterCard));
 -    const fileName =
 +    const fileName = `${sanitize(characterCard.data.name)}.png`;
 -        downloadResult.headers.get('content-disposition')?.split('filename=')[1]?.replace(/["']/g, '') ||
 +    const fileType = 'image/png';
 -        `${sanitize(projectName)}.png`;
 -    const fileType = downloadResult.headers.get('content-type');
     return { buffer, fileName, fileType };
+}