scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

Merge pull request #24373 from Chesars/fix/zhipu-finish-reason-mapping

Browse source 
fix: map Zhipu GLM non-standard finish_reason values
This commit is contained in:
Cesar Garcia 2026-03-22 18:42:04 -03:00 • committed by GitHub
commit de91bbb9ff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/litellm_core_utils/core_helpers.py
View file

@ -89,6 +89,9 @@ _FINISH_REASON_MAP: dict[str, OpenAIChatCompletionFinishReason] = {
"IMAGE_PROHIBITED_CONTENT": "content_filter",
"TOO_MANY_TOOL_CALLS": "stop",
"MALFORMED_RESPONSE": "stop",
# Zhipu GLM
"network_error": "stop",
"sensitive": "content_filter",
# Bedrock
"guardrail_intervened": "content_filter",
# OpenAI passthrough

8
tests/test_litellm/litellm_core_utils/test_core_helpers.py
View file

@ -126,6 +126,14 @@ class TestMapFinishReasonBedrock:
assert map_finish_reason("guardrail_intervened") == "content_filter"
class TestMapFinishReasonZhipu:
def test_network_error(self):
assert map_finish_reason("network_error") == "stop"
def test_sensitive(self):
assert map_finish_reason("sensitive") == "content_filter"
class TestMapFinishReasonOpenAIPassthrough:
@pytest.mark.parametrize(
"reason", ["stop", "length", "tool_calls", "function_call", "content_filter"]