spec-gen-cli 1.0.0

package/dist/core/services/llm-service.d.ts

@@ -0,0 +1,336 @@
+/**
+ * LLM Service
+ *
+ * Provides a clean interface for LLM interactions with proper error handling,
+ * retry logic, token management, and cost tracking.
+ */
+/**
+ * Claude Code CLI provider
+ *
+ * Routes LLM calls through the local `claude` CLI binary in non-interactive
+ * mode (`claude -p ...`).  Authentication is handled by the Claude Code session
+ * (Max/Pro subscription) — no ANTHROPIC_API_KEY is required.
+ */
+export declare class ClaudeCodeProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private model;
+    constructor(model?: string);
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    countTokens(text: string): number;
+}
+/**
+ * Mistral Vibe CLI provider
+ *
+ * Routes LLM calls through the local `mistral-vibe` CLI binary (standalone, no npm).
+ * No API key required — uses local LLM execution.
+ * If the binary is not on PATH, set MISTRAL_VIBE_CLI to its full path.
+ * The CLI is invoked as `vibe` (not `mistral-vibe`).
+ */
+export declare class MistralVibeProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private model;
+    constructor(model?: string);
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    countTokens(text: string): number;
+}
+/**
+ * Completion request parameters
+ */
+export interface CompletionRequest {
+    systemPrompt: string;
+    userPrompt: string;
+    temperature?: number;
+    maxTokens?: number;
+    stopSequences?: string[];
+    responseFormat?: 'text' | 'json';
+}
+/**
+ * Completion response
+ */
+export interface CompletionResponse {
+    content: string;
+    usage: {
+        inputTokens: number;
+        outputTokens: number;
+        totalTokens: number;
+    };
+    model: string;
+    finishReason: 'stop' | 'length' | 'error';
+}
+/**
+ * LLM provider interface
+ */
+export interface LLMProvider {
+    name: string;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    countTokens(text: string): number;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+}
+export type ProviderName = 'anthropic' | 'openai' | 'openai-compat' | 'gemini' | 'gemini-cli' | 'claude-code' | 'mistral-vibe';
+/**
+ * Token usage tracking
+ */
+export interface TokenUsage {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+    requests: number;
+}
+/**
+ * Cost tracking
+ */
+export interface CostTracking {
+    estimatedCost: number;
+    currency: string;
+    byProvider: Record<string, number>;
+}
+/**
+ * LLM service options
+ */
+export interface LLMServiceOptions {
+    /** Primary provider to use */
+    provider?: ProviderName;
+    /** Model override */
+    model?: string;
+    /** Custom API base URL (e.g., for local/enterprise OpenAI-compatible servers) */
+    apiBase?: string;
+    /** Disable SSL verification (for internal/self-signed certificates) */
+    sslVerify?: boolean;
+    /** Base URL for openai-compat provider (overrides OPENAI_COMPAT_BASE_URL env var) */
+    openaiCompatBaseUrl?: string;
+    /** Maximum retry attempts */
+    maxRetries?: number;
+    /** Initial retry delay in ms */
+    initialDelay?: number;
+    /** Maximum retry delay in ms */
+    maxDelay?: number;
+    /** Request timeout in ms */
+    timeout?: number;
+    /** Cost warning threshold in USD */
+    costWarningThreshold?: number;
+    /** Log directory for prompts/responses */
+    logDir?: string;
+    /** Enable prompt logging */
+    enableLogging?: boolean;
+}
+/**
+ * Parse the number of milliseconds to wait before retrying a 429 response.
+ *
+ * Checks (in order):
+ *  1. Standard `Retry-After` HTTP header (seconds as integer, or HTTP-date)
+ *  2. `Limit resets at: YYYY-MM-DD HH:MM:SS UTC` in the response body
+ *
+ * Returns `undefined` when nothing useful is found so the caller can fall back
+ * to its own exponential-backoff delay.
+ */
+export declare function parseRetryAfterMs(body: string, retryAfterHeader?: string | null): number | undefined;
+/**
+ * Exported for use in pre-flight cost estimation.
+ * Look up pricing for a model ID using prefix/family matching.
+ * Exact match first, then longest prefix match, then provider default.
+ *
+ * This is robust to minor version suffixes like "claude-sonnet-4-6-20251120"
+ * matching the "claude-sonnet-4" family entry.
+ */
+export declare function lookupPricing(providerName: string, modelId: string): {
+    input: number;
+    output: number;
+};
+/**
+ * Estimate token count from text (rough approximation)
+ * ~4 characters per token for English text
+ */
+export declare function estimateTokens(text: string): number;
+/**
+ * Anthropic Claude provider
+ */
+export declare class AnthropicProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private apiKey;
+    private model;
+    private baseUrl;
+    constructor(apiKey: string, model?: string, baseUrl?: string, sslVerify?: boolean);
+    countTokens(text: string): number;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    private parseError;
+}
+/**
+ * OpenAI provider
+ */
+export declare class OpenAIProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private apiKey;
+    private model;
+    private baseUrl;
+    constructor(apiKey: string, model?: string, baseUrl?: string, sslVerify?: boolean);
+    countTokens(text: string): number;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    private parseError;
+}
+/**
+ * Generic OpenAI-compatible provider.
+ * Works with any API that implements the OpenAI chat completions format:
+ * Mistral AI, Groq, Together AI, Ollama, LM Studio, etc.
+ *
+ * Required env vars:
+ *   OPENAI_COMPAT_API_KEY   — API key (use "ollama" for local setups without auth)
+ *   OPENAI_COMPAT_BASE_URL  — Base URL, e.g. https://api.mistral.ai/v1
+ */
+export declare class OpenAICompatibleProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private apiKey;
+    private model;
+    private baseUrl;
+    constructor(apiKey: string, baseUrl: string, model?: string);
+    countTokens(text: string): number;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+}
+/**
+ * Gemini CLI provider
+ *
+ * Routes LLM calls through the local `gemini` CLI binary in non-interactive
+ * mode (`gemini -p ...`).  Authentication is handled by the Google account
+ * session — no GEMINI_API_KEY is required.
+ * If the binary is not on PATH, set GEMINI_CLI to its full path.
+ */
+export declare class GeminiCLIProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private model;
+    constructor(model?: string);
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    countTokens(text: string): number;
+}
+/**
+ * Google Gemini provider
+ */
+export declare class GeminiProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private apiKey;
+    private model;
+    private baseUrl;
+    constructor(apiKey: string, model?: string);
+    countTokens(text: string): number;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+}
+/**
+ * Mock provider for testing
+ */
+export declare class MockLLMProvider implements LLMProvider {
+    name: string;
+    maxContextTokens: number;
+    maxOutputTokens: number;
+    private responses;
+    private defaultResponse;
+    callHistory: CompletionRequest[];
+    shouldFail: boolean;
+    failCount: number;
+    private currentFailCount;
+    setResponse(promptContains: string, response: string): void;
+    setDefaultResponse(response: string): void;
+    countTokens(text: string): number;
+    generateCompletion(request: CompletionRequest): Promise<CompletionResponse>;
+    reset(): void;
+}
+/**
+ * LLM Service - main interface for LLM interactions
+ */
+export declare class LLMService {
+    private provider;
+    private retryConfig;
+    private options;
+    private tokenUsage;
+    private costTracking;
+    private requestLog;
+    constructor(provider: LLMProvider, options?: LLMServiceOptions);
+    /**
+     * Get the provider name
+     */
+    getProviderName(): string;
+    /**
+     * Get maximum context tokens for the provider
+     */
+    getMaxContextTokens(): number;
+    /**
+     * Count tokens in text
+     */
+    countTokens(text: string): number;
+    /**
+     * Get current token usage
+     */
+    getTokenUsage(): TokenUsage;
+    /**
+     * Get current cost tracking
+     */
+    getCostTracking(): CostTracking;
+    /**
+     * Reset usage tracking
+     */
+    resetTracking(): void;
+    /**
+     * Generate a completion with retry logic
+     */
+    complete(request: CompletionRequest): Promise<CompletionResponse>;
+    /**
+     * Generate a completion expecting JSON response
+     */
+    completeJSON<T>(request: CompletionRequest, schema?: object): Promise<T>;
+    /**
+     * Execute request with timeout
+     */
+    private executeWithTimeout;
+    /**
+     * Update tracking after a successful request
+     */
+    private updateTracking;
+    /**
+     * Calculate cost for a response
+     */
+    private calculateCost;
+    /**
+     * Log request/response
+     */
+    private logRequest;
+    /**
+     * Redact potential secrets from request
+     */
+    private redactSecrets;
+    /**
+     * Simple schema validation
+     */
+    private validateSchema;
+    /**
+     * Save logs to disk
+     */
+    saveLogs(): Promise<void>;
+    /**
+     * Sleep helper
+     */
+    private sleep;
+}
+/**
+ * Create an LLM service with the specified provider
+ */
+export declare function createLLMService(options?: LLMServiceOptions): LLMService;
+/**
+ * Create an LLM service with a mock provider (for testing)
+ */
+export declare function createMockLLMService(options?: LLMServiceOptions): {
+    service: LLMService;
+    provider: MockLLMProvider;
+};
+//# sourceMappingURL=llm-service.d.ts.map

package/dist/core/services/llm-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm-service.d.ts","sourceRoot":"","sources":["../../../src/core/services/llm-service.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAUH;;;;;;GAMG;AACH,qBAAa,kBAAmB,YAAW,WAAW;IACpD,IAAI,SAAiB;IACrB,gBAAgB,SAAW;IAC3B,eAAe,SAAU;IACzB,OAAO,CAAC,KAAK,CAAqB;gBAEtB,KAAK,CAAC,EAAE,MAAM;IAOpB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsDjF,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;CAGlC;AAMD;;;;;;;GAOG;AACH,qBAAa,mBAAoB,YAAW,WAAW;IACrD,IAAI,SAAkB;IACtB,gBAAgB,SAAW;IAC3B,eAAe,SAAS;IACxB,OAAO,CAAC,KAAK,CAAqB;gBAEtB,KAAK,CAAC,EAAE,MAAM;IAKpB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAuEjF,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;CAGlC;AAMD;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,GAAG,QAAQ,GAAG,OAAO,CAAC;CAC3C;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC5E,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,gBAAgB,EAAE,MAAM,CAAC;IACzB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,MAAM,YAAY,GAAG,WAAW,GAAG,QAAQ,GAAG,eAAe,GAAG,QAAQ,GAAG,YAAY,GAAG,aAAa,GAAG,cAAc,CAAC;AAE/H;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,8BAA8B;IAC9B,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,qBAAqB;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iFAAiF;IACjF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,uEAAuE;IACvE,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,qFAAqF;IACrF,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,6BAA6B;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gCAAgC;IAChC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gCAAgC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,4BAA4B;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oCAAoC;IACpC,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,0CAA0C;IAC1C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,4BAA4B;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AA6DD;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,MAAM,GAAG,SAAS,CA4BpG;AAuED;;;;;;;GAOG;AACH,wBAAgB,aAAa,CAC3B,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,MAAM,GACd;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAmBnC;AAMD;;;GAGG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CASnD;AAMD;;GAEG;AACH,qBAAa,iBAAkB,YAAW,WAAW;IACnD,IAAI,SAAe;IACnB,gBAAgB,SAAU;IAC1B,eAAe,SAAQ;IAEvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,KAAK,SAA+B,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,UAAO;IAOpG,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI3B,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkDjF,OAAO,CAAC,UAAU;CASnB;AAMD;;GAEG;AACH,qBAAa,cAAe,YAAW,WAAW;IAChD,IAAI,SAAY;IAChB,gBAAgB,SAAU;IAC1B,eAAe,SAAQ;IAEvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,KAAK,SAAW,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,UAAO;IAOhF,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI3B,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAmDjF,OAAO,CAAC,UAAU;CASnB;AAMD;;;;;;;;GAQG;AACH,qBAAa,wBAAyB,YAAW,WAAW;IAC1D,IAAI,SAAmB;IACvB,gBAAgB,SAAU;IAC1B,eAAe,SAAQ;IAEvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,SAAyB;IAM3E,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI3B,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;CAqDlF;AAMD;;;;;;;GAOG;AACH,qBAAa,iBAAkB,YAAW,WAAW;IACnD,IAAI,SAAgB;IACpB,gBAAgB,SAAa;IAC7B,eAAe,SAAS;IACxB,OAAO,CAAC,KAAK,CAAqB;gBAEtB,KAAK,CAAC,EAAE,MAAM;IAIpB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAiEjF,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;CAGlC;AAMD;;GAEG;AACH,qBAAa,cAAe,YAAW,WAAW;IAChD,IAAI,SAAY;IAChB,gBAAgB,SAAW;IAC3B,eAAe,SAAQ;IAEvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAA6D;gBAEhE,MAAM,EAAE,MAAM,EAAE,KAAK,SAAqB;IAKtD,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI3B,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA4DlF;AAMD;;GAEG;AACH,qBAAa,eAAgB,YAAW,WAAW;IACjD,IAAI,SAAU;IACd,gBAAgB,SAAU;IAC1B,eAAe,SAAQ;IAEvB,OAAO,CAAC,SAAS,CAAkC;IACnD,OAAO,CAAC,eAAe,CAAiC;IACjD,WAAW,EAAE,iBAAiB,EAAE,CAAM;IACtC,UAAU,UAAS;IACnB,SAAS,SAAK;IACrB,OAAO,CAAC,gBAAgB,CAAK;IAE7B,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAI3D,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAI1C,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI3B,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAmCjF,KAAK,IAAI,IAAI;CAOd;AAMD;;GAEG;AACH,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,OAAO,CAA8B;IAC7C,OAAO,CAAC,UAAU,CAAgF;IAClG,OAAO,CAAC,YAAY,CAAuE;IAC3F,OAAO,CAAC,UAAU,CAA+G;gBAErH,QAAQ,EAAE,WAAW,EAAE,OAAO,GAAE,iBAAsB;IAwBlE;;OAEG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,mBAAmB,IAAI,MAAM;IAI7B;;OAEG;IACH,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIjC;;OAEG;IACH,aAAa,IAAI,UAAU;IAI3B;;OAEG;IACH,eAAe,IAAI,YAAY;IAI/B;;OAEG;IACH,aAAa,IAAI,IAAI;IAMrB;;OAEG;IACG,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAqEvE;;OAEG;IACG,YAAY,CAAC,CAAC,EAAE,OAAO,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC;IAoE9E;;OAEG;YACW,kBAAkB;IAchC;;OAEG;IACH,OAAO,CAAC,cAAc;IAYtB;;OAEG;IACH,OAAO,CAAC,aAAa;IAOrB;;OAEG;IACH,OAAO,CAAC,UAAU;IAWlB;;OAEG;IACH,OAAO,CAAC,aAAa;IAiBrB;;OAEG;IACH,OAAO,CAAC,cAAc;IAatB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAmB/B;;OAEG;IACH,OAAO,CAAC,KAAK;CAGd;AAMD;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,GAAE,iBAAsB,GAAG,UAAU,CAkD5E;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,GAAE,iBAAsB,GAAG;IAAE,OAAO,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE,eAAe,CAAA;CAAE,CAIxH"}