fix tests

pydantic · Oct 30, 2024 · 39914b8 · 39914b8
1 parent 40c68d6
commit 39914b8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_agent.py b/tests/test_agent.py
@@ -269,7 +269,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