scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

test_message_with_name

Browse source
This commit is contained in:
Ishaan Jaff 2025-02-18 21:53:47 -08:00
parent 4828fb1544
commit 0d398c2295
1 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/llm_translation/base_llm_unit_tests.py
View file

@ -196,15 +196,18 @@ class BaseLLMChatTest(ABC):
assert response is not None
def test_message_with_name(self):
litellm.set_verbose = True
base_completion_call_args = self.get_base_completion_call_args()
messages = [
{"role": "user", "content": "Hello", "name": "test_name"},
]
response = self.completion_function(
**base_completion_call_args, messages=messages
)
assert response is not None
try:
litellm.set_verbose = True
base_completion_call_args = self.get_base_completion_call_args()
messages = [
{"role": "user", "content": "Hello", "name": "test_name"},
]
response = self.completion_function(
**base_completion_call_args, messages=messages
)
assert response is not None
except litellm.RateLimitError:
pass
@pytest.mark.parametrize(
"response_format",