summaryrefslogtreecommitdiff
path: root/components/provider.rb
diff options
context:
space:
mode:
authoricebaker <icebaker@proton.me>2024-01-06 22:09:23 -0300
committericebaker <icebaker@proton.me>2024-01-06 22:09:23 -0300
commitc4807b26f0d530ef99ff87b6c5c45a4953ba958a (patch)
tree8d2709d35089ec8afb60bd20c6855de4623e3d1b /components/provider.rb
parentbfe0e76e3683a71bb8ce5bfdaae99b0252e7be05 (diff)
adding new providers
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