thread openapi_config and pool_threads through as well

pinecone-io · Oct 25, 2024 · 32a999b · 32a999b
1 parent a8bd692
commit 32a999b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pinecone/data/index.py b/pinecone/data/index.py
@@ -100,14 +100,15 @@ def __init__(
         self._config = ConfigBuilder.build(
             api_key=api_key, host=host, additional_headers=additional_headers, **kwargs
         )
-        openapi_config = ConfigBuilder.build_openapi_config(self._config, openapi_config)
+        self._openapi_config = ConfigBuilder.build_openapi_config(self._config, openapi_config)
+        self._pool_threads = pool_threads
 
         self._vector_api = setup_openapi_client(
             api_client_klass=ApiClient,
             api_klass=DataPlaneApi,
             config=self._config,
-            openapi_config=openapi_config,
-            pool_threads=pool_threads,
+            openapi_config=self._openapi_config,
+            pool_threads=self._pool_threads,
             api_version=API_VERSION,
         )
 
@@ -121,8 +122,8 @@ def _load_plugins(self):
             # halt client initialization.
             openapi_client_builder = build_plugin_setup_client(
                 config=self._config,
-                openapi_config=self.openapi_config,
-                pool_threads=self.pool_threads,
+                openapi_config=self._openapi_config,
+                pool_threads=self._pool_threads,
             )
             install_plugins(self, openapi_client_builder)
         except Exception as e: