Added LiteLLM to the stack

Development/litellm/tests/test_litellm/llms/cohere/chat/test_transformation.py

@@ -0,0 +1,51 @@
+import os
+import sys
+from unittest.mock import MagicMock
+
+sys.path.insert(
+    0, os.path.abspath("../../../../..")
+)  # Adds the parent directory to the system path
+
+from litellm.llms.cohere.chat.transformation import CohereChatConfig
+
+
+class TestCohereTransform:
+    def setup_method(self):
+        self.config = CohereChatConfig()
+        self.model = "command-r-plus-latest"
+        self.logging_obj = MagicMock()
+
+    def test_map_cohere_params(self):
+        """Test that parameters are correctly mapped"""
+        test_params = {
+            "temperature": 0.7,
+            "max_tokens": 200,
+            "max_completion_tokens": 256,
+        }
+
+        result = self.config.map_openai_params(
+            non_default_params=test_params,
+            optional_params={},
+            model=self.model,
+            drop_params=False,
+        )
+
+        # The function should properly map max_completion_tokens to max_tokens and override max_tokens
+        assert result == {"temperature": 0.7, "max_tokens": 256}
+
+    def test_cohere_max_tokens_backward_compat(self):
+        """Test that parameters are correctly mapped"""
+        test_params = {
+            "temperature": 0.7,
+            "max_tokens": 200,
+        }
+
+        result = self.config.map_openai_params(
+            non_default_params=test_params,
+            optional_params={},
+            model=self.model,
+            drop_params=False,
+        )
+
+        # The function should properly map max_tokens if max_completion_tokens is not provided
+        assert result == {"temperature": 0.7, "max_tokens": 200}