From c4807b26f0d530ef99ff87b6c5c45a4953ba958a Mon Sep 17 00:00:00 2001
From: icebaker <icebaker@proton.me>
Date: Sat, 6 Jan 2024 22:09:23 -0300
Subject: adding new providers

---
 logic/cartridge/streaming.rb       |  4 ++--
 logic/providers/maritaca/tokens.rb | 14 ++++++++++++++
 logic/providers/ollama/tokens.rb   | 14 ++++++++++++++
 3 files changed, 30 insertions(+), 2 deletions(-)
 create mode 100644 logic/providers/maritaca/tokens.rb
 create mode 100644 logic/providers/ollama/tokens.rb

(limited to 'logic')

diff --git a/logic/cartridge/streaming.rb b/logic/cartridge/streaming.rb
index 23e88ac..b04dc6e 100644
--- a/logic/cartridge/streaming.rb
+++ b/logic/cartridge/streaming.rb
@@ -8,9 +8,9 @@ module NanoBot
       module Streaming
         def self.enabled?(cartridge, interface)
           provider_stream = case Helpers::Hash.fetch(cartridge, %i[provider id])
-                            when 'openai', 'mistral', 'cohere'
+                            when 'openai', 'mistral', 'cohere', 'ollama'
                               Helpers::Hash.fetch(cartridge, %i[provider settings stream])
-                            when 'google'
+                            when 'google', 'maritaca'
                               Helpers::Hash.fetch(cartridge, %i[provider options stream])
                             end
 
diff --git a/logic/providers/maritaca/tokens.rb b/logic/providers/maritaca/tokens.rb
new file mode 100644
index 0000000..1ae2219
--- /dev/null
+++ b/logic/providers/maritaca/tokens.rb
@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module NanoBot
+  module Logic
+    module Maritaca
+      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/ollama/tokens.rb b/logic/providers/ollama/tokens.rb
new file mode 100644
index 0000000..45500fb
--- /dev/null
+++ b/logic/providers/ollama/tokens.rb
@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module NanoBot
+  module Logic
+    module Ollama
+      module Tokens
+        def self.apply_policies!(_cartridge, payload)
+          payload[:messages] = payload[:messages].map { |message| message.except(:_meta) }
+          payload
+        end
+      end
+    end
+  end
+end
-- 
cgit v1.2.3