@michelabboud/visual-forge-mcp 0.7.0 → 0.9.0

package/CHANGELOG.md

@@ -9,6 +9,278 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ---
 
+## [0.9.0] - 2026-01-16
+
+### Added 🧪 **Model Testing & Comparison System**
+
+#### Model Selection Workflow (Phase 1)
+- **Enhanced Provider Configuration**: `configure_provider` now returns available models
+  - Shows list of all models offered by the provider
+  - Displays model details: name, cost, description, capabilities
+  - Indicates which model is currently set as default
+  - Guides user to use `set_default_model` for customization
+- **New MCP Tool: `set_default_model`**
+  - Choose preferred model for a provider
+  - Validates model exists before saving
+  - Persists choice in `~/.visual-forge-mcp/user-config.json`
+  - Applies to all future generations unless overridden
+- **New MCP Tool: `get_model_info`**
+  - Retrieve detailed information about a specific model
+  - Shows capabilities, pricing, and description
+  - Displays test results if model was previously tested
+  - Helps users make informed model selection decisions
+- **User Configuration Manager**: `src/utils/user-config-manager.ts`
+  - Persistent storage for user preferences
+  - Default model per provider
+  - Model test results with quality scores
+  - Atomic file writes for crash safety
+
+#### Model Testing (Phase 2)
+- **ModelTester Utility**: `src/quality/model-tester.ts`
+  - Comprehensive testing framework for AI models
+  - Standard automated tests and custom prompt tests
+  - Multi-provider comparison with side-by-side results
+  - Quality scoring algorithm with weighted metrics
+
+- **Quality Scoring Algorithm**:
+  - **Sharpness (30%)**: Laplacian variance analysis for edge detection
+  - **Brightness (20%)**: Average brightness in optimal range (30-240)
+  - **Text Rendering (40%)**: Estimated OCR accuracy and text clarity
+  - **Color Accuracy (10%)**: Heuristic-based color validation
+  - **Overall Score**: Weighted average, pass threshold 60/100
+  - **Auto-Recording**: Test results saved for future reference
+
+- **Standard Test Prompt**:
+  - Professional quality validation image
+  - Multiple text elements (title, subtitle, detailed text)
+  - Geometric shapes (red circle, blue square, green triangle)
+  - Color gradient background (#1a365d to #0891b2)
+  - Technical diagram (simple flowchart)
+  - Designed to test sharpness, color accuracy, and text rendering
+
+- **New MCP Tool: `test_model`**
+  - Two modes:
+    - **Standard Test**: Automated quality validation with predefined elements
+    - **Custom Prompt**: Test with user's actual use case
+  - Cost permission flow (requires confirmation for paid models)
+  - Automatic quality scoring and pass/fail determination
+  - Test images saved to `generated-images/tests/`
+  - Results include quality metrics, generation time, and cost
+
+- **New MCP Tool: `compare_models`**
+  - Side-by-side comparison of multiple providers/models
+  - Same prompt tested across all selected models
+  - Automatic ranking by quality score
+  - Intelligent recommendation with reasoning
+  - Alternative suggestions for budget-conscious users
+  - Total cost and time tracking
+  - Detailed results per model with success/failure handling
+
+- **Permission Flow**:
+  - First call shows cost estimate and requires confirmation
+  - User confirms by calling again with `skipPermission: true`
+  - Prevents accidental spending on paid models
+  - Free models (Gemini, HuggingFace) can skip permission
+
+#### Files Added
+- `src/quality/model-tester.ts` - Model testing and comparison utility
+- `docs/design/model-selection-workflow.md` - Complete workflow specification
+- `docs/design/IMPLEMENTATION_STATUS.md` - Implementation tracking
+
+#### Files Modified
+- `src/server/mcp-server.ts`:
+  - Added `userConfigManager` import
+  - Added `ModelTester` instance
+  - Updated `configure_provider` handler to return models
+  - Added `handleSetDefaultModel()` handler
+  - Added `handleGetModelInfo()` handler
+  - Added `handleTestModel()` handler
+  - Added `handleCompareModels()` handler
+  - Added routing for 5 new tools
+  - Added 'zai' to all provider enums
+- `src/quality/index.ts` - Exported ModelTester
+- `src/utils/index.ts` - Exported UserConfigManager (from 0.8.0)
+
+### Benefits
+
+#### For Users
+- ✅ **Informed Decisions**: Test before committing to a model
+- ✅ **Quality Assurance**: Validate model performance on your use case
+- ✅ **Cost Awareness**: See estimated costs before generation
+- ✅ **Traceability**: Historical test results for comparison
+- ✅ **Flexibility**: Different models for different use cases
+- ✅ **Side-by-Side Comparison**: Objective ranking with recommendations
+
+#### For Workflows
+- ✅ **Standard Tests**: Quick automated validation of model quality
+- ✅ **Custom Tests**: Real-world testing with actual prompts
+- ✅ **Multi-Provider**: Compare 2-8 providers in single call
+- ✅ **Automatic Ranking**: Data-driven recommendations
+- ✅ **Error Handling**: Graceful failures don't block comparisons
+
+### Example Usage
+
+```typescript
+// 1. Configure provider and see models
+configure_provider({ provider: "zai", apiKey: "zai-..." })
+// Returns: List of available models with costs
+
+// 2. Set preferred model
+set_default_model({ provider: "zai", modelId: "glm-image" })
+
+// 3. Test with standard prompt (automated)
+test_model({
+  provider: "zai",
+  modelId: "glm-image",
+  useStandardTest: true
+})
+
+// 4. Test with custom prompt (real use case)
+test_model({
+  provider: "gemini",
+  modelId: "gemini-2.5-flash-image",
+  prompt: "AWS VPC architecture diagram with public/private subnets"
+})
+
+// 5. Compare multiple models
+compare_models({
+  prompt: "Technical diagram showing microservices architecture",
+  providers: [
+    { provider: "gemini", model: "gemini-2.5-flash-image" },
+    { provider: "zai", model: "glm-image" },
+    { provider: "huggingface", model: "black-forest-labs/FLUX.1-dev" }
+  ]
+})
+// Returns: Ranked results with recommendation
+```
+
+### Technical Details
+
+- **Test Image Storage**: `generated-images/tests/`
+- **Config Storage**: `~/.visual-forge-mcp/user-config.json`
+- **Quality Metrics**: Sharpness, brightness, text rendering, color accuracy
+- **Pass Threshold**: 60/100 overall score
+- **Concurrent Testing**: Up to 3 parallel model tests in comparison mode
+
+---
+
+## [0.8.0] - 2026-01-16
+
+### Added 🎯 **Multi-Model Architecture & Z.ai Provider**
+
+#### Multi-Model Architecture
+- **Provider-Model Separation**: Distinguish between providers (companies) and models (specific AI implementations)
+  - One provider can offer multiple models with different capabilities and pricing
+  - Example: OpenAI offers `gpt-image-1` (standard, $0.04) and `gpt-image-1-hd` (HD, $0.12)
+  - Example: Gemini offers `gemini-2.5-flash-image` (2K) and `gemini-2.5-flash-image-pro` (4K)
+  - Example: Replicate offers `flux-schnell` ($0.003), `flux-dev` ($0.025), `flux-pro` ($0.055)
+- **New Type System**:
+  - `ProviderModel` interface: Represents individual models with id, name, cost, capabilities
+  - `IImageProvider.getModels()`: Returns array of available models
+  - `IImageProvider.getDefaultModel()`: Returns provider's default model
+  - `IImageProvider.estimateCost(spec, modelId?)`: Model-aware cost estimation
+- **Backward Compatibility**:
+  - Optional `models` and `defaultModel` fields in `ProviderConfig`
+  - Legacy `model` field still supported for single-model providers
+  - Automatic fallback to legacy config if models array not provided
+- **Files Modified**:
+  - `src/types/provider.ts` - Added ProviderModel interface, updated IImageProvider
+  - `src/providers/base-provider.ts` - Implemented getModels(), getDefaultModel(), updated estimateCost()
+  - `src/providers/index.ts` - Updated all provider initializations
+  - `config/pricing.json` - Restructured to support models array per provider (v2.1.0)
+- **Benefits**:
+  - ✅ Future-proof architecture for providers with multiple model offerings
+  - ✅ Granular cost control per model
+  - ✅ Better model selection and capabilities reporting
+  - ✅ Full backward compatibility with existing code
+
+#### Z.ai GLM-Image Provider (8th Provider)
+- **New Provider**: Z.ai (Zhipu AI) with GLM-Image model
+  - **Model**: GLM-Image - 16B parameter hybrid autoregressive + diffusion model
+  - **Specialty**: Excellent for text-heavy diagrams, posters, and knowledge-dense images
+  - **Performance**: Beats Google Gemini on CVTG-2k text rendering benchmark (0.9116 vs 0.7788)
+  - **Pricing**: $0.015 per image (2nd cheapest paid option after Replicate $0.003)
+  - **Rate Limit**: 15 images/minute
+  - **Resolution**: Up to 2048x2048
+  - **Supported Formats**: PNG
+- **Files Added**:
+  - `src/providers/zai/zai-provider.ts` - Complete Z.ai provider implementation
+  - `src/types/generation.ts` - Added 'zai' to ProviderType
+- **Configuration**:
+  - Environment variable: `ZAI_API_KEY=zai-...`
+  - Runtime configuration via `configure_provider` MCP tool
+- **Features**:
+  - Custom prompt adaptation for text-heavy content
+  - Emphasizes accurate text rendering and clear labels
+  - Professional diagram focus
+  - Multilingual support
+- **Benefits**:
+  - ✅ Best-in-class text rendering for technical diagrams
+  - ✅ Cost-effective at $0.015/image
+  - ✅ Perfect for documentation with text-heavy visualizations
+
+#### Comprehensive Documentation
+- **New Guide**: `docs/guides/comprehensive-guide.md` - Complete user and developer reference
+  - **Description**: Overview of Visual Forge MCP features and use cases
+  - **Installation**: Step-by-step setup from prerequisites to MCP client configuration
+  - **Architecture**: Detailed system components and data flow diagrams
+  - **Environment Variables**: Complete reference for all configuration options
+  - **Provider & Model System**: Guide to 8 providers, model selection, capabilities
+  - **Usage Workflows**: Basic to advanced workflows with code examples
+  - **MCP Tools Reference**: All 13+ MCP tools with parameters and response examples
+  - **Testing**: Test infrastructure, running tests, writing tests, manual testing
+  - **Troubleshooting**: Common issues and debug strategies
+  - **Advanced Topics**: Custom providers, pricing config, backups, optimization
+- **Updated**:
+  - `README.md` - Added Z.ai provider, multi-model architecture feature, comprehensive guide link
+  - Version badge updated to 0.8.0
+  - Provider count updated to 8
+  - Added `ZAI_API_KEY` to environment variable examples
+
+### Changed
+
+#### Provider Priority Order
+- **New Default Priority**: Updated to prioritize cost-effectiveness
+  1. `replicate` - Cheapest ($0.003)
+  2. `zai` - 2nd cheapest ($0.015) ✨ NEW
+  3. `gemini` - Free tier
+  4. `huggingface` - Free tier
+  5. Others by cost
+
+#### Pricing Configuration
+- **Updated**: `config/pricing.json` version 2.1.0
+  - Added Z.ai provider with GLM-Image model
+  - Updated cost comparison to highlight Z.ai as 2nd cheapest
+  - Added recommended provider section (Z.ai for technical documentation)
+  - Complete model specifications for all providers
+
+### Fixed
+
+#### TypeScript Type Safety
+- **Fixed**: Handling of optional `models` field in ProviderConfig
+  - BaseProvider.init() now safely accesses models array
+  - BaseProvider.getDefaultModel() handles undefined models array
+  - BaseProvider.estimateCost() checks models existence before access
+  - ZaiProvider.generateImage() properly extracts prompt text
+- **Fixed**: Removed unused imports (ProviderModel, fs, path, fileURLToPath from ProviderFactory)
+- **Resolved**: All TypeScript compilation errors for multi-model architecture
+
+#### Backward Compatibility
+- **Ensured**: Legacy single-model providers continue to work
+  - `config.model` field still supported
+  - Automatic conversion to ProviderModel format
+  - Fallback to legacy cost if models array missing
+
+### Testing
+
+#### Test Results
+- **All Tests Passed**: 77 tests across 4 test suites
+  - ProviderFactory: 21 tests (provider initialization, selection, fallback, priority)
+  - All existing tests remain green with multi-model changes
+  - Backward compatibility verified
+
+---
+
 ## [0.7.0] - 2026-01-16
 
 ### Added 🎨 **Professional Image Generation Pipeline**

package/README.md

@@ -7,7 +7,7 @@
 An MCP (Model Context Protocol) server that automates AI image generation for technical documentation. Parse markdown files containing image prompts and generate professional images using multiple AI providers.
 
 [![npm version](https://img.shields.io/npm/v/@michelabboud/visual-forge-mcp.svg)](https://www.npmjs.com/package/@michelabboud/visual-forge-mcp)
-[![Version](https://img.shields.io/badge/version-0.4.0-blue.svg)](https://github.com/michelabboud/visual-forge-mcp)
+[![Version](https://img.shields.io/badge/version-0.9.0-blue.svg)](https://github.com/michelabboud/visual-forge-mcp)
 [![License](https://img.shields.io/badge/License-Personal%20Use%20Only-red.svg)](LICENSE)
 [![Node.js](https://img.shields.io/badge/node-%3E%3D24.0.0-brightgreen.svg)](https://nodejs.org/)
 
@@ -16,7 +16,7 @@ An MCP (Model Context Protocol) server that automates AI image generation for te
 ## 🌟 Features
 
 - **📝 Multi-Format Support**: Extract image specifications from Markdown (.md) and HTML (.html) files
-- **🔌 Multi-Provider Support**: 7 AI providers with automatic fallback
+- **🔌 Multi-Provider Support**: 8 AI providers with automatic fallback
   - OpenAI GPT Image
   - Google Gemini 2.5 Flash Image (Nano Banana)
   - Stability AI SDXL
@@ -24,6 +24,14 @@ An MCP (Model Context Protocol) server that automates AI image generation for te
   - Leonardo Phoenix
   - HuggingFace Inference
   - xAI Grok 2 Image
+  - Z.ai GLM-Image ✨ NEW - Excellent for text-heavy diagrams
+- **🎯 Multi-Model Architecture** ✨ NEW v0.8.0: Each provider can offer multiple models with different capabilities and pricing (e.g., OpenAI GPT Image 1 vs GPT Image 1 HD)
+- **🧪 Model Testing & Comparison** ✨ NEW v0.9.0: Test and compare AI models before production use
+  - Standard automated tests with quality scoring (sharpness, brightness, text rendering, color accuracy)
+  - Custom prompt testing with real use cases
+  - Side-by-side multi-provider comparison with intelligent recommendations
+  - Permission flow for cost-aware testing
+  - Persistent test results for historical tracking
 - **🎨 Detailed Global Context**: Comprehensive styling system with hex colors, typography, layout rules, and audience targeting for dramatically better, more consistent images
 - **🖼️ Multi-Format Optimization** ✨ NEW v0.7.0: Automatic generation of WebP (94% smaller), JPEG (85% smaller), and optional lossy PNG (70% smaller) with professional watermarking
 - **🔍 Quality Validation & Auto-Regeneration** ✨ NEW v0.7.0: OCR-based text detection, sharpness/brightness analysis, and automatic retry on quality failure
@@ -96,6 +104,9 @@ OPENAI_API_KEY=sk-...
 # xAI Grok 2 Image ($0.07/image)
 XAI_API_KEY=xai-...
 
+# Z.ai GLM-Image ($0.015/image) - NEW: Excellent for text-heavy diagrams
+ZAI_API_KEY=zai-...
+
 # Leonardo Phoenix ($0.02/image)
 LEONARDO_API_KEY=...
 
@@ -115,6 +126,7 @@ IMAGE_GEN_DEFAULT_PROVIDER=gemini
 - **Stability AI**: [https://platform.stability.ai](https://platform.stability.ai)
 - **OpenAI**: [https://platform.openai.com](https://platform.openai.com)
 - **xAI**: [https://console.x.ai](https://console.x.ai)
+- **Z.ai**: [https://z.ai](https://z.ai) ✨ NEW - Best for text-heavy diagrams
 - **Leonardo**: [https://leonardo.ai](https://leonardo.ai)
 - **HuggingFace**: [https://huggingface.co](https://huggingface.co)
 
@@ -928,6 +940,187 @@ Remove API key for a provider.
 
 **Example**: "Remove my OpenAI API key"
 
+### Model Selection & Testing Tools ✨ NEW v0.9.0
+
+#### `set_default_model`
+Set the default model for a provider. This model will be used for all future generations unless overridden.
+
+```json
+{
+  "provider": "zai",
+  "modelId": "glm-image"
+}
+```
+
+**Example**: "Set Z.ai to use the GLM-Image model"
+
+**Returns**:
+```json
+{
+  "success": true,
+  "provider": "zai",
+  "modelId": "glm-image",
+  "modelName": "GLM-Image",
+  "message": "Default model set to 'GLM-Image' for Z.ai GLM-Image..."
+}
+```
+
+#### `get_model_info`
+Get detailed information about a specific model, including test results if available.
+
+```json
+{
+  "provider": "gemini",
+  "modelId": "gemini-2.5-flash-image"
+}
+```
+
+**Example**: "What are the details for Gemini Flash Image?"
+
+**Returns**:
+```json
+{
+  "success": true,
+  "provider": "gemini",
+  "providerName": "Google Gemini 2.5 Flash Image",
+  "model": {
+    "id": "gemini-2.5-flash-image",
+    "name": "Gemini 2.5 Flash Image",
+    "costPerImage": 0.0,
+    "description": "Fast, free-tier image generation",
+    "capabilities": {
+      "maxResolution": "2048x2048",
+      "supportedAspectRatios": ["1:1", "16:9", "4:3", "9:16"]
+    }
+  },
+  "testResult": {
+    "testedAt": "2026-01-16T10:30:00.000Z",
+    "qualityScore": 85.5,
+    "passed": true
+  },
+  "message": "Model tested on 1/16/2026 with quality score 85.5/100"
+}
+```
+
+#### `test_model`
+Test a model with either standard test (automated) or custom prompt (user-provided). Records quality score for future reference.
+
+**Standard Test (Automated Validation)**:
+```json
+{
+  "provider": "zai",
+  "modelId": "glm-image",
+  "useStandardTest": true,
+  "aspectRatio": "16:9"
+}
+```
+
+**Custom Prompt Test (Real Use Case)**:
+```json
+{
+  "provider": "gemini",
+  "modelId": "gemini-2.5-flash-image",
+  "prompt": "AWS VPC architecture diagram showing public/private subnets, NAT gateway, and EC2 instances",
+  "aspectRatio": "16:9"
+}
+```
+
+**Example**: "Test the Z.ai GLM-Image model with a standard quality test"
+
+**Returns**:
+```json
+{
+  "success": true,
+  "provider": "zai",
+  "providerName": "Z.ai GLM-Image",
+  "model": "GLM-Image",
+  "testImage": {
+    "filepath": "generated-images/tests/zai-glm-image-test.png",
+    "generationTime": 12000,
+    "actualCost": 0.015
+  },
+  "qualityScore": {
+    "overall": 87.5,
+    "sharpness": 89.2,
+    "brightness": 145,
+    "textRendering": 85.0,
+    "colorAccuracy": 90.0,
+    "passed": true
+  },
+  "message": "Model test passed! Quality score: 87.5/100. Model is ready for production use."
+}
+```
+
+**Quality Metrics**:
+- **Sharpness (30%)**: Laplacian variance analysis for edge detection
+- **Brightness (20%)**: Average brightness in optimal range (30-240)
+- **Text Rendering (40%)**: Estimated OCR accuracy and text clarity
+- **Color Accuracy (10%)**: Heuristic-based color validation
+- **Overall Score**: Weighted average, pass threshold 60/100
+
+#### `compare_models`
+Compare multiple providers/models side-by-side with the same prompt. Generates quality scores and recommendation.
+
+```json
+{
+  "prompt": "Technical diagram showing microservices architecture with API gateway, service mesh, and databases",
+  "providers": [
+    { "provider": "gemini", "model": "gemini-2.5-flash-image" },
+    { "provider": "zai", "model": "glm-image" },
+    { "provider": "huggingface", "model": "black-forest-labs/FLUX.1-dev" }
+  ],
+  "aspectRatio": "16:9"
+}
+```
+
+**Example**: "Compare Gemini, Z.ai, and HuggingFace FLUX models for generating a microservices architecture diagram"
+
+**Returns**:
+```json
+{
+  "success": true,
+  "prompt": "Technical diagram showing microservices architecture...",
+  "totalCost": 0.015,
+  "totalTime": 35000,
+  "results": [
+    {
+      "provider": "zai",
+      "model": "GLM-Image",
+      "qualityScore": { "overall": 92.1, "textRendering": 95.8 },
+      "cost": 0.015,
+      "rank": 1
+    },
+    {
+      "provider": "gemini",
+      "model": "Gemini Flash Image",
+      "qualityScore": { "overall": 85.5 },
+      "cost": 0.0,
+      "rank": 2
+    },
+    {
+      "provider": "huggingface",
+      "model": "FLUX.1-dev",
+      "qualityScore": { "overall": 78.3 },
+      "cost": 0.0,
+      "rank": 3
+    }
+  ],
+  "recommendation": {
+    "provider": "zai",
+    "model": "glm-image",
+    "reason": "Highest overall quality (92.1/100), especially excellent text rendering (95.8/100). Worth the $0.015 cost for technical diagrams.",
+    "alternatives": [
+      {
+        "provider": "gemini",
+        "model": "gemini-2.5-flash-image",
+        "reason": "Free alternative with good quality (85.5/100)"
+      }
+    ]
+  },
+  "message": "Comparison complete! zai glm-image scored highest (rank 1)..."
+}
+```
+
 ### Image Generation Tools
 
 #### `parse_markdown`
@@ -1327,6 +1520,7 @@ For commercial licensing inquiries, please contact the author.
 ## 📚 Resources
 
 ### Documentation
+- **📖 Comprehensive Guide** ✨ NEW: Complete guide covering installation, architecture, environment variables, usage workflows, and testing - [docs/guides/comprehensive-guide.md](docs/guides/comprehensive-guide.md)
 - **Visual MCP Server Specification**: [docs/VISUAL_MCP_SERVER.md](docs/VISUAL_MCP_SERVER.md)
 - **Implementation Plan**: [docs/IMPLEMENTATION-PLAN.md](docs/IMPLEMENTATION-PLAN.md)
 - **Implementation Report**: [docs/IMPLEMENTATION_REPORT.md](docs/IMPLEMENTATION_REPORT.md)

package/config/pricing.json

@@ -1,8 +1,8 @@
 {
   "$schema": "./pricing-schema.json",
-  "version": "2.0.0",
-  "lastUpdated": "2026-01-13",
-  "lastChecked": "2026-01-15T17:00:08.513Z",
+  "version": "2.1.0",
+  "lastUpdated": "2026-01-16",
+  "lastChecked": "2026-01-16T06:15:00.000Z",
   "providers": {
     "openai": {
       "name": "OpenAI",
@@ -184,6 +184,24 @@
           "notes": "Preview model, pricing may change"
         }
       }
+    },
+    "zai": {
+      "name": "Z.ai (Zhipu AI)",
+      "pricingUrl": "https://docs.z.ai/guides/overview/pricing",
+      "defaultModel": "glm-image",
+      "models": {
+        "glm-image": {
+          "name": "GLM-Image",
+          "costPerImage": 0.015,
+          "rateLimit": 15,
+          "timeout": 90000,
+          "maxDimensions": {
+            "width": 2048,
+            "height": 2048
+          },
+          "notes": "16B hybrid autoregressive + diffusion model. Excellent for text-heavy diagrams, posters, and knowledge-dense images. Beats Gemini on text rendering benchmarks."
+        }
+      }
     }
   },
   "costComparison": {
@@ -192,10 +210,20 @@
       "model": "black-forest-labs/flux-schnell",
       "cost": 0.003
     },
+    "secondCheapest": {
+      "provider": "zai",
+      "model": "glm-image",
+      "cost": 0.015
+    },
     "recommended": {
+      "provider": "zai",
+      "model": "glm-image",
+      "reason": "Best for text-heavy diagrams and technical documentation. Excellent text rendering, 2nd cheapest paid option."
+    },
+    "recommendedAlternative": {
       "provider": "replicate",
       "model": "black-forest-labs/flux-schnell",
-      "reason": "Best quality/cost ratio"
+      "reason": "Cheapest option, excellent quality/cost ratio for general images"
     },
     "fastest": [
       {

package/dist/providers/base-provider.d.ts

@@ -1,7 +1,7 @@
 /**
  * Base provider class - all providers extend this
  */
-import { IImageProvider, ProviderType, ProviderConfig, ProviderCapabilities, GenerationOptions, ImageSpec, GlobalContext, GeneratedImage } from '../types/index.js';
+import { IImageProvider, ProviderType, ProviderConfig, ProviderCapabilities, ProviderModel, GenerationOptions, ImageSpec, GlobalContext, GeneratedImage } from '../types/index.js';
 import { HttpClient } from '../utils/index.js';
 export declare abstract class BaseProvider implements IImageProvider {
     abstract readonly name: ProviderType;
@@ -13,6 +13,14 @@ export declare abstract class BaseProvider implements IImageProvider {
      * Initialize provider (call after construction)
      */
     protected init(): void;
+    /**
+     * Get available models for this provider
+     */
+    getModels(): ProviderModel[];
+    /**
+     * Get default model for this provider
+     */
+    getDefaultModel(): ProviderModel;
     /**
      * Create output path for generated image with new index-based directory structure
      * Structure: generated-images/{index}-{filename}/{provider}/{image-id}.png
@@ -29,7 +37,7 @@ export declare abstract class BaseProvider implements IImageProvider {
      * Save image metadata and generation log
      */
     private saveImageMetadata;
-    estimateCost(_spec: ImageSpec): number;
+    estimateCost(_spec: ImageSpec, modelId?: string): number;
     /**
      * Provider-specific generation logic - must be implemented by subclasses
      */

package/dist/providers/base-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base-provider.d.ts","sourceRoot":"","sources":["../../src/providers/base-provider.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,cAAc,EACd,YAAY,EACZ,cAAc,EACd,oBAAoB,EACpB,iBAAiB,EACjB,SAAS,EACT,aAAa,EACb,cAAc,EACf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,UAAU,EAWX,MAAM,mBAAmB,CAAC;AAM3B,8BAAsB,YAAa,YAAW,cAAc;IAC1D,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC;IACrC,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAEtC,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC;IAC7B,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;gBAErB,MAAM,EAAE,cAAc;IAWlC;;OAEG;IACH,SAAS,CAAC,IAAI,IAAI,IAAI;IAStB;;;;;;;OAOG;cACa,gBAAgB,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,GAAE,MAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAmC7F,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED,YAAY,IAAI,OAAO;IASjB,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,CAAC;IA0MrF;;OAEG;YACW,iBAAiB;IAkH/B,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM;IAKtC;;OAEG;IACH,SAAS,CAAC,QAAQ,CAAC,aAAa,CAC9B,IAAI,EAAE,SAAS,EACf,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,cAAc,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,MAAM,GAAG,GAAG;IAE3E;;OAEG;IACH,QAAQ,CAAC,eAAe,IAAI,oBAAoB;IAEhD;;;OAGG;IACH,QAAQ,CAAC,cAAc,IAAI,OAAO,CAAC;QACjC,OAAO,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IAEF;;OAEG;IACH,SAAS,CAAC,cAAc,IAAI,IAAI;IAMhC;;OAEG;IACH,SAAS,CAAC,uBAAuB,CAAC,WAAW,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;CAY1F"}
+{"version":3,"file":"base-provider.d.ts","sourceRoot":"","sources":["../../src/providers/base-provider.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,cAAc,EACd,YAAY,EACZ,cAAc,EACd,oBAAoB,EACpB,aAAa,EACb,iBAAiB,EACjB,SAAS,EACT,aAAa,EACb,cAAc,EACf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,UAAU,EAWX,MAAM,mBAAmB,CAAC;AAM3B,8BAAsB,YAAa,YAAW,cAAc;IAC1D,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC;IACrC,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAEtC,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC;IAC7B,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;gBAErB,MAAM,EAAE,cAAc;IAWlC;;OAEG;IACH,SAAS,CAAC,IAAI,IAAI,IAAI;IAWtB;;OAEG;IACH,SAAS,IAAI,aAAa,EAAE;IAI5B;;OAEG;IACH,eAAe,IAAI,aAAa;IAgChC;;;;;;;OAOG;cACa,gBAAgB,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,GAAE,MAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAmC7F,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED,YAAY,IAAI,OAAO;IASjB,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,CAAC;IA0MrF;;OAEG;YACW,iBAAiB;IAkH/B,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM;IAkBxD;;OAEG;IACH,SAAS,CAAC,QAAQ,CAAC,aAAa,CAC9B,IAAI,EAAE,SAAS,EACf,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,cAAc,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,MAAM,GAAG,GAAG;IAE3E;;OAEG;IACH,QAAQ,CAAC,eAAe,IAAI,oBAAoB;IAEhD;;;OAGG;IACH,QAAQ,CAAC,cAAc,IAAI,OAAO,CAAC;QACjC,OAAO,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IAEF;;OAEG;IACH,SAAS,CAAC,cAAc,IAAI,IAAI;IAMhC;;OAEG;IACH,SAAS,CAAC,uBAAuB,CAAC,WAAW,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;CAY1F"}