diff --git a/packages/mistralai_azure/src/mistralai_azure/_hooks/custom_user_agent.py b/packages/mistralai_azure/src/mistralai_azure/_hooks/custom_user_agent.py index 00bac95..2b71a96 100644 --- a/packages/mistralai_azure/src/mistralai_azure/_hooks/custom_user_agent.py +++ b/packages/mistralai_azure/src/mistralai_azure/_hooks/custom_user_agent.py @@ -5,18 +5,18 @@ from .types import BeforeRequestContext, BeforeRequestHook -prefix = "mistral-client-python/" +PREFIX = "mistral-client-python/" class CustomUserAgentHook(BeforeRequestHook): def before_request( self, hook_ctx: BeforeRequestContext, request: httpx.Request ) -> Union[httpx.Request, Exception]: current = request.headers["user-agent"] - if current.startswith(prefix): + if current.startswith(PREFIX): return request request.headers["user-agent"] = ( - prefix + current.split(" ")[1] + PREFIX + current.split(" ")[1] ) return request \ No newline at end of file diff --git a/packages/mistralai_gcp/src/mistralai_gcp/_hooks/custom_user_agent.py b/packages/mistralai_gcp/src/mistralai_gcp/_hooks/custom_user_agent.py index 6a7083d..77df6ae 100644 --- a/packages/mistralai_gcp/src/mistralai_gcp/_hooks/custom_user_agent.py +++ b/packages/mistralai_gcp/src/mistralai_gcp/_hooks/custom_user_agent.py @@ -5,18 +5,18 @@ from .types import BeforeRequestContext, BeforeRequestHook -prefix = "mistral-client-python/" +PREFIX = "mistral-client-python/" class CustomUserAgentHook(BeforeRequestHook): def before_request( self, hook_ctx: BeforeRequestContext, request: httpx.Request ) -> Union[httpx.Request, Exception]: current = request.headers["user-agent"] - if current.startswith(prefix): + if current.startswith(PREFIX): return request request.headers["user-agent"] = ( - prefix + current.split(" ")[1] + PREFIX + current.split(" ")[1] ) return request diff --git a/src/mistralai/_hooks/custom_user_agent.py b/src/mistralai/_hooks/custom_user_agent.py index 29675c0..212f217 100644 --- a/src/mistralai/_hooks/custom_user_agent.py +++ b/src/mistralai/_hooks/custom_user_agent.py @@ -5,18 +5,18 @@ from .types import BeforeRequestContext, BeforeRequestHook -prefix = "mistral-client-python/" +PREFIX = "mistral-client-python/" class CustomUserAgentHook(BeforeRequestHook): def before_request( self, hook_ctx: BeforeRequestContext, request: httpx.Request ) -> Union[httpx.Request, Exception]: current = request.headers["user-agent"] - if current.startswith(prefix): + if current.startswith(PREFIX): return request request.headers["user-agent"] = ( - prefix + current.split(" ")[1] + PREFIX + current.split(" ")[1] ) return request