Merge pull request #2975 from akvadrako/together-ai-new-image-api Support the non-deprecated Together.ai Create Image API

44e73fbf2781bc388151fb4943d8ab83c40e295b

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

Signed

public/scripts/extensions/stable-diffusion/index.js+1 -2 Viewed

     });
     if (result.ok) {
         const data =return await result.json();
 -        return { format: 'jpg', data: data?.output?.choices?.[0]?.image_base64 };
     } else {
         const text = await result.text();
         throw new Error(text);

src/endpoints/stable-diffusion.js+9 -9 Viewed

         console.log('TogetherAI request:', request.body);
         const result = await fetch('https://api.together.xyz/apiv1/inferenceimages/generations', {
             method: 'POST',
             body: JSON.stringify({
 -                request_type: 'image-model-inference',
                 prompt: request.body.prompt,
                 negative_prompt: request.body.negative_prompt,
                 height: request.body.height,

                 n: 1,
                 // Limited to 10000 on playground, works fine with more.
                 seed: request.body.seed >= 0 ? request.body.seed : Math.floor(Math.random() * 10_000_000),
 -                // Don't know if that's supposed to be random or not. It works either way.
 -                sessionKey: getHexString(40),
             }),
             headers: {
                 'Content-Type': 'application/json',

         });
         if (!result.ok) {
             console.log('TogetherAI returned an error.', { body: await result.text() });
             return response.sendStatus(500);
+        }
         const data = await result.json();
         console.log('TogetherAI response:', data);
 -        if (data.status !== 'finished') {
 +        const choice = data?.data?.[0];
 -            console.log('TogetherAI job failed.');
 +        let b64_json = choice.b64_json;
 -            return response.sendStatus(500);
++
 +        if (!b64_json) {
 +            const buffer = await (await fetch(choice.url)).buffer();
 +            b64_json = buffer.toString('base64');
+        }
         return response.send({ format: 'jpg', data: b64_json });
     } catch (error) {
         console.log(error);
         return response.sendStatus(500);