From ffefa326519c031b4a078c5ae8c16955363ba7da Mon Sep 17 00:00:00 2001 From: Samuel Colvin Date: Fri, 8 Nov 2024 19:49:37 +0000 Subject: [PATCH] fix for older python --- pydantic_ai/models/function.py | 4 ++-- tests/test_streaming.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pydantic_ai/models/function.py b/pydantic_ai/models/function.py index 18b137da8..6a2880a0d 100644 --- a/pydantic_ai/models/function.py +++ b/pydantic_ai/models/function.py @@ -3,7 +3,7 @@ from collections.abc import Iterable, Iterator, Mapping, Sequence from dataclasses import dataclass from itertools import chain -from typing import Callable, cast +from typing import Callable, Union, cast from typing_extensions import TypeAlias, overload @@ -37,7 +37,7 @@ class DeltaToolCall: DeltaToolCalls = dict[int, DeltaToolCall] FunctionDef: TypeAlias = Callable[[list[Message], AgentInfo], LLMMessage] -StreamFunctionDef: TypeAlias = Callable[[list[Message], AgentInfo], Iterable[str] | Iterable[DeltaToolCalls]] +StreamFunctionDef: TypeAlias = Callable[[list[Message], AgentInfo], Union[Iterable[str], Iterable[DeltaToolCalls]]] @dataclass diff --git a/tests/test_streaming.py b/tests/test_streaming.py index 8807b84a8..c87ee2062 100644 --- a/tests/test_streaming.py +++ b/tests/test_streaming.py @@ -1,3 +1,5 @@ +from __future__ import annotations as _annotations + import json from collections.abc import Iterable