test(test_function_call_parsing.py): handle anthropic internal server error

2024-09-05 09:27:42 -07:00 · 2024-09-05 09:27:42 -07:00 · e32d4a15b2
commit e32d4a15b2
parent 773914bcb7
1 changed files with 3 additions and 3 deletions
--- a/litellm/tests/test_function_call_parsing.py
+++ b/litellm/tests/test_function_call_parsing.py
@ -134,11 +134,11 @@ def trade(model_name: str) -> List[Trade]:
                "function": {"name": tool_spec["function"]["name"]},  # type: ignore
            },
        )
+        calls = response.choices[0].message.tool_calls
+        trades = [trade for call in calls for trade in parse_call(call)]
+        return trades
    except litellm.InternalServerError:
        pass
-    calls = response.choices[0].message.tool_calls
-    trades = [trade for call in calls for trade in parse_call(call)]
-    return trades


@pytest.mark.parametrize(