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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.30
+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.30 → 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.30 → 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.30 → 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):

{vikingdb_python_sdk-0.1.30 → 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.30 → 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.30 → 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.30'
+__version__ = '0.1.32'

{vikingdb_python_sdk-0.1.30 → 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.30
+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