generate-image-sdk.ts

import path from 'path';
import dotenv from 'dotenv';
import { writeFile } from 'fs';

import { Session } from '@yandex-cloud/nodejs-sdk/dist/session';

import { initOperationSdk } from '@yandex-cloud/nodejs-sdk/sdk/operation';
import { imageGeneration } from '@yandex-cloud/nodejs-sdk/ai-foundation_models-v1';
import { initImageGenerationSdk } from '@yandex-cloud/nodejs-sdk/sdk/ai-foundation_models-v1';

dotenv.config({ path: path.resolve(__dirname, '.env') });

const getEnv = (envName: string, defaultValue?: string): string => {
    const envValue = process.env[envName] || defaultValue;

    if (!envValue) {
        throw new Error(`Env variable ${envName} is not defined`);
    }

    return envValue;
};

const iamToken = getEnv('YC_IAM_TOKEN');
const folderId = getEnv('YC_FOLDER_ID');

(async function () {
    const session = new Session({ iamToken });

    const operationSdk = initOperationSdk(session);

    const imageGenerationSdk = initImageGenerationSdk(session);

    const generateImageOperation = await imageGenerationSdk.generateImage({
        folderId,
        modelId: 'yandex-art',
        generationOptions: {
            mimeType: 'image/jpeg',
        },
        messages: [{ text: 'Three cats', weight: 1 }],
    });

    const imageGenerationResponse = await operationSdk.pollOperation(
        generateImageOperation,
        1_000,
        {
            operationCallback: console.log,
        },
    );

    writeFile('./image.png', imageGenerationResponse.image, { encoding: 'base64' }, function (err) {
        console.log('File created');
    });
})();