feat(mistral): Enhance Mistral provider with new models and API integration

- Updated model aliases in aliases.json to include specific paths for Mistral models.
- Added `mistral_api_base` configuration option in configuration.rb.
- Introduced new models in models.json, including Mistral Tiny, Pixtral Large, and others.
- Refactored Mistral provider methods to support new API structure and error handling.
- Removed deprecated image generation methods from the Mistral provider.
- Updated tests to reflect changes in model handling and API responses.

Author: OriPekelman

lib/ruby_llm/aliases.json

@@ -78,12 +78,12 @@
     "openrouter": "openai/o3-mini"
   },
   "mistral-medium": {
-    "mistral": "mistral-medium-latest"
+    "mistral": "mistral/mistral-medium-latest"
   },
   "mistral-large": {
-    "mistral": "mistral-large-latest"
+    "mistral": "mistral/mistral-large"
   },
   "mistral-small": {
-    "mistral": "mistral-small-latest"
+    "mistral": "mistral/mistral-small"
   }
 }

lib/ruby_llm/configuration.rb

@@ -22,6 +22,7 @@ class Configuration
                   :bedrock_session_token,
                   :openrouter_api_key,
                   :ollama_api_base,
+                  :mistral_api_base,
                   :mistral_api_key,
                   # Default models
                   :default_model,

lib/ruby_llm/models.json

@@ -11517,8 +11517,62 @@
     }
   },
   {
-    "id": "mistralai/mistral-embed",
-    "created_at": "2024-03-15T00:00:00Z",
+    "id": "mistral-tiny",
+    "created_at": "2024-12-23T00:00:00Z",
+    "display_name": "Mistral Tiny",
+    "provider": "mistral",
+    "context_window": 32768,
+    "max_tokens": 32768,
+    "type": "chat",
+    "family": "mistral_tiny",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.14,
+    "output_price_per_million": 0.42,
+    "metadata": {
+      "description": "Most affordable model in the Mistral lineup"
+    }
+  },
+  {
+    "id": "pixtral-large",
+    "created_at": "2024-11-24T00:00:00Z",
+    "display_name": "Pixtral Large",
+    "provider": "mistral",
+    "context_window": 131072,
+    "max_tokens": 131072,
+    "type": "chat",
+    "family": "pixtral_large",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 3,
+    "output_price_per_million": 9,
+    "metadata": {
+      "description": "Pixtral Large - frontier-class multimodal model"
+    }
+  },
+  {
+    "id": "codestral",
+    "created_at": "2025-01-25T00:00:00Z",
+    "display_name": "Codestral",
+    "provider": "mistral",
+    "context_window": 262144,
+    "max_tokens": 262144,
+    "type": "chat",
+    "family": "codestral",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 1,
+    "output_price_per_million": 3,
+    "metadata": {
+      "description": "Codestral - specialized model for code generation"
+    }
+  },
+  {
+    "id": "mistral-embed",
+    "created_at": "2023-12-23T00:00:00Z",
     "display_name": "Mistral Embed",
     "provider": "mistral",
     "context_window": 32768,
@@ -11534,6 +11588,114 @@
       "description": "Mistral Embed - specialized model for generating text embeddings"
     }
   },
+  {
+    "id": "mistral-large-latest",
+    "created_at": "2024-11-24T00:00:00Z",
+    "display_name": "Mistral Large (Latest)",
+    "provider": "mistral",
+    "context_window": 131072,
+    "max_tokens": 131072,
+    "type": "chat",
+    "family": "mistral_large",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 3,
+    "output_price_per_million": 9,
+    "metadata": {
+      "description": "Mistral Large 2 - top-tier reasoning model for high-complexity tasks"
+    }
+  },
+  {
+    "id": "mistral-small-latest",
+    "created_at": "2025-03-25T00:00:00Z",
+    "display_name": "Mistral Small (Latest)",
+    "provider": "mistral",
+    "context_window": 131072,
+    "max_tokens": 131072,
+    "type": "chat",
+    "family": "mistral_small",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.2,
+    "output_price_per_million": 0.6,
+    "metadata": {
+      "description": "Mistral Small (Latest) - balanced model with good performance/cost ratio"
+    }
+  },
+  {
+    "id": "pixtral-12b-latest",
+    "created_at": "2024-09-25T00:00:00Z",
+    "display_name": "Pixtral 12B (Latest)",
+    "provider": "mistral",
+    "context_window": 131072,
+    "max_tokens": 131072,
+    "type": "chat",
+    "family": "pixtral",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.15,
+    "output_price_per_million": 0.15,
+    "metadata": {
+      "description": "Pixtral 12B - Mistral's first multimodal model with vision capabilities"
+    }
+  },
+  {
+    "id": "mistral-saba",
+    "created_at": "2025-02-25T00:00:00Z",
+    "display_name": "Mistral Saba",
+    "provider": "mistral",
+    "context_window": 32768,
+    "max_tokens": 32768,
+    "type": "chat",
+    "family": "mistral_saba",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.3,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "description": "Mistral Saba - powerful and efficient model for languages from the Middle East and South Asia"
+    }
+  },
+  {
+    "id": "mistral-moderation",
+    "created_at": "2024-11-24T00:00:00Z",
+    "display_name": "Mistral Moderation",
+    "provider": "mistral",
+    "context_window": 32768,
+    "max_tokens": 32768,
+    "type": "moderation",
+    "family": "mistral_moderation",
+    "supports_vision": false,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.0,
+    "output_price_per_million": 0.0,
+    "metadata": {
+      "description": "Mistral Moderation - service that enables users to detect harmful text content"
+    }
+  },
+  {
+    "id": "mistral-ocr",
+    "created_at": "2025-03-25T00:00:00Z",
+    "display_name": "Mistral OCR",
+    "provider": "mistral",
+    "context_window": 32768,
+    "max_tokens": 32768,
+    "type": "ocr",
+    "family": "mistral_ocr",
+    "supports_vision": true,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.0,
+    "output_price_per_million": 0.0,
+    "metadata": {
+      "description": "Mistral OCR - service that enables users to extract interleaved text and images"
+    }
+  },
   {
     "id": "moonshotai/kimi-vl-a3b-thinking:free",
     "created_at": "2025-04-10T19:07:21+02:00",

lib/ruby_llm/providers/mistral.rb

@@ -11,6 +11,7 @@ module Mistral
       extend Mistral::Tools
       extend Mistral::Capabilities
       extend Mistral::Media
+      # Mistral::Images removed as API doesn't support image generation
 
       def self.extended(base)
         base.extend(Provider)
@@ -25,13 +26,42 @@ def self.extended(base)
 
       module_function
 
-      def api_base
-        'https://api.mistral.ai/v1'
+      def api_base(config)
+        config.mistral_api_base || 'https://api.mistral.ai/v1'
       end
 
-      def headers
+      # Add provider-specific error parsing
+      def parse_error(response)
+        return if response.body.empty?
+
+        body = try_parse_json(response.body)
+        error_message = case body
+        when Hash
+          # Check for standard { detail: [{ msg: ... }] } structure
+          if body['detail'].is_a?(Array)
+            body['detail'].map { |err| err['msg'] }.join("; ")
+          # Check for simple { message: ... } structure (some Mistral errors use this)
+          elsif body['message']
+            body['message']
+          # Fallback for other hash structures
+          else
+            body.to_s
+          end
+        else
+          body.to_s # Return raw body if not a hash
+        end
+
+        # Format the error message
+        if error_message.include?('Input should be a valid')
+          "Invalid message format: The message content is not properly formatted"
+        else
+          error_message
+        end
+      end
+
+      def headers(config)
         {
-          'Authorization' => "Bearer #{RubyLLM.config.mistral_api_key}"
+          'Authorization' => "Bearer #{config.mistral_api_key}"
         }
       end
 

lib/ruby_llm/providers/mistral/capabilities.rb

@@ -51,17 +51,8 @@ def output_price_for(model_id)
         # @param model_id [String] the model identifier
         # @return [Boolean] true if the model supports vision
         def supports_vision?(model_id)
-          # Check for all vision-capable models in Mistral's lineup
-          #
-          # NOTE: While this correctly identifies models that support vision capabilities,
-          # there are currently issues with the image handling in the test suite.
-          # The 'pixtral-12b-latest can understand images' test fails because image data
-          # isn't properly passed to the API. This requires fixes in the core Content handling code.
-          #
-          # Known vision-capable models in Mistral's lineup:
-          # - pixtral-12b-latest
-          # - pixtral-large-latest
-          model_id.match?(/pixtral|mistral-large-vision|other-vision-models/)
+          # Determine vision support based on model ID pattern
+          model_id.match?(/pixtral/)
         end
 
         # Determines if the model supports function calling