vikingdb-python-sdk 0.1.29__tar.gz → 0.1.32__tar.gz

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.29
+Version: 0.1.32
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md
@@ -64,6 +64,8 @@ client = VikingDB(
     auth=auth,
     scheme="https",
     timeout=30,
+    pool_connections=20,
+    pool_maxsize=50,
 )
 index = client.index(
     collection_name=os.environ["VIKINGDB_COLLECTION"],
@@ -85,6 +87,8 @@ client = VikingMem(
     region="cn-beijing",
     auth=auth,
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 # Get collection
@@ -132,6 +136,8 @@ client = VikingKnowledge(
     host="api-knowledgebase.mlp.cn-beijing.volces.com",
     region="cn-beijing",
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 collection = client.collection(
@@ -149,6 +155,25 @@ print(f"request_id={resp.request_id} hits={len(resp.result.result_list or [])}")
 svc_client = VikingKnowledge(auth=APIKey(api_key=os.getenv("VIKING_SERVICE_API_KEY")))
 ```
 
+### Connection Pool Tuning
+
+For high-concurrency synchronous workloads, the client constructors support
+`pool_connections` and `pool_maxsize` and pass them through to
+the underlying `requests` `HTTPAdapter`.
+
+```python
+client = VikingDB(
+    host=os.environ["VIKINGDB_HOST"],
+    region=os.environ["VIKINGDB_REGION"],
+    auth=auth,
+    pool_connections=20,
+    pool_maxsize=50,
+)
+```
+
+- `pool_connections`: number of cached connection pools
+- `pool_maxsize`: max reusable connections per host
+
 ### Example Guides
 
 #### Vector Examples

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/README.md

@@ -36,6 +36,8 @@ client = VikingDB(
     auth=auth,
     scheme="https",
     timeout=30,
+    pool_connections=20,
+    pool_maxsize=50,
 )
 index = client.index(
     collection_name=os.environ["VIKINGDB_COLLECTION"],
@@ -57,6 +59,8 @@ client = VikingMem(
     region="cn-beijing",
     auth=auth,
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 # Get collection
@@ -104,6 +108,8 @@ client = VikingKnowledge(
     host="api-knowledgebase.mlp.cn-beijing.volces.com",
     region="cn-beijing",
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 collection = client.collection(
@@ -121,6 +127,25 @@ print(f"request_id={resp.request_id} hits={len(resp.result.result_list or [])}")
 svc_client = VikingKnowledge(auth=APIKey(api_key=os.getenv("VIKING_SERVICE_API_KEY")))
 ```
 
+### Connection Pool Tuning
+
+For high-concurrency synchronous workloads, the client constructors support
+`pool_connections` and `pool_maxsize` and pass them through to
+the underlying `requests` `HTTPAdapter`.
+
+```python
+client = VikingDB(
+    host=os.environ["VIKINGDB_HOST"],
+    region=os.environ["VIKINGDB_REGION"],
+    auth=auth,
+    pool_connections=20,
+    pool_maxsize=50,
+)
+```
+
+- `pool_connections`: number of cached connection pools
+- `pool_maxsize`: max reusable connections per host
+
 ### Example Guides
 
 #### Vector Examples

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/_client.py

@@ -17,6 +17,7 @@ from volcengine.ServiceInfo import ServiceInfo
 from volcengine.base.Request import Request
 from volcengine.base.Service import Service
 import requests
+from requests.adapters import HTTPAdapter
 
 from .auth import Auth, IAM, APIKey, HeaderAuth
 from .exceptions import (
@@ -26,6 +27,8 @@ from .exceptions import (
 
 
 _REQUEST_ID_HEADER = "X-Tt-Logid"
+_DEFAULT_POOL_CONNECTIONS = 10
+_DEFAULT_POOL_MAXSIZE = 10
 
 
 class Client(Service, ABC):
@@ -41,6 +44,9 @@ class Client(Service, ABC):
         sts_token: str = "",
         scheme: str = "http",
         timeout: int = 30,
+        pool_connections: int = _DEFAULT_POOL_CONNECTIONS,
+        pool_maxsize: int = _DEFAULT_POOL_MAXSIZE,
+        pool_block: bool = False,
     ):
         self.region = region
         self.service = service
@@ -57,7 +63,11 @@ class Client(Service, ABC):
         if isinstance(auth, (IAM, APIKey, HeaderAuth)):
             # volcengine Service.init() 可能读取环境变量或 ~/.volc/config 覆盖 AK/SK，
             # 这里确保使用用户传入的 IAM 凭证，因此不使用super().__init__初始化
-            self.session = requests.session()
+            self.session = self._build_session(
+                pool_connections=pool_connections,
+                pool_maxsize=pool_maxsize,
+                pool_block=pool_block,
+            )
         else:
             raise ValueError("auth must be IAM, APIKey or HeaderAuth type")
 
@@ -85,6 +95,28 @@ class Client(Service, ABC):
             scheme=scheme,
         )
 
+    @staticmethod
+    def _build_session(
+        *,
+        pool_connections: int,
+        pool_maxsize: int,
+        pool_block: bool,
+    ) -> requests.Session:
+        if pool_connections <= 0:
+            raise ValueError("pool_connections must be greater than 0")
+        if pool_maxsize <= 0:
+            raise ValueError("pool_maxsize must be greater than 0")
+
+        session = requests.session()
+        adapter = HTTPAdapter(
+            pool_connections=pool_connections,
+            pool_maxsize=pool_maxsize,
+            pool_block=pool_block,
+        )
+        session.mount("http://", adapter)
+        session.mount("https://", adapter)
+        return session
+
     def prepare_request(self, api_info: ApiInfo, params: Optional[Mapping[str, Any]], doseq: int = 0):
         """Prepare a volcengine request without adding implicit headers."""
         request = Request()

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/knowledge/client.py

@@ -43,6 +43,8 @@ class VikingKnowledge(Client):
         sts_token: str = "",
         scheme: str = "http",
         timeout: int = 30,
+        pool_connections: int = 10,
+        pool_maxsize: int = 10,
     ):
         super().__init__(
             host=host,
@@ -52,6 +54,8 @@ class VikingKnowledge(Client):
             sts_token=sts_token,
             scheme=scheme,
             timeout=timeout,
+            pool_connections=pool_connections,
+            pool_maxsize=pool_maxsize,
         )
 
     def _build_api_info(self):
@@ -148,7 +152,7 @@ class VikingKnowledge(Client):
             payload["rerank_instruction"] = rerank_instruction
         if endpoint_id is not None:
             payload["endpoint_id"] = endpoint_id
-        res = self.json_exception("Rerank", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.json_exception("Rerank", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = RerankResponse.parse_with(res)
         return response
 
@@ -166,11 +170,11 @@ class VikingKnowledge(Client):
         )
         if bool(payload.get("stream")):
             def _gen():
-                for res in self.stream_json_exception("ChatCompletion", {}, json.dumps(payload), headers=headers, timeout=timeout):
+                for res in self.stream_json_exception("ChatCompletion", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout):
                     yield ChatCompletionResponse.parse_with(res)
             return _gen()
         else:
-            res = self.json_exception("ChatCompletion", {}, json.dumps(payload), headers=headers, timeout=timeout)
+            res = self.json_exception("ChatCompletion", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
             return ChatCompletionResponse.parse_with(res)
 
     def service_chat(
@@ -187,9 +191,9 @@ class VikingKnowledge(Client):
         )
         if bool(payload.get("stream")):
             def _gen():
-                for res in self.stream_json_exception("ServiceChat", {}, json.dumps(payload), headers=headers, timeout=timeout):
+                for res in self.stream_json_exception("ServiceChat", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout):
                     yield ServiceChatResponse.parse_with(res)
             return _gen()
         else:
-            res = self.json_exception("ServiceChat", {}, json.dumps(payload), headers=headers, timeout=timeout)
+            res = self.json_exception("ServiceChat", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
             return ServiceChatResponse.parse_with(res)

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/knowledge/collection.py

@@ -47,7 +47,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("AddDoc", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddDoc", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = AddDocResponse.parse_with(res)
         return response
 
@@ -64,7 +64,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("AddDocV2", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddDocV2", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = AddDocV2Response.parse_with(res)
         return response
 
@@ -76,7 +76,7 @@ class KnowledgeCollection:
         timeout: Optional[int] = None,
     ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id}
-        res = self.client.json_exception("DeleteDoc", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("DeleteDoc", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = CommonResponse.model_validate(res)
         return response
 
@@ -91,7 +91,7 @@ class KnowledgeCollection:
         payload = {**self._meta_payload, "doc_id": doc_id}
         if return_token_usage:
             payload["return_token_usage"] = True
-        res = self.client.json_exception("GetDocInfo", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("GetDocInfo", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         data_obj = res["data"] if isinstance(res, dict) and "data" in res else {}
         if not isinstance(data_obj, dict):
             data_obj = {}
@@ -113,7 +113,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("ListDocs", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("ListDocs", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = ListDocsResponse.parse_with(res)
         return response
 
@@ -130,7 +130,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("ListDocsV2", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("ListDocsV2", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = ListDocsV2Response.parse_with(res)
         return response
 
@@ -147,7 +147,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("SearchDocsByFilter", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchDocsByFilter", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = SearchDocsByFilterResponse.parse_with(res)
         return response
 
@@ -160,7 +160,7 @@ class KnowledgeCollection:
         timeout: Optional[int] = None,
     ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id, "meta": [item.model_dump(by_alias=True) for item in meta]}
-        res = self.client.json_exception("UpdateDocMeta", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("UpdateDocMeta", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = CommonResponse.model_validate(res)
         return response
 
@@ -173,7 +173,7 @@ class KnowledgeCollection:
         timeout: Optional[int] = None,
     ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id, "doc_name": doc_name}
-        res = self.client.json_exception("UpdateDoc", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("UpdateDoc", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = CommonResponse.model_validate(res)
         return response
 
@@ -188,7 +188,7 @@ class KnowledgeCollection:
         payload = {**self._meta_payload, "point_id": point_id}
         if get_attachment_link:
             payload["get_attachment_link"] = True
-        res = self.client.json_exception("GetPointInfo", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("GetPointInfo", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         data_obj = res["data"] if isinstance(res, dict) and "data" in res else {}
         if not isinstance(data_obj, dict):
             data_obj = {}
@@ -210,7 +210,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("ListPoints", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("ListPoints", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = ListPointsResponse.parse_with(res)
         return response
 
@@ -227,7 +227,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("AddPoint", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddPoint", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = PointAddResponse.parse_with(res)
         return response
 
@@ -245,7 +245,7 @@ class KnowledgeCollection:
             else dict(update)
         )
         payload = {**self._meta_payload, "point_id": point_id, **upd_payload}
-        res = self.client.json_exception("UpdatePoint", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("UpdatePoint", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = CommonResponse.model_validate(res)
         return response
 
@@ -262,7 +262,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload = {**self._meta_payload, **req_payload}
-        res = self.client.json_exception("DeletePoint", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("DeletePoint", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = CommonResponse.model_validate(res)
         return response
 
@@ -279,7 +279,7 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload, "name": self._meta.collection_name}
-        res = self.client.json_exception("SearchCollection", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchCollection", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = SearchResponse.parse_with(res)
         return response
 
@@ -297,6 +297,6 @@ class KnowledgeCollection:
             else dict(request)
         )
         payload: dict = {**self._meta_payload, **req_payload, "name": self._meta.collection_name}
-        res = self.client.json_exception("SearchKnowledge", {}, json.dumps(payload), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchKnowledge", {}, json.dumps(payload, ensure_ascii=False), headers=headers, timeout=timeout)
         response = SearchKnowledgeResponse.parse_with(res)
         return response

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/memory/client.py

@@ -38,6 +38,8 @@ class VikingMem(Client):
         sts_token: str = "",
         scheme: str = "http",
         timeout: int = 30,
+        pool_connections: int = 10,
+        pool_maxsize: int = 10,
     ):
         """
         Initialize Viking Memory Service
@@ -63,6 +65,8 @@ class VikingMem(Client):
             sts_token=sts_token,
             scheme=scheme,
             timeout=timeout,
+            pool_connections=pool_connections,
+            pool_maxsize=pool_maxsize,
         )
 
     def ping(self):

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/memory/collection.py

@@ -90,7 +90,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("AddEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_add_event(
@@ -139,7 +139,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("AddEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("AddEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def update_event(
@@ -180,7 +180,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("UpdateEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("UpdateEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_update_event(
@@ -221,7 +221,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("UpdateEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("UpdateEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def delete_event(
@@ -251,7 +251,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("DeleteEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("DeleteEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_delete_event(
@@ -281,7 +281,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("DeleteEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("DeleteEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def batch_delete_event(
@@ -315,7 +315,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("BatchDeleteEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("BatchDeleteEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_batch_delete_event(
@@ -349,7 +349,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("BatchDeleteEvent", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("BatchDeleteEvent", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     # ==================== Profile Operations ====================
     
@@ -398,7 +398,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("AddProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_add_profile(
@@ -446,7 +446,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("AddProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("AddProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def update_profile(
@@ -479,7 +479,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("UpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("UpdateProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_update_profile(
@@ -512,7 +512,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("UpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("UpdateProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     def delete_profile(
@@ -542,7 +542,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("DeleteProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("DeleteProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_delete_profile(
@@ -572,7 +572,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("DeleteProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("DeleteProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     def trigger_update_profile(
@@ -606,7 +606,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     async def async_trigger_update_profile(
@@ -631,7 +631,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def batch_delete_profile(
@@ -661,7 +661,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("BatchDeleteProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("BatchDeleteProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_batch_delete_profile(
@@ -691,7 +691,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("BatchDeleteProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("BatchDeleteProfile", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     # ==================== Session Operations ====================
     
@@ -736,7 +736,7 @@ class Collection:
         if store_file is not None:
             params["store_file"] = store_file
 
-        res = self.client.json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_add_session(
@@ -780,7 +780,7 @@ class Collection:
         if store_file is not None:
             params["store_file"] = store_file
 
-        res = await self.client.async_json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     # ==================== Session Info Operations ====================
@@ -812,7 +812,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("GetSessionInfo", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("GetSessionInfo", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_get_session_info(
@@ -842,7 +842,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("GetSessionInfo", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("GetSessionInfo", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     # ==================== Search Operations ====================
@@ -882,7 +882,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("SearchMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     async def async_search_memory(
@@ -920,7 +920,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("SearchMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("SearchMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
     
     def search_event_memory(
@@ -948,7 +948,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("SearchEventMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchEventMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     def search_profile_memory(
@@ -973,7 +973,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = self.client.json_exception("SearchProfileMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = self.client.json_exception("SearchProfileMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     async def async_search_event_memory(
@@ -1004,7 +1004,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("SearchEventMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("SearchEventMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     async def async_search_profile_memory(
@@ -1032,7 +1032,7 @@ class Collection:
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
-        res = await self.client.async_json_exception("SearchProfileMemory", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        res = await self.client.async_json_exception("SearchProfileMemory", {}, json.dumps(params, cls=EnumEncoder, ensure_ascii=False), headers=headers, timeout=timeout)
         return res
 
     

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/vector/client.py

@@ -66,6 +66,8 @@ class VikingDB(Client):
         scheme: str = "https",
         sts_token: str = "",
         timeout: int = 30,
+        pool_connections: int = 10,
+        pool_maxsize: int = 10,
     ) -> None:
         if auth is None:
             raise ValueError("auth is required for VikingDB")
@@ -78,6 +80,8 @@ class VikingDB(Client):
             sts_token=sts_token,
             scheme=scheme,
             timeout=timeout,
+            pool_connections=pool_connections,
+            pool_maxsize=pool_maxsize,
         )
         try:
             resp = self.session.get(f"{scheme}://{host}/api/vikingdb/Ping")
@@ -317,6 +321,8 @@ class VikingVector(VikingDB):
         scheme: str = "https",
         sts_token: str = "",
         timeout: int = 30,
+        pool_connections: int = 10,
+        pool_maxsize: int = 10,
     ) -> None:
         warnings.warn(
             "VikingVector is deprecated; use VikingDB instead.",
@@ -330,4 +336,6 @@ class VikingVector(VikingDB):
             scheme=scheme,
             sts_token=sts_token,
             timeout=timeout,
+            pool_connections=pool_connections,
+            pool_maxsize=pool_maxsize,
         )

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb/version.py

@@ -1,4 +1,4 @@
 # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
-__version__ = '0.1.29'
+__version__ = '0.1.32'

{vikingdb_python_sdk-0.1.29 → vikingdb_python_sdk-0.1.32}/vikingdb_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.29
+Version: 0.1.32
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md
@@ -64,6 +64,8 @@ client = VikingDB(
     auth=auth,
     scheme="https",
     timeout=30,
+    pool_connections=20,
+    pool_maxsize=50,
 )
 index = client.index(
     collection_name=os.environ["VIKINGDB_COLLECTION"],
@@ -85,6 +87,8 @@ client = VikingMem(
     region="cn-beijing",
     auth=auth,
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 # Get collection
@@ -132,6 +136,8 @@ client = VikingKnowledge(
     host="api-knowledgebase.mlp.cn-beijing.volces.com",
     region="cn-beijing",
     scheme="http",
+    pool_connections=20,
+    pool_maxsize=50,
 )
 
 collection = client.collection(
@@ -149,6 +155,25 @@ print(f"request_id={resp.request_id} hits={len(resp.result.result_list or [])}")
 svc_client = VikingKnowledge(auth=APIKey(api_key=os.getenv("VIKING_SERVICE_API_KEY")))
 ```
 
+### Connection Pool Tuning
+
+For high-concurrency synchronous workloads, the client constructors support
+`pool_connections` and `pool_maxsize` and pass them through to
+the underlying `requests` `HTTPAdapter`.
+
+```python
+client = VikingDB(
+    host=os.environ["VIKINGDB_HOST"],
+    region=os.environ["VIKINGDB_REGION"],
+    auth=auth,
+    pool_connections=20,
+    pool_maxsize=50,
+)
+```
+
+- `pool_connections`: number of cached connection pools
+- `pool_maxsize`: max reusable connections per host
+
 ### Example Guides
 
 #### Vector Examples