Update hook

Author: andrefpoliveira

src/mollie/_hooks/mollie_hooks.py

@@ -1,4 +1,5 @@
 import uuid
+import sys
 
 from httpx import Request
 from typing import Union
@@ -27,14 +28,14 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: Request) -> Un
         :param request: The HTTP request to modify.
         :return: The modified request or an exception.
         """
-        idempotency_key = "idempotency-key"
-
         # Create a copy of the headers
         headers = dict(request.headers or {})
 
         # Add the idempotency key if it doesn't already exist
-        if idempotency_key not in headers or not headers[idempotency_key]:
-            headers[idempotency_key] = generate_idempotency_key()
+        headers = self._handle_idempotency_key(headers)
+
+        # Customize the User Agent header
+        headers = self._customize_user_agent(headers, hook_ctx)
 
         return Request(
             method = request.method,
@@ -43,3 +44,25 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: Request) -> Un
             content = request.content,
             extensions=request.extensions
         )
+    
+    def _handle_idempotency_key(self, headers: dict) -> dict:
+        idempotency_key = "idempotency-key"
+        if idempotency_key not in headers or not headers[idempotency_key]:
+            headers[idempotency_key] = generate_idempotency_key()
+        return headers
+    
+    def _customize_user_agent(self, headers: dict, hook_ctx: BeforeRequestContext) -> dict:
+        user_agent_key = "user-agent"
+
+        gen_version = hook_ctx.config.gen_version
+        sdk_version = hook_ctx.config.sdk_version
+        python_version = sys.version.split(" ", maxsplit=1)[0]
+        package_name = "mollie-api-python-beta"
+
+        new_user_agent = f"Speakeasy/{gen_version} Python/{python_version} {package_name}/{sdk_version}"
+        if hook_ctx.config.globals.custom_user_agent:
+            new_user_agent = f"{new_user_agent} {hook_ctx.config.globals.custom_user_agent}"
+
+        headers[user_agent_key] = new_user_agent
+
+        return headers