Added LiteLLM to the stack

Development/litellm/tests/test_litellm/llms/test_volcengine.py

@@ -0,0 +1,135 @@
+import os
+import sys
+from unittest.mock import MagicMock, patch
+
+from pydantic import BaseModel
+
+from litellm.llms.volcengine import VolcEngineConfig
+from litellm.utils import get_optional_params
+
+
+class TestVolcEngineConfig:
+    def test_get_optional_params(self):
+        config = VolcEngineConfig()
+        supported_params = config.get_supported_openai_params(model="doubao-seed-1.6")
+        assert "thinking" in supported_params
+
+        # Test thinking disabled - should NOT appear in extra_body
+        mapped_params = config.map_openai_params(
+            non_default_params={
+                "thinking": {"type": "disabled"},
+            },
+            optional_params={},
+            model="doubao-seed-1.6",
+            drop_params=False,
+        )
+
+        # Fixed: thinking disabled should be omitted from extra_body
+        assert mapped_params == {}
+
+        e2e_mapped_params = get_optional_params(
+            model="doubao-seed-1.6",
+            custom_llm_provider="volcengine",
+            thinking={"type": "enabled"},
+            drop_params=False,
+        )
+
+        assert "thinking" in e2e_mapped_params["extra_body"] and e2e_mapped_params[
+            "extra_body"
+        ]["thinking"] == {
+            "type": "enabled",
+        }
+
+    def test_thinking_parameter_handling(self):
+        """Test comprehensive thinking parameter handling scenarios"""
+        config = VolcEngineConfig()
+        
+        # Test 1: thinking enabled - should appear in extra_body
+        result_enabled = config.map_openai_params(
+            non_default_params={"thinking": {"type": "enabled"}},
+            optional_params={},
+            model="doubao-seed-1.6",
+            drop_params=False,
+        )
+        assert result_enabled == {
+            "extra_body": {"thinking": {"type": "enabled"}}
+        }
+        
+        # Test 2: thinking None - should appear in extra_body as None
+        result_none = config.map_openai_params(
+            non_default_params={"thinking": None},
+            optional_params={},
+            model="doubao-seed-1.6", 
+            drop_params=False,
+        )
+        assert result_none == {
+            "extra_body": {"thinking": None}
+        }
+        
+        # Test 3: thinking with custom value - should appear in extra_body
+        result_custom = config.map_openai_params(
+            non_default_params={"thinking": "custom_mode"},
+            optional_params={},
+            model="doubao-seed-1.6",
+            drop_params=False,
+        )
+        assert result_custom == {
+            "extra_body": {"thinking": "custom_mode"}
+        }
+        
+        # Test 4: thinking disabled - should NOT appear in extra_body
+        result_disabled = config.map_openai_params(
+            non_default_params={"thinking": {"type": "disabled"}},
+            optional_params={},
+            model="doubao-seed-1.6",
+            drop_params=False,
+        )
+        assert result_disabled == {}
+        
+        # Test 5: No thinking parameter - should return empty dict
+        result_no_thinking = config.map_openai_params(
+            non_default_params={},
+            optional_params={},
+            model="doubao-seed-1.6",
+            drop_params=False,
+        )
+        assert result_no_thinking == {}
+
+    def test_e2e_completion(self):
+        from openai import OpenAI
+
+        from litellm import completion
+        from litellm.types.utils import ModelResponse
+
+        client = OpenAI(api_key="test_api_key")
+
+        mock_raw_response = MagicMock()
+        mock_raw_response.headers = {
+            "x-request-id": "123",
+            "openai-organization": "org-123",
+            "x-ratelimit-limit-requests": "100",
+            "x-ratelimit-remaining-requests": "99",
+        }
+        mock_raw_response.parse.return_value = ModelResponse()
+
+        with patch.object(
+            client.chat.completions.with_raw_response, "create", mock_raw_response
+        ) as mock_create:
+            completion(
+                model="volcengine/doubao-seed-1.6",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": "**Tell me your model detail information.**",
+                    }
+                ],
+                user="guest",
+                stream=True,
+                thinking={"type": "disabled"},
+                client=client,
+            )
+
+            mock_create.assert_called_once()
+            print(mock_create.call_args.kwargs)
+            # Fixed: thinking disabled should NOT appear in extra_body
+            assert "extra_body" not in mock_create.call_args.kwargs or "thinking" not in mock_create.call_args.kwargs.get("extra_body", {})