test(groq): add test_litellm regression for assistant metadata strip

This commit is contained in:
Utsab Dahal 2026-04-12 14:18:41 +05:45
parent fe656d8d4f
commit 6170840a35

View file

@ -0,0 +1,31 @@
from typing import List, cast
from litellm.llms.groq.chat.transformation import GroqChatConfig
from litellm.types.llms.openai import AllMessageValues
def test_transform_messages_strips_assistant_provider_specific_fields() -> None:
"""Groq payloads should not include LiteLLM internal assistant metadata."""
config = GroqChatConfig()
messages = cast(
List[AllMessageValues],
[
{
"role": "assistant",
"content": "Tool metadata",
"provider_specific_fields": {
"mcp_list_tools": [{"name": "weather"}],
"mcp_tool_calls": [{"id": "call_123"}],
},
}
],
)
transformed = cast(
List[AllMessageValues],
config._transform_messages(messages=messages, model="qwen/qwen3-32b"),
)
assert transformed[0]["role"] == "assistant"
assert transformed[0].get("content") == "Tool metadata"
assert "provider_specific_fields" not in transformed[0]