@hasna/connectors 1.0.2 → 1.0.3

package/connectors/connect-leonardo/src/api/client.ts

@@ -0,0 +1,164 @@
+import type {
+  LeonardoConfig,
+  GenerateRequest,
+  GenerateResponse,
+  GetGenerationResponse,
+  ListGenerationsResponse,
+  ListModelsResponse,
+  ListPlatformModelsResponse,
+  VariationRequest,
+  VariationResponse,
+  UserInfo,
+} from '../types';
+import { LeonardoApiError } from '../types';
+
+const DEFAULT_BASE_URL = 'https://cloud.leonardo.ai/api/rest/v1';
+
+export interface RequestOptions {
+  method?: 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+  params?: Record<string, string | number | boolean | undefined>;
+  body?: Record<string, unknown> | unknown[] | string;
+  headers?: Record<string, string>;
+}
+
+export class LeonardoClient {
+  private readonly apiKey: string;
+  private readonly baseUrl: string;
+
+  constructor(config: LeonardoConfig) {
+    if (!config.apiKey) {
+      throw new Error('API key is required');
+    }
+    this.apiKey = config.apiKey;
+    this.baseUrl = config.baseUrl || DEFAULT_BASE_URL;
+  }
+
+  private buildUrl(path: string, params?: Record<string, string | number | boolean | undefined>): string {
+    const url = new URL(`${this.baseUrl}${path}`);
+
+    if (params) {
+      Object.entries(params).forEach(([key, value]) => {
+        if (value !== undefined && value !== null && value !== '') {
+          url.searchParams.append(key, String(value));
+        }
+      });
+    }
+
+    return url.toString();
+  }
+
+  async request<T>(path: string, options: RequestOptions = {}): Promise<T> {
+    const { method = 'GET', params, body, headers = {} } = options;
+
+    const url = this.buildUrl(path, params);
+
+    const requestHeaders: Record<string, string> = {
+      'Authorization': `Bearer ${this.apiKey}`,
+      'Accept': 'application/json',
+      ...headers,
+    };
+
+    if (body && ['POST', 'PUT', 'PATCH'].includes(method)) {
+      requestHeaders['Content-Type'] = 'application/json';
+    }
+
+    const fetchOptions: RequestInit = {
+      method,
+      headers: requestHeaders,
+    };
+
+    if (body && ['POST', 'PUT', 'PATCH'].includes(method)) {
+      fetchOptions.body = typeof body === 'string' ? body : JSON.stringify(body);
+    }
+
+    const response = await fetch(url, fetchOptions);
+
+    if (response.status === 204) {
+      return {} as T;
+    }
+
+    let data: unknown;
+    const contentType = response.headers.get('content-type') || '';
+
+    if (contentType.includes('application/json')) {
+      const text = await response.text();
+      if (text) {
+        try {
+          data = JSON.parse(text);
+        } catch {
+          data = text;
+        }
+      }
+    } else {
+      data = await response.text();
+    }
+
+    if (!response.ok) {
+      const errorMessage = typeof data === 'object' && data !== null
+        ? JSON.stringify(data)
+        : String(data || response.statusText);
+      throw new LeonardoApiError(errorMessage, response.status);
+    }
+
+    return data as T;
+  }
+
+  async get<T>(path: string, params?: Record<string, string | number | boolean | undefined>): Promise<T> {
+    return this.request<T>(path, { method: 'GET', params });
+  }
+
+  async post<T>(path: string, body?: Record<string, unknown> | unknown[] | string | object): Promise<T> {
+    return this.request<T>(path, { method: 'POST', body: body as Record<string, unknown> });
+  }
+
+  // ============================================
+  // Generation Methods
+  // ============================================
+
+  async generate(params: GenerateRequest): Promise<GenerateResponse> {
+    return this.post<GenerateResponse>('/generations', params);
+  }
+
+  async getGeneration(generationId: string): Promise<GetGenerationResponse> {
+    return this.get<GetGenerationResponse>(`/generations/${generationId}`);
+  }
+
+  async listGenerations(userId: string, limit?: number, offset?: number): Promise<ListGenerationsResponse> {
+    return this.get<ListGenerationsResponse>(`/generations/user/${userId}`, { limit, offset });
+  }
+
+  // ============================================
+  // Model Methods
+  // ============================================
+
+  async listModels(): Promise<ListPlatformModelsResponse> {
+    return this.get<ListPlatformModelsResponse>('/platformModels');
+  }
+
+  async listCustomModels(userId: string): Promise<ListModelsResponse> {
+    return this.get<ListModelsResponse>(`/models?userId=${userId}`);
+  }
+
+  // ============================================
+  // Variation Methods
+  // ============================================
+
+  async createVariation(params: VariationRequest): Promise<VariationResponse> {
+    return this.post<VariationResponse>('/variations', params);
+  }
+
+  // ============================================
+  // User Methods
+  // ============================================
+
+  async getUser(): Promise<UserInfo> {
+    return this.get<UserInfo>('/me');
+  }
+
+  getApiKeyPreview(): string {
+    if (this.apiKey.length > 10) {
+      return `${this.apiKey.substring(0, 6)}...${this.apiKey.substring(this.apiKey.length - 4)}`;
+    }
+    return '***';
+  }
+}

package/connectors/connect-leonardo/src/api/index.ts

@@ -0,0 +1,63 @@
+import type {
+  LeonardoConfig,
+  GenerateRequest,
+  GenerateResponse,
+  GetGenerationResponse,
+  ListGenerationsResponse,
+  ListPlatformModelsResponse,
+  ListModelsResponse,
+  VariationRequest,
+  VariationResponse,
+  UserInfo,
+} from '../types';
+import { LeonardoClient } from './client';
+
+export class Leonardo {
+  private readonly client: LeonardoClient;
+
+  constructor(config: LeonardoConfig) {
+    this.client = new LeonardoClient(config);
+  }
+
+  async generate(params: GenerateRequest): Promise<GenerateResponse> {
+    return this.client.generate(params);
+  }
+
+  async getGeneration(generationId: string): Promise<GetGenerationResponse> {
+    return this.client.getGeneration(generationId);
+  }
+
+  async listGenerations(userId: string, limit?: number, offset?: number): Promise<ListGenerationsResponse> {
+    return this.client.listGenerations(userId, limit, offset);
+  }
+
+  async listModels(): Promise<ListPlatformModelsResponse> {
+    return this.client.listModels();
+  }
+
+  async listCustomModels(userId: string): Promise<ListModelsResponse> {
+    return this.client.listCustomModels(userId);
+  }
+
+  async createVariation(params: VariationRequest): Promise<VariationResponse> {
+    return this.client.createVariation(params);
+  }
+
+  async getUser(): Promise<UserInfo> {
+    return this.client.getUser();
+  }
+
+  static fromEnv(): Leonardo {
+    const apiKey = process.env.LEONARDO_API_KEY;
+    if (!apiKey) {
+      throw new Error('LEONARDO_API_KEY environment variable is required');
+    }
+    return new Leonardo({ apiKey });
+  }
+
+  getApiKeyPreview(): string {
+    return this.client.getApiKeyPreview();
+  }
+}
+
+export { LeonardoClient } from './client';

package/connectors/connect-leonardo/src/cli/index.ts

@@ -0,0 +1,344 @@
+#!/usr/bin/env bun
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { Leonardo } from '../api';
+import {
+  getApiKey,
+  setApiKey,
+  clearConfig,
+  getConfigDir,
+  setProfileOverride,
+  getCurrentProfile,
+  setCurrentProfile,
+  listProfiles,
+  createProfile,
+  deleteProfile,
+  profileExists,
+  loadProfile,
+} from '../utils/config';
+import type { OutputFormat } from '../utils/output';
+import { success, error, info, print } from '../utils/output';
+
+const CONNECTOR_NAME = 'connect-leonardo';
+const VERSION = '0.0.1';
+
+const program = new Command();
+
+program
+  .name(CONNECTOR_NAME)
+  .description('Leonardo AI connector CLI - AI image generation with multi-profile support')
+  .version(VERSION)
+  .option('-k, --api-key <key>', 'API key (overrides config)')
+  .option('-f, --format <format>', 'Output format (json, pretty)', 'pretty')
+  .option('-p, --profile <profile>', 'Use a specific profile')
+  .hook('preAction', (thisCommand) => {
+    const opts = thisCommand.opts();
+    if (opts.profile) {
+      if (!profileExists(opts.profile)) {
+        error(`Profile "${opts.profile}" does not exist. Create it with "${CONNECTOR_NAME} profile create ${opts.profile}"`);
+        process.exit(1);
+      }
+      setProfileOverride(opts.profile);
+    }
+    if (opts.apiKey) {
+      process.env.LEONARDO_API_KEY = opts.apiKey;
+    }
+  });
+
+function getFormat(cmd: Command): OutputFormat {
+  const parent = cmd.parent;
+  return (parent?.opts().format || 'pretty') as OutputFormat;
+}
+
+function getClient(): Leonardo {
+  const apiKey = getApiKey();
+  if (!apiKey) {
+    error(`No API key configured. Run "${CONNECTOR_NAME} config set-key <key>" or set LEONARDO_API_KEY environment variable.`);
+    process.exit(1);
+  }
+  return new Leonardo({ apiKey });
+}
+
+// ============================================
+// Profile Commands
+// ============================================
+const profileCmd = program
+  .command('profile')
+  .description('Manage configuration profiles');
+
+profileCmd
+  .command('list')
+  .description('List all profiles')
+  .action(() => {
+    const profiles = listProfiles();
+    const current = getCurrentProfile();
+    if (profiles.length === 0) {
+      info('No profiles found. Use "profile create <name>" to create one.');
+      return;
+    }
+    success(`Profiles:`);
+    profiles.forEach(p => {
+      const isActive = p === current ? chalk.green(' (active)') : '';
+      console.log(`  ${p}${isActive}`);
+    });
+  });
+
+profileCmd
+  .command('use <name>')
+  .description('Switch to a profile')
+  .action((name: string) => {
+    if (!profileExists(name)) {
+      error(`Profile "${name}" does not exist. Create it with "profile create ${name}"`);
+      process.exit(1);
+    }
+    setCurrentProfile(name);
+    success(`Switched to profile: ${name}`);
+  });
+
+profileCmd
+  .command('create <name>')
+  .description('Create a new profile')
+  .option('--api-key <key>', 'API key')
+  .option('--use', 'Switch to this profile after creation')
+  .action((name: string, opts) => {
+    if (profileExists(name)) {
+      error(`Profile "${name}" already exists`);
+      process.exit(1);
+    }
+    createProfile(name, { apiKey: opts.apiKey });
+    success(`Profile "${name}" created`);
+    if (opts.use) {
+      setCurrentProfile(name);
+      info(`Switched to profile: ${name}`);
+    }
+  });
+
+profileCmd
+  .command('delete <name>')
+  .description('Delete a profile')
+  .action((name: string) => {
+    if (name === 'default') {
+      error('Cannot delete the default profile');
+      process.exit(1);
+    }
+    if (deleteProfile(name)) {
+      success(`Profile "${name}" deleted`);
+    } else {
+      error(`Profile "${name}" not found`);
+      process.exit(1);
+    }
+  });
+
+profileCmd
+  .command('show [name]')
+  .description('Show profile configuration')
+  .action((name?: string) => {
+    const profileName = name || getCurrentProfile();
+    const config = loadProfile(profileName);
+    const active = getCurrentProfile();
+    console.log(chalk.bold(`Profile: ${profileName}${profileName === active ? chalk.green(' (active)') : ''}`));
+    info(`API Key: ${config.apiKey ? `${config.apiKey.substring(0, 8)}...` : chalk.gray('not set')}`);
+  });
+
+// ============================================
+// Config Commands
+// ============================================
+const configCmd = program
+  .command('config')
+  .description('Manage CLI configuration (for active profile)');
+
+configCmd
+  .command('set-key <apiKey>')
+  .description('Set API key')
+  .action((apiKey: string) => {
+    setApiKey(apiKey);
+    success(`API key saved to profile: ${getCurrentProfile()}`);
+  });
+
+configCmd
+  .command('show')
+  .description('Show current configuration')
+  .action(() => {
+    const profileName = getCurrentProfile();
+    const apiKey = getApiKey();
+    console.log(chalk.bold(`Active Profile: ${profileName}`));
+    info(`Config directory: ${getConfigDir()}`);
+    info(`API Key: ${apiKey ? `${apiKey.substring(0, 8)}...` : chalk.gray('not set')}`);
+  });
+
+configCmd
+  .command('clear')
+  .description('Clear configuration for active profile')
+  .action(() => {
+    clearConfig();
+    success(`Configuration cleared for profile: ${getCurrentProfile()}`);
+  });
+
+// ============================================
+// Image Generation Commands
+// ============================================
+const imageCmd = program
+  .command('image')
+  .description('Image generation commands');
+
+imageCmd
+  .command('generate <prompt>')
+  .description('Generate images from text prompt')
+  .option('-m, --model <id>', 'Model ID')
+  .option('-w, --width <number>', 'Image width', '1024')
+  .option('-h, --height <number>', 'Image height', '1024')
+  .option('-n, --num <number>', 'Number of images', '1')
+  .option('-g, --guidance <number>', 'Guidance scale', '7')
+  .option('--negative <prompt>', 'Negative prompt')
+  .option('--seed <number>', 'Random seed')
+  .option('--alchemy', 'Enable Alchemy')
+  .option('--photo-real', 'Enable PhotoReal')
+  .action(async (prompt: string, opts) => {
+    try {
+      const client = getClient();
+      const result = await client.generate({
+        prompt,
+        modelId: opts.model,
+        width: parseInt(opts.width),
+        height: parseInt(opts.height),
+        num_images: parseInt(opts.num),
+        guidance_scale: parseFloat(opts.guidance),
+        negative_prompt: opts.negative,
+        seed: opts.seed ? parseInt(opts.seed) : undefined,
+        alchemy: opts.alchemy || undefined,
+        photoReal: opts.photoReal || undefined,
+      });
+
+      const format = getFormat(imageCmd);
+      if (format === 'json') {
+        print(result, format);
+      } else {
+        success(`Generation started`);
+        info(`Generation ID: ${result.sdGenerationJob.generationId}`);
+        info(`Use "connect-leonardo image get ${result.sdGenerationJob.generationId}" to check status`);
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+imageCmd
+  .command('get <generationId>')
+  .description('Get generation by ID')
+  .action(async (generationId: string) => {
+    try {
+      const client = getClient();
+      const result = await client.getGeneration(generationId);
+      const format = getFormat(imageCmd);
+      if (format === 'json') {
+        print(result, format);
+      } else {
+        const gen = result.generations_by_pk;
+        console.log(chalk.bold(`Generation: ${gen.id}`));
+        info(`Status: ${gen.status}`);
+        info(`Prompt: ${gen.prompt}`);
+        if (gen.generated_images.length > 0) {
+          success(`Generated ${gen.generated_images.length} image(s)`);
+          gen.generated_images.forEach((img, i) => {
+            console.log(chalk.bold(`\nImage ${i + 1}:`));
+            info(`URL: ${img.url}`);
+          });
+        }
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+imageCmd
+  .command('variation <imageId>')
+  .description('Create a variation of an image')
+  .option('-t, --type <type>', 'Transform type (OUTPAINT, INPAINT, UPSCALE, UNZOOM)', 'UPSCALE')
+  .action(async (imageId: string, opts) => {
+    try {
+      const client = getClient();
+      const result = await client.createVariation({
+        id: imageId,
+        isVariation: true,
+        transformType: opts.type as 'OUTPAINT' | 'INPAINT' | 'UPSCALE' | 'UNZOOM',
+      });
+
+      const format = getFormat(imageCmd);
+      if (format === 'json') {
+        print(result, format);
+      } else {
+        success(`Variation started`);
+        info(`Generation ID: ${result.sdGenerationJob.generationId}`);
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// ============================================
+// Model Commands
+// ============================================
+const modelCmd = program
+  .command('model')
+  .description('Model commands');
+
+modelCmd
+  .command('list')
+  .description('List platform models')
+  .action(async () => {
+    try {
+      const client = getClient();
+      const result = await client.listModels();
+      const format = getFormat(modelCmd);
+      if (format === 'json') {
+        print(result, format);
+      } else {
+        success(`Platform Models:`);
+        result.platform_models.forEach(model => {
+          console.log(chalk.bold(`\n${model.name}`));
+          info(`ID: ${model.id}`);
+          if (model.description) info(`Description: ${model.description}`);
+        });
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// ============================================
+// User Commands
+// ============================================
+const userCmd = program
+  .command('user')
+  .description('User commands');
+
+userCmd
+  .command('me')
+  .description('Get current user info')
+  .action(async () => {
+    try {
+      const client = getClient();
+      const result = await client.getUser();
+      const format = getFormat(userCmd);
+      if (format === 'json') {
+        print(result, format);
+      } else {
+        const user = result.user;
+        console.log(chalk.bold(`User: ${user.username}`));
+        info(`ID: ${user.id}`);
+        info(`Subscription Tokens: ${user.subscriptionTokens}`);
+        info(`GPT Tokens: ${user.subscriptionGptTokens}`);
+        info(`Model Tokens: ${user.subscriptionModelTokens}`);
+        info(`Token Renewal: ${user.tokenRenewalDate}`);
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+program.parse();

package/connectors/connect-leonardo/src/index.ts

@@ -0,0 +1,19 @@
+// Leonardo AI Connector
+// TypeScript wrapper for Leonardo AI image generation API
+
+export { Leonardo } from './api';
+export * from './types';
+export { LeonardoClient } from './api';
+
+export {
+  getApiKey,
+  setApiKey,
+  getCurrentProfile,
+  setCurrentProfile,
+  listProfiles,
+  createProfile,
+  deleteProfile,
+  loadProfile,
+  saveProfile,
+  clearConfig,
+} from './utils/config';