File tree Expand file tree Collapse file tree 3 files changed +22
-4
lines changed
mistralai_azure/src/mistralai_azure/_hooks
mistralai_gcp/src/mistralai_gcp/_hooks Expand file tree Collapse file tree 3 files changed +22
-4
lines changed Original file line number Diff line number Diff line change 55
66from .types import BeforeRequestContext , BeforeRequestHook
77
8+ prefix = "mistral-client-python/"
89
910class 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
Original file line number Diff line number Diff line change 55
66from .types import BeforeRequestContext , BeforeRequestHook
77
8+ prefix = "mistral-client-python/"
89
910class 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
Original file line number Diff line number Diff line change 55
66from .types import BeforeRequestContext , BeforeRequestHook
77
8+ prefix = "mistral-client-python/"
89
910class 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
You can’t perform that action at this time.
0 commit comments