module RubyLLM::Providers::DeepSeek::Capabilities

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 31
def context_window_for(_model_id)
  DEFAULT_CONTEXT_WINDOW
end

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 39
def critical_capabilities_for(model_id)
  v4_model = model_id.start_with?('deepseek-v4-')
  capabilities = ['function_calling']
  capabilities << 'structured_output' if v4_model
  capabilities << 'reasoning' if model_id == 'deepseek-reasoner' || v4_model
  capabilities
end

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 35
def max_tokens_for(_model_id)
  DEFAULT_MAX_OUTPUT_TOKENS
end

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 47
def pricing_for(model_id)
  prices = model_id == 'deepseek-v4-pro' ? PRO_PRICES : DEFAULT_PRICES

  {
    text_tokens: {
      standard: {
        input_per_million: prices[:input],
        output_per_million: prices[:output],
        cache_read_input_per_million: prices[:cache_read]
      }
    }
  }
end

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 23
def supports_tool_choice?(_model_id)
  true
end

# File lib/ruby_llm/providers/deepseek/capabilities.rb, line 27
def supports_tool_parallel_control?(_model_id)
  false
end