Skip to content

Commit 3d0f547

Browse files
committed
Remove dependency on requests_toolbelt
Signed-off-by: Mattt Zmuda <mattt@replicate.com>
1 parent 7dbe666 commit 3d0f547

File tree

3 files changed

+15
-12
lines changed

3 files changed

+15
-12
lines changed

pyproject.toml

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,7 @@ readme = "README.md"
1010
license = { file = "LICENSE" }
1111
authors = [{ name = "Replicate, Inc." }]
1212
requires-python = ">=3.8"
13-
dependencies = [
14-
"packaging",
15-
"pydantic>1",
16-
"requests>2",
17-
"requests_toolbelt>=1"
18-
]
13+
dependencies = ["packaging", "pydantic>1", "requests>2"]
1914
optional-dependencies = { dev = [
2015
"black",
2116
"mypy",

replicate/client.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
import requests
77
from requests.adapters import HTTPAdapter, Retry
8-
from requests_toolbelt.cookies.forgetful import ForgetfulCookieJar # type: ignore
8+
from requests.cookies import RequestsCookieJar
99

1010
from replicate.__about__ import __version__
1111
from replicate.exceptions import ModelError, ReplicateError
@@ -141,7 +141,19 @@ def run(self, model_version: str, **kwargs) -> Union[Any, Iterator[Any]]:
141141
return prediction.output
142142

143143

144+
class _NonpersistentCookieJar(RequestsCookieJar):
145+
"""
146+
A cookie jar that doesn't persist cookies between requests.
147+
"""
148+
149+
def set(self, name, value, **kwargs) -> None:
150+
return
151+
152+
def set_cookie(self, cookie, *args, **kwargs) -> None:
153+
return
154+
155+
144156
def _create_session() -> requests.Session:
145157
s = requests.Session()
146-
s.cookies = ForgetfulCookieJar()
158+
s.cookies = _NonpersistentCookieJar()
147159
return s

requirements.txt

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,6 @@ packaging==23.0
1515
pydantic==1.10.7
1616
# via replicate (pyproject.toml)
1717
requests==2.28.2
18-
# via
19-
# replicate (pyproject.toml)
20-
# requests-toolbelt
21-
requests-toolbelt==1.0.0
2218
# via replicate (pyproject.toml)
2319
typing-extensions==4.5.0
2420
# via pydantic

0 commit comments

Comments
 (0)