scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(cloudflare): handle API errors and missing response keys gracefully

Browse source 
Signed-off-by: Soumik Sarker <ronodhirsoumik@gmail.com>
This commit is contained in:
ronodhirSoumik 2026-04-19 01:33:32 +06:00
parent 319839d8d6
commit dd8f3624d4
1 changed files with 23 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
litellm/llms/cloudflare/chat/transformation.py
View file

@ -147,9 +147,29 @@ class CloudflareChatConfig(BaseConfig):
) -> ModelResponse:
completion_response = raw_response.json()
model_response.choices[0].message.content = completion_response["result"][ # type: ignore
"response"
]
if not completion_response.get("success", True):
errors = completion_response.get("errors", [])
raise CloudflareError(
status_code=raw_response.status_code,
message=f"Cloudflare Error: {errors}",
)
result = completion_response.get("result", {})
if result is None:
raise CloudflareError(
status_code=raw_response.status_code,
message=f"Cloudflare Error: result is None. Response: {completion_response}",
)
if "response" in result:
model_response.choices[0].message.content = result["response"]
elif isinstance(result, str):
model_response.choices[0].message.content = result
else:
raise CloudflareError(
status_code=raw_response.status_code,
message=f"Cloudflare Error: 'response' key missing from result. Response: {completion_response}",
)
prompt_tokens = litellm.utils.get_token_count(messages=messages, model=model)
completion_tokens = len(