summaryrefslogtreecommitdiff
path: root/components/providers
diff options
context:
space:
mode:
authoricebaker <icebaker@proton.me>2023-12-16 11:45:13 -0300
committericebaker <icebaker@proton.me>2023-12-16 11:45:13 -0300
commit57f44096dd22c32b2575268a00fd467469400d44 (patch)
treebb7ee0e35be9202144240b3edcbcbc024bbc0f8e /components/providers
parent510403034eb15dd676f2c03ee65b874cc74a20ed (diff)
making google credentials and options flexible
Diffstat (limited to 'components/providers')
-rw-r--r--components/providers/google.rb20
1 files changed, 2 insertions, 18 deletions
diff --git a/components/providers/google.rb b/components/providers/google.rb
index 92a9cb2..a522d06 100644
--- a/components/providers/google.rb
+++ b/components/providers/google.rb
@@ -26,25 +26,9 @@ module NanoBot
def initialize(options, settings, credentials, _environment)
@settings = settings
- gemini_credentials = if credentials[:'api-key']
- {
- service: credentials[:service],
- api_key: credentials[:'api-key'],
- project_id: credentials[:'project-id'],
- region: credentials[:region]
- }
- else
- {
- service: credentials[:service],
- file_path: credentials[:'file-path'],
- project_id: credentials[:'project-id'],
- region: credentials[:region]
- }
- end
-
@client = Gemini.new(
- credentials: gemini_credentials,
- options: { model: options[:model], stream: options[:stream] }
+ credentials: credentials.transform_keys { |key| key.to_s.gsub('-', '_').to_sym },
+ options: options.transform_keys { |key| key.to_s.gsub('-', '_').to_sym }
)
end