Skip to content

Commit e0e7886

Browse files
authored
fix: only replaced ua if needed (#163)
* fix: don't replace ua when already modified * add eof * fix for providers
1 parent dcaabe1 commit e0e7886

File tree

3 files changed

+22
-4
lines changed

3 files changed

+22
-4
lines changed

packages/mistralai_azure/src/mistralai_azure/_hooks/custom_user_agent.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,18 @@
55

66
from .types import BeforeRequestContext, BeforeRequestHook
77

8+
prefix = "mistral-client-python/"
89

910
class CustomUserAgentHook(BeforeRequestHook):
1011
def before_request(
1112
self, hook_ctx: BeforeRequestContext, request: httpx.Request
1213
) -> Union[httpx.Request, Exception]:
14+
current = request.headers["user-agent"]
15+
if current.startswith(prefix):
16+
return request
17+
1318
request.headers["user-agent"] = (
14-
"mistral-client-python/" + request.headers["user-agent"].split(" ")[1]
19+
prefix + current.split(" ")[1]
1520
)
16-
return request
21+
22+
return request

packages/mistralai_gcp/src/mistralai_gcp/_hooks/custom_user_agent.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,18 @@
55

66
from .types import BeforeRequestContext, BeforeRequestHook
77

8+
prefix = "mistral-client-python/"
89

910
class CustomUserAgentHook(BeforeRequestHook):
1011
def before_request(
1112
self, hook_ctx: BeforeRequestContext, request: httpx.Request
1213
) -> Union[httpx.Request, Exception]:
14+
current = request.headers["user-agent"]
15+
if current.startswith(prefix):
16+
return request
17+
1318
request.headers["user-agent"] = (
14-
"mistral-client-python/" + request.headers["user-agent"].split(" ")[1]
19+
prefix + current.split(" ")[1]
1520
)
21+
1622
return request

src/mistralai/_hooks/custom_user_agent.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,18 @@
55

66
from .types import BeforeRequestContext, BeforeRequestHook
77

8+
prefix = "mistral-client-python/"
89

910
class CustomUserAgentHook(BeforeRequestHook):
1011
def before_request(
1112
self, hook_ctx: BeforeRequestContext, request: httpx.Request
1213
) -> Union[httpx.Request, Exception]:
14+
current = request.headers["user-agent"]
15+
if current.startswith(prefix):
16+
return request
17+
1318
request.headers["user-agent"] = (
14-
"mistral-client-python/" + request.headers["user-agent"].split(" ")[1]
19+
prefix + current.split(" ")[1]
1520
)
21+
1622
return request

0 commit comments

Comments
 (0)