feat: timeout configuration for LLM clients + vertex (#2972)

2025-06-23 16:55:23 -07:00
parent b95d7b4b13
commit 343dbb5359
9 changed files with 65 additions and 22 deletions
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -249,7 +249,9 @@ def test_coerce_dict_args_unsupported_complex_annotation():
    annotations = {"f": "CustomClass[int]"}
    function_args = {"f": "CustomClass(42)"}

-    with pytest.raises(ValueError, match="Failed to coerce argument 'f' to CustomClass\[int\]: Unsupported annotation: CustomClass\[int\]"):
+    with pytest.raises(
+        ValueError, match=r"Failed to coerce argument 'f' to CustomClass\[int\]: Unsupported annotation: CustomClass\[int\]"
+    ):
        coerce_dict_args_by_annotations(function_args, annotations)