feat: support api indicator

baidubce · Jan 11, 2024 · cec0244 · cec0244
1 parent 5c507b6
commit cec0244
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/src/qianfan/resources/console/utils.py b/src/qianfan/resources/console/utils.py
@@ -57,7 +57,7 @@ def inner(*args: Any, **kwargs: Any) -> QfResponse:
             ),
         )
         req = func(*args, **kwargs)
-        req.headers["request-source"] = f"qianfan_py_sdk-{VERSION}"
+        req.headers["request-source"] = f"qianfan_py_sdk_v{VERSION}"
         return ConsoleAPIRequestor(**kwargs)._request_console_api(
             req, ak, sk, retry_config
         )
@@ -97,7 +97,7 @@ async def inner(*args: Any, **kwargs: Any) -> QfResponse:
             ),
         )
         req = await func(*args, **kwargs)
-        req.headers["request-source"] = f"qianfan_py_sdk-{VERSION}"
+        req.headers["request-source"] = f"qianfan_py_sdk_v{VERSION}"
         return await ConsoleAPIRequestor(**kwargs)._async_request_console_api(
             req, ak, sk, retry_config
         )

diff --git a/src/qianfan/resources/llm/base.py b/src/qianfan/resources/llm/base.py
@@ -489,7 +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}"}
+        kwargs["extra_parameters"] = {"user_agent": f"qianfan_py_sdk_v{VERSION}"}
         return kwargs
 
     def _data_postprocess(self, data: QfResponse) -> QfResponse:

diff --git a/src/qianfan/tests/completion_test.py b/src/qianfan/tests/completion_test.py
@@ -427,5 +427,6 @@ async def test_batch_predict_async():
 def test_sdk_indicator():
     res = qianfan.Completion().do("hi")
     assert (
-        res["_request"]["extra_parameters"]["user_agent"] == f"qianfan_py_sdk-{VERSION}"
+        res["_request"]["extra_parameters"]["user_agent"]
+        == f"qianfan_py_sdk_v{VERSION}"
     )
diff --git a/src/qianfan/tests/service_test.py b/src/qianfan/tests/service_test.py
@@ -78,4 +78,4 @@ def test_service_list():
 def test_sdk_console_indicator():
     res = Service.list()
     # header不区分大小写，flask受到后将起转换成大写：
-    assert res["_header"]["Request-Source"] == f"qianfan_py_sdk-{VERSION}"
+    assert res["_header"]["Request-Source"] == f"qianfan_py_sdk_v{VERSION}"