summaryrefslogtreecommitdiff
path: root/components/provider.rb
diff options
context:
space:
mode:
authoricebaker <icebaker@proton.me>2023-12-14 22:49:06 -0300
committericebaker <icebaker@proton.me>2023-12-14 22:49:06 -0300
commite6f0374cc8844d4a053db4e68feee23ffc793d73 (patch)
treea06163503c7a4f754be1a91bbaaee9adf9e7c7b2 /components/provider.rb
parentfb96658a1ca4b6e3e0505e7a39f660e1a05b3c6e (diff)
adding support for google gemini
Diffstat (limited to 'components/provider.rb')
-rw-r--r--components/provider.rb5
1 files changed, 3 insertions, 2 deletions
diff --git a/components/provider.rb b/components/provider.rb
index 3414009..2ad35f4 100644
--- a/components/provider.rb
+++ b/components/provider.rb
@@ -1,8 +1,7 @@
# frozen_string_literal: true
-require 'openai'
-
require_relative 'providers/openai'
+require_relative 'providers/google'
module NanoBot
module Components
@@ -11,6 +10,8 @@ module NanoBot
case provider[:id]
when 'openai'
Providers::OpenAI.new(provider[:settings], provider[:credentials], environment:)
+ when 'google'
+ Providers::Google.new(provider[:model], provider[:settings], provider[:credentials], environment:)
else
raise "Unsupported provider \"#{provider[:id]}\""
end