Merge pull request #13 from icebaker/ib-cohere

Adding support to Cohere

1 files changed, 17 insertions, 0 deletions

diff --git a/logic/providers/cohere/tokens.rb b/logic/providers/cohere/tokens.rb
new file mode 100644
index 0000000..f7d3f55
--- /dev/null
+++ b/logic/providers/cohere/tokens.rb
@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module NanoBot
+  module Logic
+    module Cohere
+      module Tokens
+        def self.apply_policies!(_cartridge, payload)
+          if payload[:chat_history]
+            payload[:chat_history] = payload[:chat_history].map { |message| message.except(:_meta) }
+          end
+
+          payload
+        end
+      end
+    end
+  end
+end