From e62882e17e0824ad7214c94d859326645f512f19 Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Wed, 10 May 2023 09:29:56 -0700 Subject: [PATCH] Adopt / convention for User-Agent header Signed-off-by: Mattt Zmuda --- replicate/client.py | 2 +- tests/test_client.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/replicate/client.py b/replicate/client.py index 4196d6aa..244a331c 100644 --- a/replicate/client.py +++ b/replicate/client.py @@ -84,7 +84,7 @@ def _request(self, method: str, path: str, **kwargs) -> requests.Response: def _headers(self) -> Dict[str, str]: return { "Authorization": f"Token {self._api_token()}", - "User-Agent": f"replicate-python@{__version__}", + "User-Agent": f"replicate-python/{__version__}", } def _api_token(self) -> str: diff --git a/tests/test_client.py b/tests/test_client.py index 1e4d57ce..05fffe08 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -25,7 +25,7 @@ def test_client_sets_authorization_token_and_user_agent_headers(): "https://api.replicate.com/v1/models/test/model/versions", match=[ matchers.header_matcher({"Authorization": "Token abc123"}), - matchers.header_matcher({"User-Agent": f"replicate-python@{__version__}"}), + matchers.header_matcher({"User-Agent": f"replicate-python/{__version__}"}), ], json={"results": []}, )