Skip to content

Commit 2ee5b60

Browse files
Use 'Union' return type to satisfy pre-commit checks for python version 3.9.
1 parent 5730bcd commit 2ee5b60

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

singlestoredb/ai/chat.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from typing import Any
33
from typing import Callable
44
from typing import Optional
5+
from typing import Union
56

67
import httpx
78

@@ -73,7 +74,7 @@ def SingleStoreChatFactory(
7374
http_client: Optional[httpx.Client] = None,
7475
obo_token_getter: Optional[Callable[[], Optional[str]]] = None,
7576
**kwargs: Any,
76-
) -> ChatOpenAI | ChatBedrockConverse:
77+
) -> Union[ChatOpenAI, ChatBedrockConverse]:
7778
"""Return a chat model instance (ChatOpenAI or ChatBedrockConverse) based on prefix.
7879
7980
The fully-qualified model name is expected to contain a prefix followed by

0 commit comments

Comments
 (0)