summaryrefslogtreecommitdiff
path: root/components/provider.rb
diff options
context:
space:
mode:
Diffstat (limited to 'components/provider.rb')
-rw-r--r--components/provider.rb14
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