[WIP] Create Mock Provider

Author: sirwolfgang

lib/active_agent/providers/mock/_types.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require_relative "options"
+require_relative "request"
+require_relative "embedding_request"
+
+module ActiveAgent
+  module Providers
+    module Mock
+      # Type for Request model
+      class RequestType < ActiveModel::Type::Value
+        def cast(value)
+          case value
+          when Request
+            value
+          when Hash
+            Request.new(**value.deep_symbolize_keys)
+          when nil
+            nil
+          else
+            raise ArgumentError, "Cannot cast #{value.class} to Request"
+          end
+        end
+
+        def serialize(value)
+          case value
+          when Request
+            value.serialize
+          when Hash
+            value
+          when nil
+            nil
+          else
+            raise ArgumentError, "Cannot serialize #{value.class}"
+          end
+        end
+
+        def deserialize(value)
+          cast(value)
+        end
+      end
+
+      # Type for EmbeddingRequest model
+      class EmbeddingRequestType < ActiveModel::Type::Value
+        def cast(value)
+          case value
+          when EmbeddingRequest
+            value
+          when Hash
+            EmbeddingRequest.new(**value.deep_symbolize_keys)
+          when nil
+            nil
+          else
+            raise ArgumentError, "Cannot cast #{value.class} to EmbeddingRequest"
+          end
+        end
+
+        def serialize(value)
+          case value
+          when EmbeddingRequest
+            value.serialize
+          when Hash
+            value
+          when nil
+            nil
+          else
+            raise ArgumentError, "Cannot serialize #{value.class}"
+          end
+        end
+
+        def deserialize(value)
+          cast(value)
+        end
+      end
+    end
+  end
+end

lib/active_agent/providers/mock/embedding_request.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "active_agent/providers/common/model"
+
+module ActiveAgent
+  module Providers
+    module Mock
+      # Embedding request model for Mock provider.
+      class EmbeddingRequest < Common::BaseModel
+        attribute :model, :string, default: "mock-embedding-model"
+        attribute :input # String or array of strings to embed
+        attribute :encoding_format, :string
+        attribute :dimensions, :integer
+      end
+    end
+  end
+end

lib/active_agent/providers/mock/messages/assistant.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require "active_agent/providers/common/model"
+
+module ActiveAgent
+  module Providers
+    module Mock
+      module Messages
+        # Assistant message for Mock provider.
+        #
+        # Drops extra fields that are part of the API response but not
+        # part of the message structure (usage, id, model, stop_reason, type, etc).
+        class Assistant < Common::BaseModel
+          attribute :role, :string, as: "assistant"
+          attribute :content # Can be string or array of content blocks
+          attribute :name, :string
+
+          validates :content, presence: true
+
+          # Drop API response fields that aren't part of the message
+          drop_attributes :usage, :id, :model, :stop_reason, :stop_sequence, :type
+
+          # Converts to common format.
+          #
+          # @return [Hash] message in canonical format with role and text content
+          def to_common
+            {
+              role: role,
+              content: extract_text_content,
+              name: name
+            }
+          end
+
+          private
+
+          # Extracts text content from the content field.
+          #
+          # @return [String] Extracted text
+          def extract_text_content
+            case content
+            when String
+              content
+            when Array
+              content
+                .select { |block| block.is_a?(Hash) && block[:type] == "text" }
+                .map { |block| block[:text] }
+                .join("\n")
+            else
+              content.to_s
+            end
+          end
+        end
+      end
+    end
+  end
+end

lib/active_agent/providers/mock/options.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require "active_agent/providers/common/model"
+
+module ActiveAgent
+  module Providers
+    module Mock
+      # Configuration options for the Mock provider.
+      #
+      # This provider doesn't make real API calls, so most options are unused.
+      # Included for consistency with other providers.
+      class Options < Common::BaseModel
+        attribute :base_url, :string, default: "https://mock.example.com"
+        attribute :api_key, :string, default: "mock-api-key"
+
+        # Common Interface Compatibility
+        alias_attribute :access_token, :api_key
+
+        def initialize(kwargs = {})
+          kwargs = kwargs.deep_symbolize_keys if kwargs.respond_to?(:deep_symbolize_keys)
+          super(**deep_compact(kwargs))
+        end
+
+        def serialize
+          super
+        end
+      end
+    end
+  end
+end

lib/active_agent/providers/mock/request.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require "active_agent/providers/common/model"
+require_relative "messages/assistant"
+
+module ActiveAgent
+  module Providers
+    module Mock
+      # Type for Messages array
+      class MessagesType < ActiveModel::Type::Value
+        def cast(value)
+          return [] if value.nil?
+          return value if value.is_a?(Array) && value.all? { |v| v.is_a?(Messages::Assistant) }
+
+          Array(value).map { |msg| cast_message(msg) }.compact
+        end
+
+        def serialize(value)
+          Array(value).map do |msg|
+            msg.respond_to?(:serialize) ? msg.serialize : msg
+          end
+        end
+
+        private
+
+        def cast_message(value)
+          case value
+          when Messages::Assistant
+            value
+          when Hash
+            role = value[:role]&.to_s || value["role"]&.to_s
+            case role
+            when "assistant"
+              Messages::Assistant.new(**value.deep_symbolize_keys)
+            else
+              # For other roles (user, system), just pass through as-is
+              value.deep_symbolize_keys
+            end
+          else
+            value
+          end
+        end
+      end
+
+      # Request model for Mock provider.
+      #
+      # Simplified request model that accepts messages and basic parameters.
+      class Request < Common::BaseModel
+        # Required parameters
+        attribute :model, :string, default: "mock-model"
+        attribute :messages, MessagesType.new
+
+        # Optional parameters
+        attribute :temperature, :float
+        attribute :max_tokens, :integer
+        attribute :stream, :boolean, default: false
+        attribute :tools # Array of tool definitions
+        attribute :tool_choice # Tool choice configuration
+
+        # Common Format Compatibility
+        def message=(value)
+          self.messages ||= []
+          self.messages << value
+        end
+      end
+    end
+  end
+end