diff options
author | icebaker <icebaker@proton.me> | 2023-12-28 19:43:00 -0300 |
---|---|---|
committer | icebaker <icebaker@proton.me> | 2023-12-28 19:43:00 -0300 |
commit | 46638e0b4e1809d683f470922f9cc27ab161c248 (patch) | |
tree | a6135a74b8c56abe435fb557c873273e32f86eb1 /logic | |
parent | 347ac029c922fbb3843ec22324527f2c6b3ba9a9 (diff) |
upgrading gemini-ai and adding support for mistral-ai
Diffstat (limited to 'logic')
-rw-r--r-- | logic/cartridge/streaming.rb | 2 | ||||
-rw-r--r-- | logic/providers/google/tokens.rb | 2 | ||||
-rw-r--r-- | logic/providers/mistral/tokens.rb | 14 | ||||
-rw-r--r-- | logic/providers/openai/tokens.rb | 2 |
4 files changed, 15 insertions, 5 deletions
diff --git a/logic/cartridge/streaming.rb b/logic/cartridge/streaming.rb index 6949b3a..0b9b19f 100644 --- a/logic/cartridge/streaming.rb +++ b/logic/cartridge/streaming.rb @@ -8,7 +8,7 @@ module NanoBot module Streaming def self.enabled?(cartridge, interface) provider_stream = case Helpers::Hash.fetch(cartridge, %i[provider id]) - when 'openai' + when 'openai', 'mistral' Helpers::Hash.fetch(cartridge, %i[provider settings stream]) when 'google' Helpers::Hash.fetch(cartridge, %i[provider options stream]) diff --git a/logic/providers/google/tokens.rb b/logic/providers/google/tokens.rb index 3d5492f..0d74928 100644 --- a/logic/providers/google/tokens.rb +++ b/logic/providers/google/tokens.rb @@ -1,7 +1,5 @@ # frozen_string_literal: true -require 'openai' - module NanoBot module Logic module Google diff --git a/logic/providers/mistral/tokens.rb b/logic/providers/mistral/tokens.rb new file mode 100644 index 0000000..7aa64b7 --- /dev/null +++ b/logic/providers/mistral/tokens.rb @@ -0,0 +1,14 @@ +# frozen_string_literal: true + +module NanoBot + module Logic + module Mistral + module Tokens + def self.apply_policies!(_cartridge, payload) + payload[:messages] = payload[:messages].map { |message| message.except(:_meta) } + payload + end + end + end + end +end diff --git a/logic/providers/openai/tokens.rb b/logic/providers/openai/tokens.rb index 60efa60..828b774 100644 --- a/logic/providers/openai/tokens.rb +++ b/logic/providers/openai/tokens.rb @@ -1,7 +1,5 @@ # frozen_string_literal: true -require 'openai' - module NanoBot module Logic module OpenAI |