From c4807b26f0d530ef99ff87b6c5c45a4953ba958a Mon Sep 17 00:00:00 2001 From: icebaker Date: Sat, 6 Jan 2024 22:09:23 -0300 Subject: adding new providers --- components/provider.rb | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'components/provider.rb') 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 -- cgit v1.2.3