fix tests

pydantic · Oct 30, 2024 · f7f1c96 · f7f1c96
1 parent b605920
commit f7f1c96
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_agent.py b/tests/test_agent.py
@@ -274,7 +274,8 @@ def validate_result(ctx: CallContext[None], r: Any) -> Any:
 def test_response_multiple_return_tools(create_module: Callable[[str], Any], union_code: str):
     module_code = f'''
 from pydantic import BaseModel
-from typing import Union, TypeAlias
+from typing import Union
+from typing_extensions import TypeAlias
 
 class Foo(BaseModel):
     a: int