fix: only replaced ua if needed (#163)

* fix: don't replace ua when already modified * add eof * fix for providers
mistralai · Dec 4, 2024 · e0e7886 · e0e7886
1 parent dcaabe1
commit e0e7886
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
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
@@ -5,12 +5,18 @@
 
 from .types import BeforeRequestContext, BeforeRequestHook
 
+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):
+            return request
+
         request.headers["user-agent"] = (
-            "mistral-client-python/" + request.headers["user-agent"].split(" ")[1]
+            prefix + current.split(" ")[1]
         )
-        return request
+
+        return request
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
@@ -5,12 +5,18 @@
 
 from .types import BeforeRequestContext, BeforeRequestHook
 
+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):
+            return request
+
         request.headers["user-agent"] = (
-            "mistral-client-python/" + request.headers["user-agent"].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
@@ -5,12 +5,18 @@
 
 from .types import BeforeRequestContext, BeforeRequestHook
 
+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):
+            return request
+
         request.headers["user-agent"] = (
-            "mistral-client-python/" + request.headers["user-agent"].split(" ")[1]
+            prefix + current.split(" ")[1]
         )
+
         return request