From cc754ff93eca3452cf4199d4e968ede120c3f7a6 Mon Sep 17 00:00:00 2001
From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
Date: Fri, 17 Oct 2025 17:26:19 +0200
Subject: [PATCH] Add attributes to user_agent (#170)

---
 src/kernels/utils.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/kernels/utils.py b/src/kernels/utils.py
index 0b375e2..958e94e 100644
--- a/src/kernels/utils.py
+++ b/src/kernels/utils.py
@@ -528,11 +528,16 @@ def _get_user_agent(
         return None
 
     if user_agent is None:
-        user_agent = {
-            "kernels": __version__,
-            "torch": torch.__version__,
-            "build_variant": build_variant(),
-            "file_type": "kernel",
-        }
-
+        user_agent = {}
+    if isinstance(user_agent, dict):
+        user_agent.update(
+            {
+                "kernels": __version__,
+                "torch": torch.__version__,
+                "build_variant": build_variant(),
+                "file_type": "kernel",
+            }
+        )
+    elif isinstance(user_agent, str):
+        user_agent += f"; kernels/{__version__}; torch/{torch.__version__}; build_variant/{build_variant()}; file_type/kernel"
     return user_agent