From c6c14492d7d8082815fdeb0bbeeec5f37fc496ce Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Thu, 8 Jan 2026 13:55:37 +0530
Subject: [PATCH] test_extract_agent_profile

---
 .../test_manus_responses_transformation.py    | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 tests/test_litellm/llms/manus/responses/test_manus_responses_transformation.py

diff --git a/tests/test_litellm/llms/manus/responses/test_manus_responses_transformation.py b/tests/test_litellm/llms/manus/responses/test_manus_responses_transformation.py
new file mode 100644
index 00000000000..b47ed77156d
--- /dev/null
+++ b/tests/test_litellm/llms/manus/responses/test_manus_responses_transformation.py
@@ -0,0 +1,60 @@
+"""
+Tests for Manus Responses API transformation
+
+Tests the ManusResponsesAPIConfig class that handles Manus-specific
+transformations for the Responses API.
+
+Source: litellm/llms/manus/responses/transformation.py
+"""
+import os
+import sys
+
+sys.path.insert(0, os.path.abspath("../../../../.."))
+
+from litellm.llms.manus.responses.transformation import ManusResponsesAPIConfig
+from litellm.types.llms.openai import ResponsesAPIOptionalRequestParams
+from litellm.types.router import GenericLiteLLMParams
+
+
+def test_extract_agent_profile():
+    """Test that agent profile is correctly extracted from model name"""
+    config = ManusResponsesAPIConfig()
+    
+    assert config._extract_agent_profile("manus/manus-1.6") == "manus-1.6"
+    assert config._extract_agent_profile("manus/manus-1.6-lite") == "manus-1.6-lite"
+    assert config._extract_agent_profile("manus/manus-1.6-max") == "manus-1.6-max"
+
+
+def test_transform_responses_api_request_adds_manus_params():
+    """Test that transform_responses_api_request adds task_mode and agent_profile"""
+    config = ManusResponsesAPIConfig()
+    
+    input_param = [
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "input_text",
+                    "text": "What's the color of the sky?",
+                }
+            ],
+        }
+    ]
+    
+    optional_params = ResponsesAPIOptionalRequestParams()
+    litellm_params = GenericLiteLLMParams()
+    headers = {}
+    
+    result = config.transform_responses_api_request(
+        model="manus/manus-1.6",
+        input=input_param,
+        response_api_optional_request_params=dict(optional_params),
+        litellm_params=litellm_params,
+        headers=headers,
+    )
+    
+    assert result["task_mode"] == "agent"
+    assert result["agent_profile"] == "manus-1.6"
+    assert "input" in result
+    assert "model" in result
+