diff --git a/pyproject.toml b/pyproject.toml index 586d919..45d9c3f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,6 +12,7 @@ authors = [{ name = "Replicate, Inc." }] requires-python = ">=3.8" dependencies = [ "httpx>=0.21.0,<1", + "httpx2>=2.12.0; python_version >= \"3.10\"", "packaging", "pydantic>1.10.7", "typing_extensions>=4.5.0", diff --git a/replicate/client.py b/replicate/client.py index 6a79813..191fa81 100644 --- a/replicate/client.py +++ b/replicate/client.py @@ -16,7 +16,10 @@ Union, ) -import httpx +try: + import httpx2 as httpx +except ImportError: + import httpx from typing_extensions import Unpack from replicate.__about__ import __version__ diff --git a/replicate/exceptions.py b/replicate/exceptions.py index f52f9fb..a865700 100644 --- a/replicate/exceptions.py +++ b/replicate/exceptions.py @@ -1,6 +1,9 @@ from typing import TYPE_CHECKING, Optional -import httpx +try: + import httpx2 as httpx +except ImportError: + import httpx if TYPE_CHECKING: from replicate.prediction import Prediction diff --git a/replicate/prediction.py b/replicate/prediction.py index b4ff047..37559d3 100644 --- a/replicate/prediction.py +++ b/replicate/prediction.py @@ -16,7 +16,10 @@ overload, ) -import httpx +try: + import httpx2 as httpx +except ImportError: + import httpx from typing_extensions import NotRequired, TypedDict, Unpack from replicate.exceptions import ModelError, ReplicateError diff --git a/replicate/stream.py b/replicate/stream.py index e837abd..73caa88 100644 --- a/replicate/stream.py +++ b/replicate/stream.py @@ -10,7 +10,10 @@ Union, ) -import httpx +try: + import httpx2 as httpx +except ImportError: + import httpx from typing_extensions import Unpack from replicate import identifier diff --git a/replicate/webhook.py b/replicate/webhook.py index 22dc9ab..826f9c2 100644 --- a/replicate/webhook.py +++ b/replicate/webhook.py @@ -11,6 +11,9 @@ from replicate.resource import Namespace, Resource if TYPE_CHECKING: + try: + import httpx2 as httpx +except ImportError: import httpx