feat[resource] add sdk indicator

src/qianfan/resources/console/utils.py

@@ -22,6 +22,7 @@
 from qianfan.errors import InvalidArgumentError
 from qianfan.resources.requestor.console_requestor import ConsoleAPIRequestor
 from qianfan.resources.typing import ParamSpec, QfRequest, QfResponse, RetryConfig
+from qianfan.version import VERSION
 
 P = ParamSpec("P")
 
@@ -56,6 +57,7 @@ def inner(*args: Any, **kwargs: Any) -> QfResponse:
             ),
         )
         req = func(*args, **kwargs)
+        req.headers["request-source"] = f"qianfan_py_sdk-{VERSION}"
         return ConsoleAPIRequestor(**kwargs)._request_console_api(
             req, ak, sk, retry_config
         )
@@ -95,6 +97,7 @@ async def inner(*args: Any, **kwargs: Any) -> QfResponse:
             ),
         )
         req = await func(*args, **kwargs)
+        req.headers["request-source"] = f"qianfan_py_sdk-{VERSION}"
         return await ConsoleAPIRequestor(**kwargs)._async_request_console_api(
             req, ak, sk, retry_config
         )

src/qianfan/resources/llm/base.py

@@ -40,6 +40,7 @@
 from qianfan.resources.requestor.openapi_requestor import create_api_requestor
 from qianfan.resources.typing import JsonBody, QfLLMInfo, QfResponse, RetryConfig
 from qianfan.utils import log_info, log_warn, utils
+from qianfan.version import VERSION
 
 # This is used when user provides `endpoint`
 # In such cases, SDK cannot know which model the user is using
@@ -488,6 +489,7 @@ def _generate_body(
                     f"The required key `{key}` is not provided."
                 )
         kwargs["stream"] = stream
+        kwargs["extra_parameters"] = {"user_agent": f"qianfan_py_sdk-{VERSION}"}
         return kwargs
 
     def _data_postprocess(self, data: QfResponse) -> QfResponse:

src/qianfan/tests/completion_test.py

@@ -24,6 +24,7 @@
 import qianfan
 import qianfan.tests.utils
 from qianfan.tests.utils import EnvHelper, fake_access_token
+from qianfan.version import VERSION
 
 QIANFAN_SUPPORT_COMPLETION_MOCK_MODEL = {
     "ERNIE-Bot",
@@ -421,3 +422,10 @@ async def test_batch_predict_async():
     assert 5 >= time.time() - start_time >= 3
     for input, output in zip(prompt_list, results):
         assert input in output["result"]
+
+
+def test_sdk_indicator():
+    res = qianfan.Completion().do("hi")
+    assert (
+        res["_request"]["extra_parameters"]["user_agent"] == f"qianfan_py_sdk-{VERSION}"
+    )

src/qianfan/tests/service_test.py

@@ -19,6 +19,7 @@
 
 from qianfan.resources import Service
 from qianfan.resources.console.consts import DeployPoolType, ServiceType
+from qianfan.version import VERSION
 
 
 def test_create_service():
@@ -72,3 +73,9 @@ def test_service_list():
         resp = Service.list(api_type_filter=[item])
         for common_service in resp["result"]["common"]:
             assert common_service["apiType"] == item.value
+
+
+def test_sdk_console_indicator():
+    res = Service.list()
+    # header不区分大小写，flask受到后将起转换成大写：
+    assert res["_header"]["Request-Source"] == f"qianfan_py_sdk-{VERSION}"