pangea-aws 0.1.0

data/lib/pangea/resources/aws_braket_local_simulator/types.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'dry-struct'
+require 'pangea/resources/types'
+require_relative 'types/validators'
+require_relative 'types/helpers'
+
+module Pangea
+  module Resources
+    module AWS
+      module Types
+        # Type-safe attributes for AWS Braket Local Simulator resources
+        class BraketLocalSimulatorAttributes < Pangea::Resources::BaseAttributes
+          extend BraketLocalSimulatorValidators
+          include BraketLocalSimulatorHelpers
+
+          transform_keys(&:to_sym)
+
+          # Simulator name (required)
+          attribute? :simulator_name, Resources::Types::String.optional
+
+          # Simulator type (required)
+          attribute? :simulator_type, Resources::Types::String.enum(
+            'braket_sv',     # State vector simulator
+            'braket_dm',     # Density matrix simulator
+            'braket_tn'      # Tensor network simulator
+          )
+
+          # Configuration (required)
+          attribute? :configuration, Resources::Types::Hash.schema(
+            backend_configuration: Resources::Types::Hash.schema(
+              device_name: Resources::Types::String.constrained(included_in: ['braket_sv_v2',
+                'braket_dm_v2',
+                'braket_tn1']),
+              shots?: Resources::Types::Integer.constrained(gteq: 1, lteq: 100_000).optional,
+              max_parallel_shots?: Resources::Types::Integer.constrained(gteq: 1, lteq: 10_000).optional,
+              seed?: Resources::Types::Integer.optional
+            ),
+            resource_configuration?: Resources::Types::Hash.schema(
+              cpu_count: Resources::Types::Integer.constrained(gteq: 1, lteq: 96),
+              memory_size_mb: Resources::Types::Integer.constrained(gteq: 1024, lteq: 768_000),
+              gpu_count?: Resources::Types::Integer.constrained(gteq: 0, lteq: 8).optional
+            ).lax.optional,
+            advanced_configuration?: Resources::Types::Hash.schema(
+              enable_parallelization?: Resources::Types::Bool.optional,
+              optimization_level?: Resources::Types::Integer.constrained(gteq: 0, lteq: 3).optional,
+              precision?: Resources::Types::String.constrained(included_in: ['single', 'double']).optional
+            ).lax.optional
+          )
+
+          # Execution environment (optional)
+          attribute? :execution_environment, Resources::Types::Hash.schema(
+            docker_image?: Resources::Types::String.optional,
+            python_version?: Resources::Types::String.constrained(included_in: ['3.8', '3.9', '3.10', '3.11']).optional,
+            environment_variables?: Resources::Types::Hash.map(Resources::Types::String, Resources::Types::String).optional
+          ).lax.optional
+
+          # Tags (optional)
+          attribute? :tags, Resources::Types::Hash.map(Resources::Types::String, Resources::Types::String).optional
+
+          # Custom validation
+          def self.new(attributes = {})
+            attrs = super(attributes)
+
+            validate_simulator_name(attrs.simulator_name)
+
+            device_name = attrs.configuration&.dig(:backend_configuration)[:device_name]
+            validate_device_consistency(attrs.simulator_type, device_name)
+
+            resource_config = attrs.configuration&.dig(:resource_configuration)
+            validate_gpu_requirements(attrs.simulator_type, resource_config)
+            validate_memory_requirements(attrs.simulator_type, resource_config)
+
+            shots = attrs.configuration&.dig(:backend_configuration)[:shots]
+            validate_shots(attrs.simulator_type, shots)
+
+            attrs
+          end
+        end
+      end
+    end
+  end
+end

data/lib/pangea/resources/aws_braket_quantum_task/resource.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+require 'pangea/resources/base'
+require 'pangea/resources/reference'
+require 'pangea/resources/aws_braket_quantum_task/types'
+require 'pangea/resource_registry'
+
+module Pangea
+  module Resources
+    module AWS
+      # Create an AWS Braket Quantum Task with type-safe attributes
+      #
+      # @param name [Symbol] The resource name
+      # @param attributes [Hash] Braket quantum task attributes
+      # @return [ResourceReference] Reference object with outputs and computed properties
+      def aws_braket_quantum_task(name, attributes = {})
+        # Validate attributes using dry-struct
+        task_attrs = Types::BraketQuantumTaskAttributes.new(attributes)
+        
+        # Generate terraform resource block via terraform-synthesizer
+        resource(:aws_braket_quantum_task, name) do
+          # Set device ARN
+          device_arn task_attrs.device_arn
+          
+          # Set action JSON
+          action task_attrs.action
+          
+          # Set device parameters if provided
+          device_parameters task_attrs.device_parameters if task_attrs.device_parameters
+          
+          # Set output S3 location
+          output_s3_bucket task_attrs.output_s3_bucket
+          output_s3_key_prefix task_attrs.output_s3_key_prefix
+          
+          # Set shots count
+          shots task_attrs.shots
+          
+          # Set job token if provided
+          job_token task_attrs.job_token if task_attrs.job_token
+          
+          # Set tags if provided
+          if task_attrs.tags && !task_attrs.tags.empty?
+            tags task_attrs.tags
+          end
+        end
+        
+        # Return resource reference with available outputs
+        ResourceReference.new(
+          type: 'aws_braket_quantum_task',
+          name: name,
+          resource_attributes: task_attrs.to_h,
+          outputs: {
+            id: "${aws_braket_quantum_task.#{name}.id}",
+            arn: "${aws_braket_quantum_task.#{name}.arn}",
+            status: "${aws_braket_quantum_task.#{name}.status}",
+            output_s3_uri: "${aws_braket_quantum_task.#{name}.output_s3_uri}",
+            created_at: "${aws_braket_quantum_task.#{name}.created_at}",
+            ended_at: "${aws_braket_quantum_task.#{name}.ended_at}"
+          },
+          computed: {
+            device_type: task_attrs.device_type,
+            is_simulator: task_attrs.is_simulator?,
+            is_quantum_hardware: task_attrs.is_quantum_hardware?,
+            quantum_circuit: task_attrs.quantum_circuit,
+            estimated_cost: task_attrs.estimated_cost,
+            output_location: task_attrs.output_location,
+            action_summary: task_attrs.action_summary
+          }
+        )
+      end
+    end
+  end
+end

data/lib/pangea/resources/aws_braket_quantum_task/types.rb

@@ -0,0 +1,167 @@
+# frozen_string_literal: true
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'dry-struct'
+require 'pangea/resources/types'
+
+module Pangea
+  module Resources
+    module AWS
+      module Types
+      # Type-safe attributes for AWS Braket Quantum Task resources
+        class BraketQuantumTaskAttributes < Pangea::Resources::BaseAttributes
+        transform_keys(&:to_sym)
+
+        # Device ARN (required)
+        attribute? :device_arn, Resources::Types::String.optional
+
+        # Action JSON specification (required)
+        attribute? :action, Resources::Types::String.optional
+
+        # Device parameters (optional)
+        attribute? :device_parameters, Resources::Types::String.optional
+
+        # Output S3 bucket (required)
+        attribute? :output_s3_bucket, Resources::Types::String.optional
+
+        # Output S3 key prefix (required)
+        attribute? :output_s3_key_prefix, Resources::Types::String.optional
+
+        # Shots - number of times to run the quantum circuit (optional)
+        attribute? :shots, Resources::Types::Integer.default(1000)
+
+        # Job token for grouping tasks (optional)
+        attribute? :job_token, Resources::Types::String.optional
+
+        # Tags (optional)
+        attribute? :tags, Resources::Types::Hash.map(Resources::Types::String, Resources::Types::String).optional
+
+        # Custom validation
+        def self.new(attributes = {})
+          attrs = super(attributes)
+
+          # Validate shots is positive
+          if attrs.shots && attrs.shots <= 0
+            raise Dry::Struct::Error, "shots must be positive, got #{attrs.shots}"
+          end
+
+          # Validate shots is within reasonable limits for quantum devices
+          if attrs.shots && attrs.shots > 100000
+            raise Dry::Struct::Error, "shots cannot exceed 100000 for quantum devices, got #{attrs.shots}"
+          end
+
+          # Validate action is valid JSON
+          begin
+            ::JSON.parse(attrs.action)
+          rescue ::JSON::ParserError => e
+            raise Dry::Struct::Error, "action must be valid JSON: #{e.message}"
+          end
+
+          # Validate device_parameters is valid JSON if provided
+          if attrs.device_parameters
+            begin
+              ::JSON.parse(attrs.device_parameters)
+            rescue ::JSON::ParserError => e
+              raise Dry::Struct::Error, "device_parameters must be valid JSON: #{e.message}"
+            end
+          end
+
+          # Validate S3 bucket name format
+          unless attrs.output_s3_bucket.match?(/\A[a-z0-9][a-z0-9\-\.]*[a-z0-9]\z/)
+            raise Dry::Struct::Error, "output_s3_bucket must be a valid S3 bucket name"
+          end
+
+          # Validate S3 key prefix doesn't start with /
+          if attrs.output_s3_key_prefix.start_with?('/')
+            raise Dry::Struct::Error, "output_s3_key_prefix should not start with '/'"
+          end
+
+          attrs
+        end
+
+        # Helper methods
+        def device_type
+          case device_arn
+          when /simulator/i
+            :simulator
+          when /qpu/i
+            :quantum_processing_unit
+          when /sv1/i
+            :state_vector_simulator
+          when /dm1/i
+            :density_matrix_simulator
+          when /tn1/i
+            :tensor_network_simulator
+          else
+            :unknown
+          end
+        end
+
+        def is_simulator?
+          device_type != :quantum_processing_unit
+        end
+
+        def is_quantum_hardware?
+          device_type == :quantum_processing_unit
+        end
+
+        def quantum_circuit
+          @quantum_circuit ||= begin
+            action_data = ::JSON.parse(action)
+            action_data['braketSchemaHeader']['name'] if action_data['braketSchemaHeader']
+          rescue
+            nil
+          end
+        end
+
+        def estimated_cost
+          # Rough cost estimation based on device type and shots
+          base_cost = case device_type
+          when :quantum_processing_unit
+            0.30 # $0.30 per task for QPU
+          when :state_vector_simulator
+            0.075 # $0.075 per minute for SV1
+          when :density_matrix_simulator
+            0.075 # $0.075 per minute for DM1
+          when :tensor_network_simulator
+            0.275 # $0.275 per minute for TN1
+          else
+            0.00 # Free for local simulators
+          end
+
+          # Adjust for shots
+          shot_multiplier = (shots / 1000.0).ceil
+          base_cost * shot_multiplier
+        end
+
+        def output_location
+          "s3://#{output_s3_bucket}/#{output_s3_key_prefix}"
+        end
+
+        def action_summary
+          action_data = ::JSON.parse(action)
+          {
+            type: action_data['braketSchemaHeader']&.fetch('name', 'unknown'),
+            version: action_data['braketSchemaHeader']&.fetch('version', 'unknown'),
+            qubit_count: action_data['instructions']&.map { |i| i['target']&.max || 0 }&.max&.+(1) || 0
+          }
+        rescue
+          { type: 'unknown', version: 'unknown', qubit_count: 0 }
+        end
+        end
+      end
+    end
+  end
+end

data/lib/pangea/resources/aws_budgets_budget/resource.rb

@@ -0,0 +1,197 @@
+# frozen_string_literal: true
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+require 'pangea/resources/base'
+require 'pangea/resources/reference'
+require 'pangea/resources/aws_budgets_budget/types'
+require 'pangea/resource_registry'
+
+module Pangea
+  module Resources
+    module AWS
+      # Create an AWS Budgets Budget with comprehensive cost management and financial governance
+      #
+      # @param name [Symbol] The resource name
+      # @param attributes [Hash] Budget configuration attributes
+      # @return [ResourceReference] Reference object with outputs and financial insights
+      def aws_budgets_budget(name, attributes = {})
+        # Validate attributes using dry-struct with extensive cost management validation
+        budget_attrs = Types::BudgetAttributes.new(attributes)
+        
+        # Generate terraform resource block via terraform-synthesizer
+        resource(:aws_budgets_budget, name) do
+          budget_name budget_attrs.budget_name
+          budget_type budget_attrs.budget_type
+          time_unit budget_attrs.time_unit
+          
+          # Budget limits configuration
+          limit_amount budget_attrs.limit_amount
+          limit_unit budget_attrs.limit_unit
+          
+          # Time period configuration
+          if budget_attrs.time_period
+            time_period do
+              start budget_attrs.time_period&.dig(:start) if budget_attrs.time_period&.dig(:start)
+              send(:end, budget_attrs.time_period&.dig(:end)) if budget_attrs.time_period&.dig(:end)
+            end
+          end
+          
+          # Cost filters for granular budget tracking
+          if budget_attrs.cost_filters
+            cost_filters do
+              if budget_attrs.cost_filters&.dig(:dimensions)
+                budget_attrs.cost_filters&.dig(:dimensions).each do |dimension_key, values|
+                  dimension do
+                    key dimension_key
+                    values values
+                  end
+                end
+              end
+              
+              if budget_attrs.cost_filters&.dig(:tags)
+                budget_attrs.cost_filters&.dig(:tags).each do |tag_key, tag_values|
+                  tag do
+                    key tag_key
+                    values tag_values
+                  end
+                end
+              end
+              
+              if budget_attrs.cost_filters&.dig(:cost_categories)
+                budget_attrs.cost_filters&.dig(:cost_categories).each do |category_key, category_values|
+                  cost_category do
+                    key category_key
+                    values category_values
+                  end
+                end
+              end
+              
+              # NOT filters for exclusions
+              if budget_attrs.cost_filters&.dig(:not)
+                not_filter = budget_attrs.cost_filters&.dig(:not)
+                if not_filter[:dimensions] || not_filter[:tags] || not_filter[:cost_categories]
+                  # Terraform syntax for NOT filters
+                  # This would need to be structured appropriately for the provider
+                end
+              end
+            end
+          end
+          
+          # Planned budget limits for variable spending patterns
+          if budget_attrs.planned_budget_limits
+            budget_attrs.planned_budget_limits.each do |period_date, budget_spend|
+              planned_budget_limit do
+                start_date period_date
+                limit_amount budget_spend[:amount]
+                limit_unit budget_spend[:unit]
+              end
+            end
+          end
+          
+          # Auto-adjust configuration for dynamic budgets
+          if budget_attrs.auto_adjust_data
+            auto_adjust_data do
+              auto_adjust_type budget_attrs.auto_adjust_data&.dig(:auto_adjust_type)
+              
+              if budget_attrs.auto_adjust_data&.dig(:historical_options)
+                historical_options do
+                  budget_adjustment_period budget_attrs.auto_adjust_data&.dig(:historical_options)[:budget_adjustment_period]
+                  lookback_available_periods budget_attrs.auto_adjust_data&.dig(:historical_options)[:lookback_available_periods] if budget_attrs.auto_adjust_data&.dig(:historical_options)[:lookback_available_periods]
+                end
+              end
+            end
+          end
+          
+          # Budget notifications for proactive cost management
+          if budget_attrs.notifications
+            budget_attrs.notifications.each_with_index do |notification, index|
+              notification_block = "notification_#{index}".to_sym
+              
+              public_send(notification_block) do
+                notification_type notification[:notification_type]
+                comparison_operator notification[:comparison_operator]
+                threshold notification[:threshold]
+                threshold_type notification[:threshold_type] if notification[:threshold_type]
+                
+                # Subscribers for notifications
+                if notification[:subscribers]
+                  notification[:subscribers].each_with_index do |subscriber, sub_index|
+                    subscriber_block = "subscriber_#{sub_index}".to_sym
+                    
+                    public_send(subscriber_block) do
+                      subscription_type subscriber[:subscription_type]
+                      address subscriber[:address]
+                    end
+                  end
+                end
+              end
+            end
+          end
+          
+          # Apply tags for resource organization and cost allocation
+          if budget_attrs.tags&.any?
+            tags do
+              budget_attrs.tags.each do |key, value|
+                public_send(key, value)
+              end
+            end
+          end
+        end
+        
+        # Return resource reference with comprehensive cost management insights
+        ResourceReference.new(
+          type: 'aws_budgets_budget',
+          name: name,
+          resource_attributes: budget_attrs.to_h,
+          outputs: {
+            # Core budget identifiers
+            budget_name: "${aws_budgets_budget.#{name}.budget_name}",
+            budget_arn: "${aws_budgets_budget.#{name}.arn}",
+            
+            # Budget configuration
+            budget_type: "${aws_budgets_budget.#{name}.budget_type}",
+            time_unit: "${aws_budgets_budget.#{name}.time_unit}",
+            limit_amount: "${aws_budgets_budget.#{name}.limit_amount}",
+            limit_unit: "${aws_budgets_budget.#{name}.limit_unit}",
+            
+            # Computed financial insights
+            monthly_budget_estimate: budget_attrs.monthly_budget_estimate,
+            annual_budget_estimate: budget_attrs.annual_budget_estimate,
+            cost_optimization_score: budget_attrs.cost_optimization_score,
+            governance_compliance_level: budget_attrs.governance_compliance_level,
+            
+            # Budget capabilities
+            has_cost_tracking: budget_attrs.has_cost_tracking?,
+            has_usage_tracking: budget_attrs.has_usage_tracking?,
+            has_ri_tracking: budget_attrs.has_ri_tracking?,
+            has_savings_plans_tracking: budget_attrs.has_savings_plans_tracking?,
+            
+            # Notification insights
+            notification_count: budget_attrs.notification_count,
+            has_email_notifications: budget_attrs.has_email_notifications?,
+            has_sns_notifications: budget_attrs.has_sns_notifications?,
+            
+            # Advanced features status
+            has_cost_filters: !budget_attrs.cost_filters.nil?,
+            has_planned_limits: !budget_attrs.planned_budget_limits.nil?,
+            has_auto_adjustment: !budget_attrs.auto_adjust_data.nil?
+          }
+        )
+      end
+    end
+  end
+end
+

data/lib/pangea/resources/aws_budgets_budget/types/attributes.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'dry-struct'
+require_relative 'enums'
+require_relative 'schemas'
+require_relative 'helpers'
+
+module Pangea
+  module Resources
+    module AWS
+      module Types
+        # Budget resource attributes with comprehensive validation
+        class BudgetAttributes < Pangea::Resources::BaseAttributes
+          include BudgetHelpers
+
+          transform_keys(&:to_sym)
+
+          attribute? :budget_name, Resources::Types::String.constrained(format: /\A[a-zA-Z0-9_\-. ]{1,100}\z/).constructor { |value|
+            cleaned = value.strip
+            if cleaned.empty?
+              raise Dry::Struct::Error, "Budget name cannot be empty"
+            end
+            if cleaned != value
+              raise Dry::Struct::Error, "Budget name cannot have leading or trailing whitespace"
+            end
+            cleaned
+          }
+
+          attribute? :budget_type, BudgetType.optional
+          attribute? :time_unit, BudgetTimeUnit.optional
+
+          # Main budget limit
+          attribute? :limit_amount, Resources::Types::String.constrained(format: /\A\d+(\.\d{1,2})?\z/).constructor { |value|
+            amount_float = value.to_f
+            if amount_float <= 0
+              raise Dry::Struct::Error, "Budget limit amount must be positive"
+            end
+            value
+          }
+
+          attribute? :limit_unit, BudgetCurrency.optional
+
+          # Optional attributes
+          attribute :time_period?, BudgetTimePeriod.optional
+          attribute :cost_filters?, BudgetCostFilters.optional
+          attribute :planned_budget_limits?, BudgetPlannedBudgetLimits.optional
+          attribute :auto_adjust_data?, BudgetAutoAdjustData.optional
+          attribute :notifications?, Resources::Types::Array.of(BudgetNotification).constrained(max_size: 5).optional
+          attribute :tags?, Resources::Types::AwsTags.optional
+        end
+      end
+    end
+  end
+end

data/lib/pangea/resources/aws_budgets_budget/types/enums.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# Copyright 2025 The Pangea Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Pangea
+  module Resources
+    module AWS
+      module Types
+        # Budget time unit types
+        BudgetTimeUnit = Resources::Types::String.constrained(included_in: ['DAILY', 'MONTHLY', 'QUARTERLY', 'ANNUALLY'])
+
+        # Budget type enumeration
+        BudgetType = Resources::Types::String.constrained(included_in: ['USAGE', 'COST', 'RI_UTILIZATION', 'RI_COVERAGE',
+          'SAVINGS_PLANS_UTILIZATION', 'SAVINGS_PLANS_COVERAGE'])
+
+        # Cost currency types
+        BudgetCurrency = Resources::Types::String.constrained(included_in: ['USD', 'EUR', 'GBP', 'JPY', 'CNY', 'CAD', 'AUD', 'BRL', 'INR'])
+
+        # Budget comparison operator
+        BudgetComparisonOperator = Resources::Types::String.constrained(included_in: ['GREATER_THAN', 'LESS_THAN', 'EQUAL_TO'])
+
+        # Threshold type for budget notifications
+        BudgetThresholdType = Resources::Types::String.constrained(included_in: ['PERCENTAGE', 'ABSOLUTE_VALUE'])
+
+        # Budget notification type
+        BudgetNotificationType = Resources::Types::String.constrained(included_in: ['ACTUAL', 'FORECASTED'])
+
+        # Budget subscriber protocol
+        BudgetSubscriberProtocol = Resources::Types::String.constrained(included_in: ['EMAIL', 'SNS'])
+
+        # Cost dimension key types for budget filters
+        CostDimensionKey = Resources::Types::String.constrained(included_in: ['AZ', 'INSTANCE_TYPE', 'LINKED_ACCOUNT', 'OPERATION', 'PURCHASE_TYPE',
+          'REGION', 'SERVICE', 'USAGE_TYPE', 'USAGE_TYPE_GROUP', 'RECORD_TYPE',
+          'OPERATING_SYSTEM', 'TENANCY', 'SCOPE', 'PLATFORM', 'SUBSCRIPTION_ID',
+          'LEGAL_ENTITY_NAME', 'DEPLOYMENT_OPTION', 'DATABASE_ENGINE',
+          'CACHE_ENGINE', 'INSTANCE_TYPE_FAMILY', 'BILLING_ENTITY', 'RESERVATION_ID',
+          'RESOURCE_ID', 'RIGHTSIZING_TYPE', 'SAVINGS_PLANS_TYPE', 'SAVINGS_PLAN_ARN',
+          'PAYMENT_OPTION', 'AGREEMENT_END_DATE_TIME_AFTER', 'AGREEMENT_END_DATE_TIME_BEFORE'])
+      end
+    end
+  end
+end