fix(utils.py): handle api assistant returning 'null' role

Fixes https://github.com/BerriAI/litellm/issues/3621
2024-05-13 16:45:12 -07:00 · 2024-05-13 16:45:12 -07:00 · 240c9550f0
commit 240c9550f0
parent 8c37c47c8d
3 changed files with 47 additions and 1 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -7709,7 +7709,7 @@ def convert_to_model_response_object(
            for idx, choice in enumerate(response_object["choices"]):
                message = Message(
                    content=choice["message"].get("content", None),
-                    role=choice["message"]["role"],
+                    role=choice["message"]["role"] or "assistant",
                    function_call=choice["message"].get("function_call", None),
                    tool_calls=choice["message"].get("tool_calls", None),
                )