Merge pull request #185 from dashscope/dev/issues-7

Merge SDK headers into single x-dashscope-sdk-client field

Author: luk384090-cloud

dashscope/acli/providers/tongyi.py

@@ -48,12 +48,14 @@ def __init__(
         request_timeout: int = 60,
         protocol: str = "openai",
         base_url: str | None = None,
+        module: str = "app",
     ):
         self.model = model
         self.api_key = api_key
         self.request_timeout = request_timeout
         self.protocol = protocol
         self.base_url = (base_url or DASHSCOPE_BASE_URL).rstrip("/")
+        self.module = module
 
     def _convert_tools(self, tools: list[dict] | None) -> list[dict] | None:
         if not tools:
@@ -126,8 +128,10 @@ def _get_headers(self) -> dict:
         if self.api_key:
             headers["Authorization"] = f"Bearer {self.api_key}"
         if not os.environ.get("DASHSCOPE_DISABLE_SDK_HEADERS"):
-            headers["x-dashscope-sdk-client"] = "acli"
-            headers["x-dashscope-sdk-version"] = __version__
+            parts = ["acli", __version__]
+            if self.module:
+                parts.append(self.module)
+            headers["x-dashscope-sdk-client"] = "/".join(parts)
             headers["x-dashscope-sdk-session-id"] = SDK_SESSION_ID
         return headers
 

dashscope/agentstudio/transport.py

@@ -230,7 +230,7 @@ def build_headers(
         HEADER_AUTHORIZATION: f"Bearer {api_key}",
         HEADER_USER_AGENT: user_agent,
     }
-    headers.update(get_sdk_headers())
+    headers.update(get_sdk_headers(module="agentstudio"))
     if json_body:
         headers["Content-Type"] = "application/json"
         headers["Accept"] = "application/json"

dashscope/api_entities/api_request_factory.py

@@ -53,6 +53,8 @@ def _build_api_request(  # pylint: disable=too-many-branches
     task_id: Optional[str] = None,
     enable_encryption: bool = False,
     pre_task_id: Optional[str] = None,
+    # SDK module tag for the x-dashscope-sdk-client header
+    sdk_module: str = "",
     # Additional parameters for API request data
     **kwargs,
 ):
@@ -192,7 +194,7 @@ def _build_api_request(  # pylint: disable=too-many-branches
             "websocket]",
         )
 
-    merged_headers = dict(get_sdk_headers())
+    merged_headers = dict(get_sdk_headers(module=sdk_module))
     if headers is not None:
         merged_headers.update(headers)
     if merged_headers:

dashscope/api_entities/encryption.py

@@ -18,6 +18,7 @@
     DEFAULT_REQUEST_TIMEOUT_SECONDS,
 )
 from dashscope.common.logging import logger
+from dashscope.common.utils import get_sdk_headers
 
 
 class Encryption:
@@ -90,6 +91,7 @@ def _get_public_keys():
         url = dashscope.base_http_api_url + "/public-keys/latest"
         headers = {
             "Authorization": f"Bearer {dashscope.api_key}",
+            **get_sdk_headers(module="utils"),
         }
 
         response = requests.get(

dashscope/app/application.py

@@ -240,6 +240,7 @@ def call(  # type: ignore[override]
             workspace=workspace,
             api_key=api_key,
             is_service=False,
+            sdk_module="app",
             **parameters,
         )
         # call request service.

dashscope/audio/http_tts/http_speech_synthesizer.py

@@ -203,7 +203,7 @@ def _http_call(
 
         # Get base URL
         import dashscope
-        from dashscope.common.utils import join_url
+        from dashscope.common.utils import get_sdk_headers, join_url
 
         if url:
             base_url = url
@@ -233,6 +233,9 @@ def _http_call(
             stream=stream,
         )
 
+        # Tag sdk module (custom headers below may still override)
+        request.add_headers(get_sdk_headers(module="audio"))
+
         # Add custom headers
         if headers:
             request.add_headers(headers)

dashscope/audio/qwen_omni/omni_realtime.py

@@ -14,7 +14,7 @@
 import dashscope
 from dashscope.common.error import ModelRequired
 from dashscope.common.logging import logger
-from dashscope.common.utils import get_user_agent
+from dashscope.common.utils import get_sdk_headers, get_user_agent
 
 
 class OmniRealtimeCallback:
@@ -259,6 +259,7 @@ def _get_websocket_header(self):
         headers = {
             "user-agent": ua,
             "Authorization": "Bearer " + self.apikey,
+            **get_sdk_headers(module="audio"),
         }
         if self.user_headers:
             headers = {**self.user_headers, **headers}

dashscope/audio/qwen_tts_realtime/qwen_tts_realtime.py

@@ -12,7 +12,7 @@
 import dashscope
 from dashscope.common.error import ModelRequired
 from dashscope.common.logging import logger
-from dashscope.common.utils import get_user_agent
+from dashscope.common.utils import get_sdk_headers, get_user_agent
 
 
 class QwenTtsRealtimeCallback:
@@ -117,6 +117,7 @@ def _get_websocket_header(self):
         headers = {
             "user-agent": ua,
             "Authorization": "Bearer " + self.apikey,
+            **get_sdk_headers(module="audio"),
         }
         if self.user_headers:
             headers = {**self.user_headers, **headers}

dashscope/audio/tts_v2/speech_synthesizer.py

@@ -22,7 +22,7 @@
     RequestFailure,
 )
 from dashscope.common.logging import logger
-from dashscope.common.utils import get_user_agent
+from dashscope.common.utils import get_sdk_headers, get_user_agent
 from dashscope.protocol.websocket import (
     ACTION_KEY,
     ERROR_MESSAGE,
@@ -183,6 +183,7 @@ def get_websocket_headers(self, headers, workspace):
         self.headers = {
             "user-agent": ua,
             "Authorization": "Bearer " + self.apikey,
+            **get_sdk_headers(module="audio"),
         }
         if headers:
             self.headers = {**self.headers, **headers}

dashscope/cli/__init__.py

@@ -22,6 +22,7 @@
 import dashscope  # noqa: E402
 from dashscope.cli.common import err_console  # noqa: E402
 from dashscope.common.error import AuthenticationError  # noqa: E402
+from dashscope.common.utils import set_sdk_client  # noqa: E402
 from dashscope.cli import (  # noqa: E402
     application,
     auth,
@@ -43,6 +44,8 @@
     video_synthesis,
 )
 
+set_sdk_client("python-cli")
+
 
 # ---------------------------------------------------------------------------
 # Legacy command compatibility layer