some cleanup

databricks · Jul 4, 2024 · da607ba · da607ba
1 parent a302e80
commit da607ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/databricks/sdk/config.py b/databricks/sdk/config.py
@@ -50,7 +50,7 @@ def with_product(product: str, product_version: str):
 
 def with_user_agent_extra(key: str, value: str):
     """[INTERNAL API] Add extra metadata to the User-Agent header when developing a library."""
-    useragent.with_user_agent_extra(key, value)
+    useragent.with_extra(key, value)
 
 
 class Config:

diff --git a/databricks/sdk/useragent.py b/databricks/sdk/useragent.py
@@ -63,7 +63,7 @@ def _reset_product():
     _product_version = "0.0.0"
 
 
-def with_user_agent_extra(key: str, value: str):
+def with_extra(key: str, value: str):
     """Add extra metadata to all requests submitted to Databricks.
 
     User-specified extra metadata can be inserted into request headers to provide additional context to Databricks
@@ -93,7 +93,7 @@ def _reset_extra(extra: List[Tuple[str, str]]):
 
 def with_partner(partner: str):
     """Adds the given partner to the metadata submitted to Databricks on every request."""
-    with_user_agent_extra("partner", partner)
+    with_extra("partner", partner)
 
 
 def _get_upstream_user_agent_info() -> List[Tuple[str, str]]: