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}`, {
+    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 = metadata.node?.max_res_url;
+    const { definition, topics } = metadata.node;
+    /** @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);
-    }
-    const downloadResult = await fetch(downloadUrl);
+    let imageBuffer = 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.from(await downloadResult.arrayBuffer());
+    const buffer = write(imageBuffer, JSON.stringify(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 };
+}