dify_llm 1.9.1 → 1.14.1

data/lib/ruby_llm/providers/bedrock/capabilities.rb

@@ -1,167 +0,0 @@
-# frozen_string_literal: true
-
-module RubyLLM
-  module Providers
-    class Bedrock
-      # Determines capabilities and pricing for AWS Bedrock models
-      module Capabilities
-        module_function
-
-        def context_window_for(model_id)
-          case model_id
-          when /anthropic\.claude-2/ then 100_000
-          else 200_000
-          end
-        end
-
-        def max_tokens_for(_model_id)
-          4_096
-        end
-
-        def input_price_for(model_id)
-          PRICES.dig(model_family(model_id), :input) || default_input_price
-        end
-
-        def output_price_for(model_id)
-          PRICES.dig(model_family(model_id), :output) || default_output_price
-        end
-
-        def supports_chat?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def supports_streaming?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def supports_images?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def supports_vision?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def supports_functions?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def supports_audio?(_model_id)
-          false
-        end
-
-        def supports_json_mode?(model_id)
-          model_id.match?(/anthropic\.claude/)
-        end
-
-        def format_display_name(model_id)
-          model_id.then { |id| humanize(id) }
-        end
-
-        def model_type(_model_id)
-          'chat'
-        end
-
-        def supports_structured_output?(_model_id)
-          false
-        end
-
-        # Model family patterns for capability lookup
-        MODEL_FAMILIES = {
-          /anthropic\.claude-3-opus/ => :claude3_opus,
-          /anthropic\.claude-3-sonnet/ => :claude3_sonnet,
-          /anthropic\.claude-3-5-sonnet/ => :claude3_sonnet,
-          /anthropic\.claude-3-7-sonnet/ => :claude3_sonnet,
-          /anthropic\.claude-3-haiku/ => :claude3_haiku,
-          /anthropic\.claude-3-5-haiku/ => :claude3_5_haiku,
-          /anthropic\.claude-v2/ => :claude2,
-          /anthropic\.claude-instant/ => :claude_instant
-        }.freeze
-
-        def model_family(model_id)
-          MODEL_FAMILIES.find { |pattern, _family| model_id.match?(pattern) }&.last || :other
-        end
-
-        # Pricing information for Bedrock models (per million tokens)
-        PRICES = {
-          claude3_opus: { input: 15.0, output: 75.0 },
-          claude3_sonnet: { input: 3.0, output: 15.0 },
-          claude3_haiku: { input: 0.25, output: 1.25 },
-          claude3_5_haiku: { input: 0.8, output: 4.0 },
-          claude2: { input: 8.0, output: 24.0 },
-          claude_instant: { input: 0.8, output: 2.4 }
-        }.freeze
-
-        def default_input_price
-          0.1
-        end
-
-        def default_output_price
-          0.2
-        end
-
-        def humanize(id)
-          id.tr('-', ' ')
-            .split('.')
-            .last
-            .split
-            .map(&:capitalize)
-            .join(' ')
-        end
-
-        def modalities_for(model_id)
-          modalities = {
-            input: ['text'],
-            output: ['text']
-          }
-
-          if model_id.match?(/anthropic\.claude/) && supports_vision?(model_id)
-            modalities[:input] << 'image'
-            modalities[:input] << 'pdf'
-          end
-
-          modalities
-        end
-
-        def capabilities_for(model_id)
-          capabilities = []
-
-          capabilities << 'streaming' if model_id.match?(/anthropic\.claude/)
-
-          capabilities << 'function_calling' if supports_functions?(model_id)
-
-          capabilities << 'reasoning' if model_id.match?(/claude-3-7/)
-
-          if model_id.match?(/claude-3\.5|claude-3-7/)
-            capabilities << 'batch'
-            capabilities << 'citations'
-          end
-
-          capabilities
-        end
-
-        def pricing_for(model_id)
-          family = model_family(model_id)
-          prices = PRICES.fetch(family, { input: default_input_price, output: default_output_price })
-
-          standard_pricing = {
-            input_per_million: prices[:input],
-            output_per_million: prices[:output]
-          }
-
-          batch_pricing = {
-            input_per_million: prices[:input] * 0.5,
-            output_per_million: prices[:output] * 0.5
-          }
-
-          {
-            text_tokens: {
-              standard: standard_pricing,
-              batch: batch_pricing
-            }
-          }
-        end
-      end
-    end
-  end
-end