Add with_library(product, version) to track usage of each upstream …

…library separately This PR adds `with_library(product, version)` method to clients & config, creating a copy with the user-agent extra propagated only for the relevant calls
databricks · Feb 28, 2024 · 751496d · 751496d
1 parent 9813180
commit 751496d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/databricks/sdk/__init__.py b/databricks/sdk/__init__.py
diff --git a/databricks/sdk/config.py b/databricks/sdk/config.py
@@ -251,6 +251,11 @@ def with_user_agent_extra(self, key: str, value: str) -> 'Config':
         self._user_agent_other_info.append(f"{key}/{value}")
         return self
 
+    def with_library(self, product: str, version: str) -> 'Config':
+        config_copy = self.copy()
+        config_copy.with_user_agent_extra(product, version)
+        return config_copy
+
     @property
     def oidc_endpoints(self) -> Optional[OidcEndpoints]:
         self._fix_host_if_needed()