File tree 3 files changed +22
-4
lines changed
mistralai_azure/src/mistralai_azure/_hooks
mistralai_gcp/src/mistralai_gcp/_hooks
3 files changed +22
-4
lines changed Original file line number Diff line number Diff line change 5
5
6
6
from .types import BeforeRequestContext , BeforeRequestHook
7
7
8
+ prefix = "mistral-client-python/"
8
9
9
10
class CustomUserAgentHook (BeforeRequestHook ):
10
11
def before_request (
11
12
self , hook_ctx : BeforeRequestContext , request : httpx .Request
12
13
) -> Union [httpx .Request , Exception ]:
14
+ current = request .headers ["user-agent" ]
15
+ if current .startswith (prefix ):
16
+ return request
17
+
13
18
request .headers ["user-agent" ] = (
14
- "mistral-client-python/" + request . headers [ "user-agent" ] .split (" " )[1 ]
19
+ prefix + current .split (" " )[1 ]
15
20
)
16
- return request
21
+
22
+ return request
Original file line number Diff line number Diff line change 5
5
6
6
from .types import BeforeRequestContext , BeforeRequestHook
7
7
8
+ prefix = "mistral-client-python/"
8
9
9
10
class CustomUserAgentHook (BeforeRequestHook ):
10
11
def before_request (
11
12
self , hook_ctx : BeforeRequestContext , request : httpx .Request
12
13
) -> Union [httpx .Request , Exception ]:
14
+ current = request .headers ["user-agent" ]
15
+ if current .startswith (prefix ):
16
+ return request
17
+
13
18
request .headers ["user-agent" ] = (
14
- "mistral-client-python/" + request . headers [ "user-agent" ] .split (" " )[1 ]
19
+ prefix + current .split (" " )[1 ]
15
20
)
21
+
16
22
return request
Original file line number Diff line number Diff line change 5
5
6
6
from .types import BeforeRequestContext , BeforeRequestHook
7
7
8
+ prefix = "mistral-client-python/"
8
9
9
10
class CustomUserAgentHook (BeforeRequestHook ):
10
11
def before_request (
11
12
self , hook_ctx : BeforeRequestContext , request : httpx .Request
12
13
) -> Union [httpx .Request , Exception ]:
14
+ current = request .headers ["user-agent" ]
15
+ if current .startswith (prefix ):
16
+ return request
17
+
13
18
request .headers ["user-agent" ] = (
14
- "mistral-client-python/" + request . headers [ "user-agent" ] .split (" " )[1 ]
19
+ prefix + current .split (" " )[1 ]
15
20
)
21
+
16
22
return request
You can’t perform that action at this time.
0 commit comments