primellm 0.1.0 → 1.0.0

package/src/index.ts

@@ -0,0 +1,411 @@
+#!/usr/bin/env node
+
+import chalk from 'chalk';
+import figlet from 'figlet';
+import ora from 'ora';
+import { execSync, spawnSync } from 'child_process';
+import { platform, homedir } from 'os';
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'fs';
+import { join } from 'path';
+import open from 'open';
+import Enquirer from 'enquirer';
+
+const { prompt } = Enquirer;
+
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+
+const PRIMELLM_BASE_URL = 'https://api.primellm.in/v1';
+const PRIMELLM_DASHBOARD_URL = 'https://primellm.in/dashboard/api-keys';
+
+// Tool installation packages
+const TOOLS = {
+    'Claude Code': {
+        command: 'claude',
+        package: '@anthropic-ai/claude-code',
+        configName: 'claude',
+    },
+    'Codex': {
+        command: 'codex',
+        package: '@openai/codex',
+        configName: 'codex',
+    },
+} as const;
+
+type ToolName = keyof typeof TOOLS;
+
+// ============================================================================
+// SYSTEM DETECTION
+// ============================================================================
+
+function getOS(): string {
+    const os = platform();
+    switch (os) {
+        case 'darwin': return 'macOS';
+        case 'linux': return 'Linux';
+        case 'win32': return 'Windows';
+        default: return os;
+    }
+}
+
+function getShell(): string {
+    const shell = process.env.SHELL || process.env.COMSPEC || 'unknown';
+    const shellName = shell.split(/[/\\]/).pop() || 'unknown';
+    return shellName;
+}
+
+function getNodeVersion(): string {
+    return process.version;
+}
+
+function commandExists(cmd: string): boolean {
+    try {
+        const isWindows = platform() === 'win32';
+        const checkCmd = isWindows ? `where ${cmd}` : `which ${cmd}`;
+        execSync(checkCmd, { stdio: 'ignore' });
+        return true;
+    } catch {
+        return false;
+    }
+}
+
+// ============================================================================
+// ASCII ART & DISPLAY
+// ============================================================================
+
+function displayBanner(): void {
+    console.clear();
+
+    const banner = figlet.textSync('PrimeLLM', {
+        font: 'Big',
+        horizontalLayout: 'default',
+    });
+
+    console.log(chalk.cyan(banner));
+    console.log(chalk.gray('━'.repeat(60)));
+    console.log(chalk.white.bold('  Configure AI coding tools with PrimeLLM'));
+    console.log(chalk.gray('━'.repeat(60)));
+    console.log();
+}
+
+function displaySystemInfo(): void {
+    console.log(chalk.white.bold('📊 System Information\n'));
+
+    const os = getOS();
+    const shell = getShell();
+    const nodeVersion = getNodeVersion();
+
+    console.log(`  ${chalk.gray('OS:')}      ${chalk.green(os)}`);
+    console.log(`  ${chalk.gray('Shell:')}   ${chalk.green(shell)}`);
+    console.log(`  ${chalk.gray('Node:')}    ${chalk.green(nodeVersion)}`);
+    console.log();
+}
+
+// ============================================================================
+// CONFIGURATION PATHS
+// ============================================================================
+
+function getConfigPaths(tool: ToolName, scope: 'system' | 'project'): { dir: string; file: string } {
+    const home = homedir();
+
+    if (tool === 'Claude Code') {
+        if (scope === 'system') {
+            return {
+                dir: join(home, '.claude'),
+                file: join(home, '.claude', 'config.json'),
+            };
+        } else {
+            return {
+                dir: '.claude',
+                file: join('.claude', 'config.json'),
+            };
+        }
+    } else {
+        // Codex uses OpenAI-compatible config
+        if (scope === 'system') {
+            return {
+                dir: join(home, '.codex'),
+                file: join(home, '.codex', 'config.json'),
+            };
+        } else {
+            return {
+                dir: '.codex',
+                file: join('.codex', 'config.json'),
+            };
+        }
+    }
+}
+
+interface Config {
+    baseUrl?: string;
+    apiKey?: string;
+    [key: string]: unknown;
+}
+
+function readExistingConfig(configPath: string): Config | null {
+    try {
+        if (existsSync(configPath)) {
+            const content = readFileSync(configPath, 'utf8');
+            return JSON.parse(content);
+        }
+    } catch {
+        // Config doesn't exist or is invalid
+    }
+    return null;
+}
+
+function writeConfig(tool: ToolName, scope: 'system' | 'project', apiKey: string): void {
+    const { dir, file } = getConfigPaths(tool, scope);
+
+    // Create directory if it doesn't exist
+    if (!existsSync(dir)) {
+        mkdirSync(dir, { recursive: true });
+    }
+
+    let config: Config;
+
+    if (tool === 'Claude Code') {
+        // Claude Code configuration format
+        config = {
+            apiKey: apiKey,
+            baseUrl: PRIMELLM_BASE_URL,
+            model: 'claude-sonnet-4-20250514', // Default model
+        };
+    } else {
+        // Codex (OpenAI-compatible) configuration format
+        config = {
+            apiKey: apiKey,
+            baseUrl: PRIMELLM_BASE_URL,
+            model: 'gpt-4.1', // Default model
+        };
+    }
+
+    // Preserve existing config values
+    const existing = readExistingConfig(file);
+    if (existing) {
+        config = { ...existing, ...config };
+    }
+
+    writeFileSync(file, JSON.stringify(config, null, 2));
+}
+
+// ============================================================================
+// TOOL INSTALLATION
+// ============================================================================
+
+async function installTool(tool: ToolName): Promise<boolean> {
+    const toolInfo = TOOLS[tool];
+    const spinner = ora(`Installing ${tool}...`).start();
+
+    try {
+        const isWindows = platform() === 'win32';
+        const npmCmd = isWindows ? 'npm.cmd' : 'npm';
+
+        const result = spawnSync(npmCmd, ['install', '-g', toolInfo.package], {
+            stdio: 'pipe',
+            shell: true,
+        });
+
+        if (result.status !== 0) {
+            spinner.fail(chalk.red(`Failed to install ${tool}`));
+            console.error(chalk.gray(result.stderr?.toString() || 'Unknown error'));
+            return false;
+        }
+
+        spinner.succeed(chalk.green(`${tool} installed successfully`));
+        return true;
+    } catch (error) {
+        spinner.fail(chalk.red(`Failed to install ${tool}`));
+        console.error(chalk.gray(String(error)));
+        return false;
+    }
+}
+
+// ============================================================================
+// VALIDATION
+// ============================================================================
+
+function validateApiKey(key: string): boolean {
+    return key.startsWith('primellm_') && key.length > 15;
+}
+
+function maskApiKey(key: string): string {
+    if (key.length <= 12) return '***';
+    return key.substring(0, 12) + '...' + key.substring(key.length - 4);
+}
+
+// ============================================================================
+// MAIN FLOW
+// ============================================================================
+
+async function main(): Promise<void> {
+    displayBanner();
+    displaySystemInfo();
+
+    const spinner = ora('Initializing...').start();
+    await new Promise(resolve => setTimeout(resolve, 500));
+    spinner.succeed('Ready');
+    console.log();
+
+    // ============ STEP 1: Tool Selection ============
+    console.log(chalk.white.bold('🔧 Tool Selection\n'));
+
+    const { selectedTool } = await prompt<{ selectedTool: ToolName }>({
+        type: 'select',
+        name: 'selectedTool',
+        message: 'Which tool would you like to configure?',
+        choices: ['Claude Code', 'Codex'],
+    });
+
+    console.log();
+
+    // ============ STEP 2: Tool Detection & Installation ============
+    const toolInfo = TOOLS[selectedTool];
+    const toolExists = commandExists(toolInfo.command);
+
+    if (toolExists) {
+        console.log(chalk.green(`✓ ${selectedTool} is already installed`));
+    } else {
+        console.log(chalk.yellow(`⚠ ${selectedTool} is not installed`));
+
+        const { confirmInstall } = await prompt<{ confirmInstall: boolean }>({
+            type: 'confirm',
+            name: 'confirmInstall',
+            message: `Would you like to install ${selectedTool}?`,
+            initial: true,
+        });
+
+        if (!confirmInstall) {
+            console.log(chalk.yellow('\n⚠ Installation cancelled. Please install manually and run again.'));
+            process.exit(0);
+        }
+
+        const installed = await installTool(selectedTool);
+        if (!installed) {
+            console.log(chalk.red('\n❌ Installation failed. Please install manually and run again.'));
+            process.exit(1);
+        }
+    }
+
+    console.log();
+
+    // ============ STEP 3: API Key ============
+    console.log(chalk.white.bold('🔑 API Key Configuration\n'));
+
+    const { hasApiKey } = await prompt<{ hasApiKey: string }>({
+        type: 'select',
+        name: 'hasApiKey',
+        message: 'Do you already have a PrimeLLM API key?',
+        choices: [
+            { name: 'yes', message: 'Yes, I have an API key' },
+            { name: 'no', message: 'No, I need to create one' },
+        ],
+    });
+
+    let apiKey: string;
+
+    if (hasApiKey === 'no') {
+        console.log(chalk.cyan('\n📝 Opening PrimeLLM dashboard to create an API key...\n'));
+
+        await open(PRIMELLM_DASHBOARD_URL);
+
+        await prompt({
+            type: 'confirm',
+            name: 'ready',
+            message: 'Press Enter when you have created your API key...',
+        });
+    }
+
+    // Prompt for API key
+    const { inputApiKey } = await prompt<{ inputApiKey: string }>({
+        type: 'password',
+        name: 'inputApiKey',
+        message: 'Enter your PrimeLLM API key:',
+        validate: (value: string) => {
+            if (!value) return 'API key is required';
+            if (!validateApiKey(value)) return 'API key must start with "primellm_"';
+            return true;
+        },
+    });
+
+    apiKey = inputApiKey;
+    console.log(chalk.green(`✓ API key validated: ${maskApiKey(apiKey)}`));
+    console.log();
+
+    // ============ STEP 4: Configuration Scope ============
+    console.log(chalk.white.bold('📁 Configuration Scope\n'));
+
+    const { configScope } = await prompt<{ configScope: 'system' | 'project' }>({
+        type: 'select',
+        name: 'configScope',
+        message: 'Choose configuration scope:',
+        choices: [
+            { name: 'system', message: 'System-level (applies to all projects)' },
+            { name: 'project', message: 'Project-level (current project only)' },
+        ],
+    });
+
+    console.log();
+
+    // ============ STEP 5: Check Existing Config ============
+    const { file: configFile } = getConfigPaths(selectedTool, configScope);
+    const existingConfig = readExistingConfig(configFile);
+
+    if (existingConfig) {
+        console.log(chalk.yellow('⚠ Existing configuration found:\n'));
+        console.log(chalk.gray(`  Base URL: ${existingConfig.baseUrl || '(not set)'}`));
+        console.log(chalk.gray(`  API Key:  ${existingConfig.apiKey ? maskApiKey(existingConfig.apiKey) : '(not set)'}`));
+        console.log();
+
+        const { confirmOverwrite } = await prompt<{ confirmOverwrite: boolean }>({
+            type: 'confirm',
+            name: 'confirmOverwrite',
+            message: 'Do you want to overwrite this configuration?',
+            initial: true,
+        });
+
+        if (!confirmOverwrite) {
+            console.log(chalk.yellow('\n⚠ Configuration cancelled.'));
+            process.exit(0);
+        }
+        console.log();
+    }
+
+    // ============ STEP 6: Write Configuration ============
+    const writeSpinner = ora('Writing configuration...').start();
+
+    try {
+        writeConfig(selectedTool, configScope, apiKey);
+        writeSpinner.succeed(chalk.green('Configuration saved'));
+    } catch (error) {
+        writeSpinner.fail(chalk.red('Failed to write configuration'));
+        console.error(chalk.gray(String(error)));
+        process.exit(1);
+    }
+
+    // ============ SUCCESS ============
+    console.log();
+    console.log(chalk.gray('━'.repeat(60)));
+    console.log();
+    console.log(chalk.green.bold('✅ PrimeLLM configured successfully!'));
+    console.log();
+    console.log(chalk.white('You can now use:'));
+    console.log(chalk.cyan(`  ${toolInfo.command}`));
+    console.log(chalk.white('with PrimeLLM as the backend.'));
+    console.log();
+    console.log(chalk.gray(`Config location: ${configFile}`));
+    console.log(chalk.gray(`Base URL: ${PRIMELLM_BASE_URL}`));
+    console.log();
+    console.log(chalk.gray('━'.repeat(60)));
+    console.log();
+    console.log(chalk.white('Need help? Visit ') + chalk.cyan('https://primellm.in/docs'));
+    console.log();
+}
+
+// Run the CLI
+main().catch((error) => {
+    console.error(chalk.red('An unexpected error occurred:'));
+    console.error(chalk.gray(String(error)));
+    process.exit(1);
+});

package/tsconfig.json

@@ -0,0 +1,27 @@
+{
+    "compilerOptions": {
+        "target": "ES2022",
+        "module": "NodeNext",
+        "moduleResolution": "NodeNext",
+        "lib": [
+            "ES2022"
+        ],
+        "outDir": "./dist",
+        "rootDir": "./src",
+        "strict": true,
+        "esModuleInterop": true,
+        "skipLibCheck": true,
+        "forceConsistentCasingInFileNames": true,
+        "resolveJsonModule": true,
+        "declaration": true,
+        "declarationMap": true,
+        "sourceMap": true
+    },
+    "include": [
+        "src/**/*"
+    ],
+    "exclude": [
+        "node_modules",
+        "dist"
+    ]
+}

package/dist/types.d.ts

@@ -1,141 +0,0 @@
-/**
- * PrimeLLM SDK Types
- *
- * This file contains all the TypeScript types used by the PrimeLLM SDK.
- * These types match the response format from the PrimeLLM API.
- *
- * Think of types like "templates" that describe what data looks like.
- * They help catch errors before your code runs!
- */
-/**
- * The role of a message in a conversation.
- * - "system": Instructions for the AI (like "be helpful")
- * - "user": Messages from the human user
- * - "assistant": Messages from the AI
- */
-export type ChatRole = "system" | "user" | "assistant";
-/**
- * A single message in a conversation.
- * Each message has a role (who said it) and content (what they said).
- */
-export interface ChatMessage {
-    role: ChatRole;
-    content: string;
-}
-/**
- * Request body for the /v1/chat and /v1/chat/completions endpoints.
- * This is what you send TO the API.
- */
-export interface ChatRequest {
-    /** Model name, e.g. "gpt-5.1", "claude-sonnet-4.5", "gemini-3.0" */
-    model: string;
-    /** List of messages in the conversation */
-    messages: ChatMessage[];
-    /** Whether to stream the response (not supported yet) */
-    stream?: boolean;
-    /** Optional extra data to pass along */
-    metadata?: Record<string, unknown>;
-    /** Temperature for randomness (0.0 = focused, 1.0 = creative) */
-    temperature?: number;
-    /** Maximum tokens to generate */
-    max_tokens?: number;
-}
-/**
- * Request body for the /generate endpoint (legacy).
- * This endpoint uses a simpler "prompt" format instead of messages.
- */
-export interface GenerateRequest {
-    /** Model name, e.g. "gpt-5.1" */
-    model: string;
-    /** The messages to send (same as ChatRequest) */
-    messages: ChatMessage[];
-    /** Maximum tokens to generate */
-    max_tokens?: number;
-    /** Temperature for randomness */
-    temperature?: number;
-    /** Whether to stream (not supported yet) */
-    stream?: boolean;
-    /** Optional extra data */
-    metadata?: Record<string, unknown>;
-}
-/**
- * A single "choice" in the API response.
- * The API can return multiple choices, but usually returns just one.
- */
-export interface ChatChoice {
-    /** Index of this choice (usually 0) */
-    index: number;
-    /** The AI's response message */
-    message: ChatMessage;
-    /** Why the AI stopped: "stop" means it finished normally */
-    finish_reason?: string | null;
-}
-/**
- * Token usage information.
- * Tokens are like "word pieces" - the AI counts usage in tokens.
- */
-export interface Usage {
-    /** Tokens used by your input (prompt) */
-    prompt_tokens: number;
-    /** Tokens used by the AI's response */
-    completion_tokens: number;
-    /** Total tokens = prompt + completion */
-    total_tokens: number;
-}
-/**
- * Credit information from your PrimeLLM account.
- * Credits are like "money" - each API call costs some credits.
- */
-export interface CreditsInfo {
-    /** How many credits you have left */
-    remaining: number;
-    /** How much this request cost */
-    cost?: number;
-}
-/**
- * The full response from /v1/chat or /v1/chat/completions.
- * This matches the OpenAI response format.
- */
-export interface ChatResponse {
-    /** Unique ID for this response */
-    id: string;
-    /** Which model was used */
-    model: string;
-    /** When this was created (Unix timestamp in seconds) */
-    created: number;
-    /** Type of object (always "chat.completion") */
-    object?: string;
-    /** The AI's response(s) */
-    choices: ChatChoice[];
-    /** Token usage information */
-    usage: Usage;
-    /** Your credit balance (PrimeLLM-specific) */
-    credits?: CreditsInfo;
-}
-/**
- * Response from the /generate endpoint (legacy format).
- */
-export interface GenerateResponse {
-    /** The AI's reply text */
-    reply: string;
-    /** Which model was used */
-    model: string;
-    /** Total tokens used */
-    tokens_used: number;
-    /** Cost of this request */
-    cost: number;
-    /** Credits remaining in your account */
-    credits_remaining: number;
-}
-/**
- * Options for creating a PrimeLLMClient.
- */
-export interface PrimeLLMClientOptions {
-    /** Your PrimeLLM API key (starts with "primellm_live_") */
-    apiKey: string;
-    /** Base URL for the API (default: "https://api.primellm.in") */
-    baseURL?: string;
-    /** Request timeout in milliseconds (default: 60000 = 60 seconds) */
-    timeoutMs?: number;
-}
-//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAMH;;;;;GAKG;AACH,MAAM,MAAM,QAAQ,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;AAEvD;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,QAAQ,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACnB;AAMD;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,oEAAoE;IACpE,KAAK,EAAE,MAAM,CAAC;IAEd,2CAA2C;IAC3C,QAAQ,EAAE,WAAW,EAAE,CAAC;IAExB,yDAAyD;IACzD,MAAM,CAAC,EAAE,OAAO,CAAC;IAEjB,wCAAwC;IACxC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAEnC,iEAAiE;IACjE,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,iCAAiC;IACjC,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,eAAe;IAC5B,iCAAiC;IACjC,KAAK,EAAE,MAAM,CAAC;IAEd,iDAAiD;IACjD,QAAQ,EAAE,WAAW,EAAE,CAAC;IAExB,iCAAiC;IACjC,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,iCAAiC;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,4CAA4C;IAC5C,MAAM,CAAC,EAAE,OAAO,CAAC;IAEjB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAMD;;;GAGG;AACH,MAAM,WAAW,UAAU;IACvB,uCAAuC;IACvC,KAAK,EAAE,MAAM,CAAC;IAEd,gCAAgC;IAChC,OAAO,EAAE,WAAW,CAAC;IAErB,4DAA4D;IAC5D,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED;;;GAGG;AACH,MAAM,WAAW,KAAK;IAClB,yCAAyC;IACzC,aAAa,EAAE,MAAM,CAAC;IAEtB,uCAAuC;IACvC,iBAAiB,EAAE,MAAM,CAAC;IAE1B,yCAAyC;IACzC,YAAY,EAAE,MAAM,CAAC;CACxB;AAED;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,qCAAqC;IACrC,SAAS,EAAE,MAAM,CAAC;IAElB,iCAAiC;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,YAAY;IACzB,kCAAkC;IAClC,EAAE,EAAE,MAAM,CAAC;IAEX,2BAA2B;IAC3B,KAAK,EAAE,MAAM,CAAC;IAEd,wDAAwD;IACxD,OAAO,EAAE,MAAM,CAAC;IAEhB,gDAAgD;IAChD,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,2BAA2B;IAC3B,OAAO,EAAE,UAAU,EAAE,CAAC;IAEtB,8BAA8B;IAC9B,KAAK,EAAE,KAAK,CAAC;IAEb,8CAA8C;IAC9C,OAAO,CAAC,EAAE,WAAW,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC7B,0BAA0B;IAC1B,KAAK,EAAE,MAAM,CAAC;IAEd,2BAA2B;IAC3B,KAAK,EAAE,MAAM,CAAC;IAEd,wBAAwB;IACxB,WAAW,EAAE,MAAM,CAAC;IAEpB,2BAA2B;IAC3B,IAAI,EAAE,MAAM,CAAC;IAEb,wCAAwC;IACxC,iBAAiB,EAAE,MAAM,CAAC;CAC7B;AAMD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IAClC,2DAA2D;IAC3D,MAAM,EAAE,MAAM,CAAC;IAEf,gEAAgE;IAChE,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,oEAAoE;IACpE,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB"}

package/dist/types.js

@@ -1,10 +0,0 @@
-/**
- * PrimeLLM SDK Types
- *
- * This file contains all the TypeScript types used by the PrimeLLM SDK.
- * These types match the response format from the PrimeLLM API.
- *
- * Think of types like "templates" that describe what data looks like.
- * They help catch errors before your code runs!
- */
-export {};