修复压测无法正常工作的问题

baidubce · Sep 24, 2024 · 3c28d39 · 3c28d39
1 parent 6434f71
commit 3c28d39
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/qianfan/resources/requestor/openapi_requestor.py b/python/qianfan/resources/requestor/openapi_requestor.py
@@ -57,8 +57,6 @@
 
 _T = TypeVar("_T")
 
-_sync_stream_thread_pool = ThreadPoolExecutor(20)
-
 
 class QfAPIRequestor(BaseAPIRequestor):
     """
@@ -72,6 +70,7 @@ def __init__(self, **kwargs: Any) -> None:
         super().__init__(**kwargs)
         self._token_limiter = TokenLimiter(**kwargs)
         self._async_token_limiter = AsyncTokenLimiter(**kwargs)
+        self._sync_stream_thread_pool = ThreadPoolExecutor(20)
 
     def _retry_if_token_expired(self, func: Callable[..., _T]) -> Callable[..., _T]:
         """
@@ -488,7 +487,7 @@ def _inner_worker() -> None:
 
                     is_closed = True
 
-                _sync_stream_thread_pool.submit(_inner_worker)
+                self._sync_stream_thread_pool.submit(_inner_worker)
                 while not is_closed or not data.empty():
                     try:
                         yield data.get(timeout=0.5)