mirror of
https://github.com/we-promise/sure.git
synced 2026-05-30 07:49:01 +00:00
Introduces Provider::Anthropic alongside Provider::Openai, implementing the LlmConcept chat_response contract over the official anthropic Ruby SDK. Batch ops, PDF, and RAG land in follow-up PRs. - Provider::Anthropic uses Messages API for sync and streaming responses - ChatConfig builds requests with ephemeral prompt-cache markers on the system prompt and the last tool definition - MessageFormatter reconstructs multi-turn history (text + tool_use + tool_result blocks) from raw Message records, including the paired user-role tool_result turn Anthropic requires after every tool_use - ChatParser maps Anthropic Message into the shared ChatResponse Data - Registry, Setting, User, Chat default model wired for ANTHROPIC_* envs and Setting.anthropic_*; LLM_PROVIDER selects between providers - Responder forwards raw conversation_history (Array<Message>) so providers without hosted conversation state can rebuild context - OpenAI provider accepts and ignores the new kwarg (no behavior change) Tests cover provider init, model gating, MessageFormatter for all turn shapes, ChatConfig request building (max_tokens, system cache, tool conversion), ChatParser for text / tool_use / mixed blocks, Registry discovery, and mocked chat_response success / error / function_request paths. Live VCR cassettes recorded in a follow-up with a real key. Stacked PRs: 2/5 batch ops + cost ledger, 3/5 PDF, 4/5 pgvector RAG, 5/5 settings UI + disclosure.
169 lines
4.3 KiB
Ruby
169 lines
4.3 KiB
Ruby
class Provider::Registry
|
|
include ActiveModel::Validations
|
|
|
|
Error = Class.new(StandardError)
|
|
|
|
CONCEPTS = %i[exchange_rates securities llm]
|
|
|
|
validates :concept, inclusion: { in: CONCEPTS }
|
|
|
|
class << self
|
|
def for_concept(concept)
|
|
new(concept.to_sym)
|
|
end
|
|
|
|
def get_provider(name)
|
|
send(name)
|
|
rescue NoMethodError
|
|
raise Error.new("Provider '#{name}' not found in registry")
|
|
end
|
|
|
|
def plaid_provider_for_region(region)
|
|
region.to_sym == :us ? plaid_us : plaid_eu
|
|
end
|
|
|
|
private
|
|
def stripe
|
|
secret_key = ENV["STRIPE_SECRET_KEY"]
|
|
webhook_secret = ENV["STRIPE_WEBHOOK_SECRET"]
|
|
|
|
return nil unless secret_key.present? && webhook_secret.present?
|
|
|
|
Provider::Stripe.new(secret_key:, webhook_secret:)
|
|
end
|
|
|
|
def twelve_data
|
|
api_key = ENV["TWELVE_DATA_API_KEY"].presence || Setting.twelve_data_api_key
|
|
|
|
return nil unless api_key.present?
|
|
|
|
Provider::TwelveData.new(api_key)
|
|
end
|
|
|
|
def plaid_us
|
|
Provider::PlaidAdapter.ensure_configuration_loaded
|
|
config = Rails.application.config.plaid
|
|
|
|
return nil unless config.present?
|
|
|
|
Provider::Plaid.new(config, region: :us)
|
|
end
|
|
|
|
def plaid_eu
|
|
Provider::PlaidEuAdapter.ensure_configuration_loaded
|
|
config = Rails.application.config.plaid_eu
|
|
|
|
return nil unless config.present?
|
|
|
|
Provider::Plaid.new(config, region: :eu)
|
|
end
|
|
|
|
def github
|
|
Provider::Github.new
|
|
end
|
|
|
|
def openai
|
|
access_token = ENV["OPENAI_ACCESS_TOKEN"].presence || Setting.openai_access_token
|
|
|
|
return nil unless access_token.present?
|
|
|
|
uri_base = ENV["OPENAI_URI_BASE"].presence || Setting.openai_uri_base
|
|
model = ENV["OPENAI_MODEL"].presence || Setting.openai_model
|
|
|
|
if uri_base.present? && model.blank?
|
|
Rails.logger.error("Custom OpenAI provider configured without a model; please set OPENAI_MODEL or Setting.openai_model")
|
|
return nil
|
|
end
|
|
|
|
Provider::Openai.new(access_token, uri_base: uri_base, model: model)
|
|
end
|
|
|
|
def anthropic
|
|
access_token = ENV["ANTHROPIC_ACCESS_TOKEN"].presence ||
|
|
ENV["ANTHROPIC_API_KEY"].presence ||
|
|
Setting.anthropic_access_token
|
|
|
|
return nil unless access_token.present?
|
|
|
|
base_url = ENV["ANTHROPIC_BASE_URL"].presence || Setting.anthropic_base_url
|
|
model = ENV["ANTHROPIC_MODEL"].presence || Setting.anthropic_model
|
|
|
|
Provider::Anthropic.new(access_token, base_url: base_url, model: model)
|
|
end
|
|
|
|
def yahoo_finance
|
|
Provider::YahooFinance.new
|
|
end
|
|
|
|
def tiingo
|
|
api_key = ENV["TIINGO_API_KEY"].presence || Setting.tiingo_api_key # pipelock:ignore
|
|
|
|
return nil unless api_key.present?
|
|
|
|
Provider::Tiingo.new(api_key)
|
|
end
|
|
|
|
def eodhd
|
|
api_key = ENV["EODHD_API_KEY"].presence || Setting.eodhd_api_key # pipelock:ignore
|
|
|
|
return nil unless api_key.present?
|
|
|
|
Provider::Eodhd.new(api_key)
|
|
end
|
|
|
|
def alpha_vantage
|
|
api_key = ENV["ALPHA_VANTAGE_API_KEY"].presence || Setting.alpha_vantage_api_key # pipelock:ignore
|
|
|
|
return nil unless api_key.present?
|
|
|
|
Provider::AlphaVantage.new(api_key)
|
|
end
|
|
|
|
def mfapi
|
|
Provider::Mfapi.new
|
|
end
|
|
|
|
def binance_public
|
|
Provider::BinancePublic.new
|
|
end
|
|
end
|
|
|
|
def initialize(concept)
|
|
@concept = concept
|
|
validate!
|
|
end
|
|
|
|
def providers
|
|
available_providers.map { |p| self.class.send(p) }.compact
|
|
end
|
|
|
|
# Returns the list of provider key names (symbols) registered for this concept.
|
|
def provider_keys
|
|
available_providers
|
|
end
|
|
|
|
def get_provider(name)
|
|
provider_method = available_providers.find { |p| p == name.to_sym }
|
|
|
|
raise Error.new("Provider '#{name}' not found for concept: #{concept}") unless provider_method.present?
|
|
|
|
self.class.send(provider_method)
|
|
end
|
|
|
|
private
|
|
attr_reader :concept
|
|
|
|
def available_providers
|
|
case concept
|
|
when :exchange_rates
|
|
%i[twelve_data yahoo_finance]
|
|
when :securities
|
|
%i[twelve_data yahoo_finance tiingo eodhd alpha_vantage mfapi binance_public]
|
|
when :llm
|
|
%i[openai anthropic]
|
|
else
|
|
%i[plaid_us plaid_eu github openai anthropic]
|
|
end
|
|
end
|
|
end
|