diff options
Diffstat (limited to 'components/provider.rb')
-rw-r--r-- | components/provider.rb | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/components/provider.rb b/components/provider.rb index ac3964d..4c409d2 100644 --- a/components/provider.rb +++ b/components/provider.rb @@ -1,9 +1,11 @@ # frozen_string_literal: true -require_relative 'providers/google' -require_relative 'providers/mistral' require_relative 'providers/openai' +require_relative 'providers/ollama' +require_relative 'providers/mistral' +require_relative 'providers/google' require_relative 'providers/cohere' +require_relative 'providers/maritaca' module NanoBot module Components @@ -12,12 +14,16 @@ module NanoBot case provider[:id] when 'openai' Providers::OpenAI.new(nil, provider[:settings], provider[:credentials], environment:) - when 'google' - Providers::Google.new(provider[:options], provider[:settings], provider[:credentials], environment:) + when 'ollama' + Providers::Ollama.new(provider[:options], provider[:settings], provider[:credentials], environment:) when 'mistral' Providers::Mistral.new(provider[:options], provider[:settings], provider[:credentials], environment:) + when 'google' + Providers::Google.new(provider[:options], provider[:settings], provider[:credentials], environment:) when 'cohere' Providers::Cohere.new(provider[:options], provider[:settings], provider[:credentials], environment:) + when 'maritaca' + Providers::Maritaca.new(provider[:options], provider[:settings], provider[:credentials], environment:) else raise "Unsupported provider \"#{provider[:id]}\"" end |