@hasna/connectors 0.4.2 → 0.5.0

package/connectors/connect-baseten/README.md

@@ -0,0 +1,193 @@
+# scaffold-connector
+
+A TypeScript scaffold for building API connector CLIs with multi-profile support.
+
+## Features
+
+- Multi-profile configuration (switch between different API keys/accounts)
+- Bearer token authentication (easily customizable)
+- Clean CLI structure with Commander.js
+- Pretty and JSON output formats
+- TypeScript with strict mode
+
+## Quick Start
+
+### 1. Clone and Rename
+
+```bash
+# Clone for your connector
+git clone https://github.com/hasna/scaffold-connector.git connect-yourapi
+cd connect-yourapi
+
+# Update package.json name
+# Change "@hasna/scaffold-connector" to "@hasna/connect-yourapi"
+```
+
+### 2. Update Configuration
+
+Search for `TODO` comments throughout the codebase and update:
+
+- `src/cli/index.ts` - Update `CONNECTOR_NAME` and description
+- `src/utils/config.ts` - Update `CONNECTOR_NAME` and env var names
+- `src/api/client.ts` - Update `DEFAULT_BASE_URL` and authentication method
+- `src/api/index.ts` - Rename `Connector` class to your API name
+- `src/types/index.ts` - Add your API's type definitions
+- `package.json` - Update name, description, bin command
+- `.env.example` - Update environment variable names
+
+### 3. Install and Test
+
+```bash
+# Install dependencies
+bun install
+
+# Run CLI
+bun run dev
+
+# Or run specific commands
+bun run dev profile list
+bun run dev config show
+```
+
+## CLI Structure
+
+```bash
+connector [options] [command]
+
+Options:
+  -k, --api-key <key>      API key (overrides config)
+  -f, --format <format>    Output format (json, pretty)
+  -p, --profile <profile>  Use a specific profile
+
+Commands:
+  profile list             List all profiles
+  profile use <name>       Switch to a profile
+  profile create <name>    Create a new profile
+  profile delete <name>    Delete a profile
+  profile show [name]      Show profile configuration
+
+  config set-key <key>     Set API key for active profile
+  config show              Show current configuration
+  config clear             Clear configuration
+
+  example list             Example API command (replace)
+  example get <id>         Example API command (replace)
+  example create           Example API command (replace)
+```
+
+## Project Structure
+
+```
+src/
+├── api/
+│   ├── client.ts     # HTTP client with authentication
+│   ├── example.ts    # Example API module (replace with your API)
+│   └── index.ts      # Main connector class
+├── cli/
+│   └── index.ts      # CLI commands
+├── types/
+│   └── index.ts      # Type definitions
+├── utils/
+│   ├── config.ts     # Multi-profile configuration
+│   └── output.ts     # CLI output formatting
+└── index.ts          # Library exports
+```
+
+## Multi-Profile Configuration
+
+Profiles are stored in `~/.connect/{connector-name}/profiles/`:
+
+```
+~/.connect/connector/
+├── current_profile   # Name of active profile
+└── profiles/
+    ├── default.json  # Default profile
+    ├── work.json     # Named profile
+    └── personal.json # Named profile
+```
+
+### Profile Commands
+
+```bash
+# Create profiles
+connector profile create work --api-key sk-xxx --use
+connector profile create personal --api-key sk-yyy
+
+# Switch profiles
+connector profile use work
+
+# Use profile for single command
+connector -p personal example list
+
+# List profiles
+connector profile list
+```
+
+## Customizing Authentication
+
+Edit `src/api/client.ts` to change authentication:
+
+```typescript
+// Bearer token (default)
+'Authorization': `Bearer ${this.apiKey}`,
+
+// API Key header
+'X-API-Key': this.apiKey,
+
+// Basic auth
+'Authorization': `Basic ${Buffer.from(`${this.apiKey}:${this.apiSecret}`).toString('base64')}`,
+```
+
+## Adding API Endpoints
+
+1. Create a new file in `src/api/` (e.g., `users.ts`)
+2. Export it from `src/api/index.ts`
+3. Add types in `src/types/index.ts`
+4. Add CLI commands in `src/cli/index.ts`
+
+Example API module:
+
+```typescript
+// src/api/users.ts
+import type { ConnectorClient } from './client';
+
+export class UsersApi {
+  constructor(private readonly client: ConnectorClient) {}
+
+  async list(options?: { limit?: number }) {
+    return this.client.get('/users', { limit: options?.limit });
+  }
+
+  async get(id: string) {
+    return this.client.get(`/users/${id}`);
+  }
+}
+```
+
+## Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `CONNECTOR_API_KEY` | API key (overrides profile config) |
+| `CONNECTOR_API_SECRET` | API secret (optional) |
+| `CONNECTOR_BASE_URL` | Override base URL (optional) |
+
+## Development
+
+```bash
+# Install dependencies
+bun install
+
+# Run CLI in development
+bun run dev
+
+# Build for distribution
+bun run build
+
+# Type check
+bun run typecheck
+```
+
+## License
+
+MIT

package/connectors/connect-baseten/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "@hasna/connect-baseten",
+  "version": "0.0.1",
+  "description": "Baseten connector CLI - ML model deployment platform with multi-profile support",
+  "type": "module",
+  "bin": {
+    "connect-baseten": "./bin/index.js"
+  },
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "scripts": {
+    "build": "bun build ./src/index.ts --outdir ./dist --target bun && bun build ./src/cli/index.ts --outdir ./bin --target bun",
+    "dev": "bun run ./src/cli/index.ts",
+    "typecheck": "tsc --noEmit",
+    "prepublishOnly": "bun run build"
+  },
+  "keywords": [
+    "baseten",
+    "ai",
+    "api",
+    "connector",
+    "cli",
+    "typescript",
+    "bun",
+    "ml",
+    "deployment"
+  ],
+  "author": "Hasna",
+  "license": "Apache-2.0",
+  "devDependencies": {
+    "@types/bun": "latest",
+    "typescript": "^5"
+  },
+  "dependencies": {
+    "commander": "^12.1.0",
+    "chalk": "^5.3.0"
+  },
+  "engines": {
+    "bun": ">=1.0.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/hasna/open-connectors.git"
+  }
+}

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

@@ -0,0 +1,71 @@
+import { BasetenApiError, type BasetenConfig, type ModelsResponse, type PredictionRequest, type PredictionResponse } from '../types';
+
+const DEFAULT_BASE_URL = 'https://api.baseten.co/v1';
+
+export class BasetenClient {
+  private readonly apiKey: string;
+  private readonly baseUrl: string;
+
+  constructor(config: BasetenConfig) {
+    if (!config.apiKey) {
+      throw new Error('API key is required');
+    }
+    this.apiKey = config.apiKey;
+    this.baseUrl = config.baseUrl || DEFAULT_BASE_URL;
+  }
+
+  private async request<T>(
+    endpoint: string,
+    options: RequestInit = {}
+  ): Promise<T> {
+    const url = `${this.baseUrl}${endpoint}`;
+
+    const response = await fetch(url, {
+      ...options,
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Api-Key ${this.apiKey}`,
+        ...options.headers,
+      },
+    });
+
+    const data = await response.json().catch(() => null);
+
+    if (!response.ok) {
+      let errorMessage = String(data || response.statusText);
+      if (typeof data === 'object' && data !== null) {
+        const errorObj = data as { error?: { message?: string }; message?: string; detail?: string };
+        errorMessage = errorObj.error?.message || errorObj.message || errorObj.detail || JSON.stringify(data);
+      }
+      throw new BasetenApiError(errorMessage, response.status);
+    }
+
+    return data as T;
+  }
+
+  async listModels(): Promise<ModelsResponse> {
+    return this.request<ModelsResponse>('/models');
+  }
+
+  async getModel(modelId: string): Promise<{ model: unknown }> {
+    return this.request<{ model: unknown }>(`/models/${modelId}`);
+  }
+
+  async predict(request: PredictionRequest): Promise<PredictionResponse> {
+    const { model_id, version_id, input } = request;
+    const endpoint = version_id
+      ? `/models/${model_id}/versions/${version_id}/predict`
+      : `/models/${model_id}/predict`;
+    return this.request<PredictionResponse>(endpoint, {
+      method: 'POST',
+      body: JSON.stringify(input),
+    });
+  }
+
+  getApiKeyPreview(): string {
+    if (this.apiKey.length > 10) {
+      return `${this.apiKey.substring(0, 6)}...${this.apiKey.substring(this.apiKey.length - 4)}`;
+    }
+    return '***';
+  }
+}

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

@@ -0,0 +1,40 @@
+import type { BasetenConfig, ModelsResponse, PredictionRequest, PredictionResponse } from '../types';
+import { BasetenClient } from './client';
+
+export class Baseten {
+  private readonly client: BasetenClient;
+
+  constructor(config: BasetenConfig) {
+    this.client = new BasetenClient(config);
+  }
+
+  static fromEnv(): Baseten {
+    const apiKey = process.env.BASETEN_API_KEY;
+    if (!apiKey) {
+      throw new Error('BASETEN_API_KEY environment variable is required');
+    }
+    return new Baseten({ apiKey });
+  }
+
+  async listModels(): Promise<ModelsResponse> {
+    return this.client.listModels();
+  }
+
+  async getModel(modelId: string): Promise<{ model: unknown }> {
+    return this.client.getModel(modelId);
+  }
+
+  async predict(request: PredictionRequest): Promise<PredictionResponse> {
+    return this.client.predict(request);
+  }
+
+  getApiKeyPreview(): string {
+    return this.client.getApiKeyPreview();
+  }
+
+  getClient(): BasetenClient {
+    return this.client;
+  }
+}
+
+export { BasetenClient } from './client';

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

@@ -0,0 +1,244 @@
+#!/usr/bin/env bun
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { Baseten } 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-baseten';
+const VERSION = '0.0.1';
+
+const program = new Command();
+
+program
+  .name(CONNECTOR_NAME)
+  .description('Baseten connector CLI - ML model deployment platform 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.BASETEN_API_KEY = opts.apiKey;
+    }
+  });
+
+function getFormat(cmd: Command): OutputFormat {
+  const parent = cmd.parent;
+  return (parent?.opts().format || 'pretty') as OutputFormat;
+}
+
+function getClient(): Baseten {
+  const apiKey = getApiKey();
+  if (!apiKey) {
+    error(`No API key configured. Run "${CONNECTOR_NAME} config set-key <key>" or set BASETEN_API_KEY environment variable.`);
+    process.exit(1);
+  }
+  return new Baseten({ 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()}`);
+  });
+
+// ============================================
+// Models Commands
+// ============================================
+const modelsCmd = program
+  .command('models')
+  .description('Manage Baseten models');
+
+modelsCmd
+  .command('list')
+  .description('List deployed models')
+  .action(async () => {
+    try {
+      const client = getClient();
+      const result = await client.listModels();
+      print(result, getFormat(modelsCmd));
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+modelsCmd
+  .command('get <modelId>')
+  .description('Get model details')
+  .action(async (modelId: string) => {
+    try {
+      const client = getClient();
+      const result = await client.getModel(modelId);
+      print(result, getFormat(modelsCmd));
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// ============================================
+// Predict Commands
+// ============================================
+const predictCmd = program
+  .command('predict')
+  .description('Run predictions on models');
+
+predictCmd
+  .command('run <modelId>')
+  .description('Run a prediction')
+  .requiredOption('-i, --input <json>', 'Input data as JSON')
+  .option('-v, --version <versionId>', 'Specific model version')
+  .action(async (modelId: string, opts) => {
+    try {
+      const client = getClient();
+      const input = JSON.parse(opts.input);
+      const result = await client.predict({
+        model_id: modelId,
+        input,
+        version_id: opts.version,
+      });
+      print(result, getFormat(predictCmd));
+    } catch (err) {
+      if (err instanceof SyntaxError) {
+        error('Invalid JSON input');
+      } else {
+        error(String(err));
+      }
+      process.exit(1);
+    }
+  });
+
+program.parse();

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

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

package/connectors/connect-baseten/src/types/index.ts

@@ -0,0 +1,55 @@
+// Baseten API Types
+
+export interface BasetenConfig {
+  apiKey: string;
+  baseUrl?: string;
+}
+
+// Model Types
+export interface BasetenModel {
+  id: string;
+  name: string;
+  created_at?: string;
+  updated_at?: string;
+  status?: string;
+}
+
+export interface ModelsResponse {
+  models: BasetenModel[];
+}
+
+// Prediction Types
+export interface PredictionRequest {
+  model_id: string;
+  input: Record<string, unknown>;
+  version_id?: string;
+}
+
+export interface PredictionResponse {
+  id: string;
+  model_id: string;
+  status: string;
+  output?: unknown;
+  created_at?: string;
+  completed_at?: string;
+}
+
+// Deployment Types
+export interface DeploymentStatus {
+  id: string;
+  model_id: string;
+  status: string;
+  created_at?: string;
+}
+
+// Error Types
+export class BasetenApiError extends Error {
+  constructor(
+    message: string,
+    public readonly status: number,
+    public readonly code?: string
+  ) {
+    super(message);
+    this.name = 'BasetenApiError';
+  }
+}