module RubyLLM::Providers::Bedrock::Models

# File lib/ruby_llm/providers/bedrock/models.rb, line 26
def create_model_info(model_data, slug, _capabilities = nil)
  model_id = model_id_with_region(model_data['modelId'], model_data)
  converse_data = model_data['converse'] || {}

  Model::Info.new(
    id: model_id,
    name: model_data['modelName'],
    provider: slug,
    family: model_data['modelFamily'] || model_data['providerName']&.downcase,
    created_at: nil,
    context_window: parse_context_window(model_data),
    max_output_tokens: converse_data['maxTokensDefault'] || converse_data['maxTokensMaximum'],
    modalities: {
      input: normalize_modalities(model_data['inputModalities']),
      output: normalize_modalities(model_data['outputModalities'])
    },
    capabilities: parse_capabilities(model_data),
    pricing: {},
    metadata: {
      provider_name: model_data['providerName'],
      model_arn: model_data['modelArn'],
      inference_types: model_data['inferenceTypesSupported'],
      converse: converse_data
    }
  )
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 53
def model_id_with_region(model_id, model_data)
  inference_types = Array(model_data['inferenceTypesSupported'])
  normalize_inference_profile_id(model_id, inference_types, @config.bedrock_region)
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 12
def models_api_base
  @config.bedrock_api_base || "https://bedrock.#{bedrock_region}.amazonaws.com"
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 16
def models_url
  '/foundation-models'
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 58
def normalize_inference_profile_id(model_id, inference_types, region)
  return model_id unless inference_types.include?('INFERENCE_PROFILE')
  return model_id if inference_types.include?('ON_DEMAND')

  with_region_prefix(model_id, region)
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 85
def normalize_modalities(modalities)
  Array(modalities).map do |modality|
    normalized = modality.to_s.downcase
    case normalized
    when 'embedding' then 'embeddings'
    when 'speech' then 'audio'
    else normalized
    end
  end
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 96
def parse_capabilities(model_data)
  capabilities = []
  capabilities << 'streaming' if model_data['responseStreamingSupported']

  converse = model_data['converse'] || {}
  capabilities << 'function_calling' if converse.is_a?(Hash)
  capabilities << 'reasoning' if converse.dig('reasoningSupported', 'embedded')
  capabilities << 'structured_output' if supports_structured_output?(model_data['modelId'])

  capabilities
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 130
def parse_context_window(model_data)
  value = model_data.dig('description', 'maxContextWindow')
  return unless value.is_a?(String)

  if value.match?(/\A\d+[kK]\z/)
    value.to_i * 1000
  elsif value.match?(/\A\d+\z/)
    value.to_i
  end
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 20
def parse_list_models_response(response, slug, _capabilities)
  Array(response.body['modelSummaries']).map do |model_data|
    create_model_info(model_data, slug)
  end
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 123
def reasoning_embedded?(model)
  metadata = RubyLLM::Utils.deep_symbolize_keys(model.metadata || {})
  converse = metadata[:converse] || {}
  reasoning_supported = converse[:reasoningSupported] || {}
  reasoning_supported[:embedded] || false
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 75
def region_prefix(region)
  prefix = region.to_s.split('-').first
  prefix = '' if prefix.nil?
  prefix.empty? ? 'us' : prefix
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 81
def region_prefixed?(model_id)
  model_id.match?(/\A(?:#{REGION_PREFIXES.join('|')})\./)
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 111
def supports_structured_output?(model_id)
  return false unless model_id

  normalized = model_id.sub(/\A(?:#{REGION_PREFIXES.join('|')})\./, '').delete_prefix('anthropic.')
  match = normalized.match(/claude-(?:opus|sonnet|haiku)-(\d+)-(\d{1,2})(?:\b|-)/)
  return false unless match

  major = match[1].to_i
  minor = match[2].to_i
  major > 4 || (major == 4 && minor >= 5)
end

# File lib/ruby_llm/providers/bedrock/models.rb, line 65
def with_region_prefix(model_id, region)
  prefix = region_prefix(region)

  if region_prefixed?(model_id)
    model_id.sub(/\A(?:#{REGION_PREFIXES.join('|')})\./, "#{prefix}.")
  else
    "#{prefix}.#{model_id}"
  end
end