kensho-kfinance 1.1.0__tar.gz → 1.2.0__tar.gz

kensho_kfinance-1.2.0/.readthedocs.yaml

@@ -0,0 +1,23 @@
+version: 2
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.10"
+    # You can also specify other tool versions:
+    # nodejs: "16"
+  # borrowed from here:
+  # https://docs.readthedocs.com/platform/stable/build-customization.html#avoid-having-a-dirty-git-index
+  jobs:
+    pre_install:
+      - git update-index --assume-unchanged docs/conf.py
+
+# Build documentation in the docs/ directory with Sphinx
+sphinx:
+   configuration: docs/conf.py
+
+# Dependencies required to build your docs
+python:
+  install:
+    - method: pip
+      path: .
+    - requirements: docs/requirements.txt

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 1.1.0
+Version: 1.2.0
 Summary: Python CLI for kFinance
 Author-email: Luke Brown <luke.brown@kensho.com>, Michelle Keoy <michelle.keoy@kensho.com>, Keith Page <keith.page@kensho.com>, Matthew Rosen <matthew.rosen@kensho.com>, Nick Roshdieh <nick.roshdieh@kensho.com>
 Project-URL: source, https://github.com/kensho-technologies/kfinance
@@ -28,6 +28,7 @@ Requires-Dist: mypy<2,>=1.15.0; extra == "dev"
 Requires-Dist: pytest<7,>=6.1.2; extra == "dev"
 Requires-Dist: pytest-cov<7,>=6.0.0; extra == "dev"
 Requires-Dist: ruff<1,>=0.9.4; extra == "dev"
+Requires-Dist: requests_mock<1.2,>=1.1; extra == "dev"
 Dynamic: license-file
 
 # kFinance

kensho_kfinance-1.2.0/docs/conf.py

@@ -0,0 +1,66 @@
+# Configuration file for the Sphinx documentation builder.
+#
+# For the full list of built-in configuration values, see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Project information -----------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+from importlib.metadata import version as get_version
+
+project = 'kensho-kfinance'
+copyright = '2025, Kensho Technologies'
+author = 'Kensho Technologies'
+
+# borrowed from here:
+# https://setuptools-scm.readthedocs.io/en/latest/usage/#usage-from-sphinx
+release: str = get_version(project)
+# for example take major/minor
+version: str = ".".join(release.split('.')[:2])
+
+# -- General configuration ---------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
+
+# borrowed from internal Kensho Sphinx configuration
+extensions = [
+    "sphinx.ext.autodoc",
+    "sphinx.ext.intersphinx",
+    "sphinx.ext.coverage",
+    "sphinx.ext.mathjax",
+    "sphinx.ext.githubpages",
+    "sphinx.ext.autosummary",
+    "sphinx.ext.viewcode",
+    "sphinx.ext.napoleon",
+    # 3rd party extensions
+    # m2r2 is to add support to .md files specifically to include README.md files.
+    # See this discussion: https://github.com/sphinx-doc/sphinx/issues/7000
+    "m2r2",
+    # A ReadTheDocs theme for Sphinx
+    "sphinx_rtd_theme",
+]
+
+napoleon_google_docstring = True
+napoleon_use_ivar = True
+
+autosummary_generate = True
+
+# Don't prepend module path prefixes to function definitions. Makes automodule docs less cluttered
+add_module_names = False
+
+# Add any paths that contain templates here, relative to this directory.
+templates_path = ["templates"]
+
+# The suffix of source filenames.
+source_suffix = [".rst", ".md"]
+
+
+
+# -- Options for HTML output -------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
+
+html_theme = 'sphinx_rtd_theme'
+
+# https://sphinx-rtd-theme.readthedocs.io/en/stable/configuring.html
+html_theme_options = {"body_min_width": 0, "body_max_width": "none"}
+
+# https://www.sphinx-doc.org/en/1.4.9/config.html#confval-html_sidebars
+html_sidebars = {"**": ["globaltoc.html", "relations.html", "searchbox.html"]}

kensho_kfinance-1.2.0/docs/index.rst

@@ -0,0 +1,18 @@
+Index
+#####################
+
+Documentation page for kensho-kfinance library.
+
+.. toctree::
+   :hidden:
+   :glob:
+   :maxdepth: 1
+
+   kfinance <kfinance>
+   llm_tools <llm_tools>
+
+Indices and tables
+==================
+
+* :ref:`genindex`
+* :ref:`modindex`

kensho_kfinance-1.2.0/docs/kfinance.rst

@@ -0,0 +1,7 @@
+kfinance
+========
+
+.. automodule:: kfinance.kfinance
+   :members:
+   :undoc-members:
+   :inherited-members:

kensho_kfinance-1.2.0/docs/llm_tools.rst

@@ -0,0 +1,7 @@
+llm_tools
+=========
+
+.. automodule:: kfinance.llm_tools
+   :members:
+   :undoc-members:
+   :inherited-members:

kensho_kfinance-1.2.0/docs/requirements.txt

@@ -0,0 +1,4 @@
+sphinx==8.1.3
+m2r2==0.3.3.post2
+sphinx-rtd-theme==3.0.2
+docutils==0.20

kensho_kfinance-1.2.0/docs/templates/apidoc/package.rst_t

@@ -0,0 +1,50 @@
+{% raw %}
+{%- macro automodule(modname, options) -%}
+.. automodule:: {{ modname }}
+{%- for option in options %}
+   :{{ option }}:
+{%- endfor %}
+{%- endmacro %}
+
+{%- macro toctree(docnames) -%}
+.. toctree::
+   :maxdepth: {{ maxdepth }}
+{% for docname in docnames %}
+   {{ docname }}
+{%- endfor %}
+{%- endmacro %}
+
+{%- if is_namespace %}
+{{- [pkgname, "namespace"] | join(" ") | e | heading }}
+{% else %}
+{{- [pkgname] | join(" ") | e | heading }}
+{% endif %}
+
+{%- if modulefirst and not is_namespace %}
+{{ automodule(pkgname, automodule_options) }}
+{% endif %}
+
+{%- if subpackages %}
+{{ toctree(subpackages) }}
+{% endif %}
+
+{%- if submodules %}
+{% if separatemodules %}
+{{ toctree(submodules) }}
+{% else %}
+{%- for submodule in submodules %}
+{% if show_headings %}
+{{- [submodule] | join(" ") | e | heading(2) }}
+{% endif %}
+{{ automodule(submodule, automodule_options) }}
+{% endfor %}
+{%- endif %}
+{%- endif %}
+
+{%- if not modulefirst and not is_namespace %}
+Module contents
+---------------
+
+{{ automodule(pkgname, automodule_options) }}
+{% endif %}
+{% endraw %}

kensho_kfinance-1.2.0/docs/templates/apidoc/toc.rst_t

@@ -0,0 +1,10 @@
+{% raw %}
+Index Tree
+==========
+
+.. toctree::
+   :maxdepth: {{ maxdepth }}
+{% for docname in docnames %}
+   {{ docname }}
+{%- endfor %}
+{% endraw %}

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/justfile

@@ -29,3 +29,9 @@ alias t := unit-test
 # Run unit tests. Use args for optional settings
 unit-test *args:
     python -m pytest {{args}}
+
+# Build the sphinx documents locally
+# First, copy the dependencies in docs/requirements.txt into pyproject.toml and install in a venv.
+# Don't merge changes to pyproject and docs/output into the remote repo!
+sphinx *args:
+    sphinx-build docs docs/output

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kensho_kfinance.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 1.1.0
+Version: 1.2.0
 Summary: Python CLI for kFinance
 Author-email: Luke Brown <luke.brown@kensho.com>, Michelle Keoy <michelle.keoy@kensho.com>, Keith Page <keith.page@kensho.com>, Matthew Rosen <matthew.rosen@kensho.com>, Nick Roshdieh <nick.roshdieh@kensho.com>
 Project-URL: source, https://github.com/kensho-technologies/kfinance
@@ -28,6 +28,7 @@ Requires-Dist: mypy<2,>=1.15.0; extra == "dev"
 Requires-Dist: pytest<7,>=6.1.2; extra == "dev"
 Requires-Dist: pytest-cov<7,>=6.0.0; extra == "dev"
 Requires-Dist: ruff<1,>=0.9.4; extra == "dev"
+Requires-Dist: requests_mock<1.2,>=1.1; extra == "dev"
 Dynamic: license-file
 
 # kFinance

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kensho_kfinance.egg-info/SOURCES.txt

@@ -1,5 +1,6 @@
 .coveragerc
 .gitignore
+.readthedocs.yaml
 AUTHORS.md
 CODE_OF_CONDUCT.md
 CONTRIBUTING.md
@@ -11,9 +12,13 @@ setup.py
 .github/workflows/ci-lint.yml
 .github/workflows/ci-test.yml
 .github/workflows/python-publish.yml
+docs/conf.py
 docs/index.rst
 docs/kfinance.rst
 docs/llm_tools.rst
+docs/requirements.txt
+docs/templates/apidoc/package.rst_t
+docs/templates/apidoc/toc.rst_t
 kensho_kfinance.egg-info/PKG-INFO
 kensho_kfinance.egg-info/SOURCES.txt
 kensho_kfinance.egg-info/dependency_links.txt
@@ -21,6 +26,7 @@ kensho_kfinance.egg-info/requires.txt
 kensho_kfinance.egg-info/top_level.txt
 kfinance/CHANGELOG.md
 kfinance/__init__.py
+kfinance/batch_request_handling.py
 kfinance/constants.py
 kfinance/fetch.py
 kfinance/kfinance.py
@@ -32,9 +38,9 @@ kfinance/server_thread.py
 kfinance/tool_schemas.py
 kfinance/version.py
 kfinance/tests/__init__.py
+kfinance/tests/test_batch_requests.py
 kfinance/tests/test_fetch.py
 kfinance/tests/test_objects.py
 scripts/copyright_line_check.sh
 scripts/lint.sh
-scripts/test.sh
-scripts/docs/make_rst.py
+scripts/test.sh

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kensho_kfinance.egg-info/requires.txt

@@ -15,3 +15,4 @@ mypy<2,>=1.15.0
 pytest<7,>=6.1.2
 pytest-cov<7,>=6.0.0
 ruff<1,>=0.9.4
+requests_mock<1.2,>=1.1

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kfinance/CHANGELOG.md

@@ -1,4 +1,7 @@
 # Changelog
+## v1.2.0
+- Add batch requests for iterable classes
+
 ## v1.1.0
 - Add market cap, TEV, and shares outstanding
 

kensho_kfinance-1.2.0/kfinance/batch_request_handling.py

@@ -0,0 +1,137 @@
+from concurrent.futures import ThreadPoolExecutor
+import functools
+from functools import cached_property
+import threading
+from typing import Any, Callable, Iterable, Protocol, Sized, Type, TypeVar
+
+from requests.exceptions import HTTPError
+
+from .fetch import KFinanceApiClient
+
+
+T = TypeVar("T")
+
+MAX_WORKERS_CAP: int = 10
+
+throttle = threading.Semaphore(MAX_WORKERS_CAP)
+
+
+def add_methods_of_singular_class_to_iterable_class(singular_cls: Type[T]) -> Callable:
+    """Returns a decorator that sets each method, property, and cached_property of"""
+    "[singular_cls] as an attribute of the decorated class."
+
+    class IterableKfinanceClass(Protocol, Sized, Iterable[T]):
+        """A protocol to represent a iterable Kfinance classes like Tickers and Companies.
+
+        Each of these classes has a kfinance_api_client attribute.
+        """
+
+        kfinance_api_client: KFinanceApiClient
+
+    def decorator(iterable_cls: Type[IterableKfinanceClass]) -> Type[IterableKfinanceClass]:
+        """Adds functions from a singular class to an iterable class.
+
+        This decorator modifies the [iterable_cls] so that when an attribute
+        (method, property, or cached property) added by the decorator is accessed,
+        it returns a dictionary. This dictionary maps each object in [iterable_cls]
+        to the result of invoking the attribute on that specific object.
+
+        For example, consider a `Company` class with a `city` property and a
+        `Companies` class that is an iterable of `Company` instances. When the
+        `Companies` class is decorated, it gains a `city` property. Accessing this
+        property will yield a dictionary where each key is a `Company` instance
+        and the corresponding value is the city of that instance. The resulting
+        dictionary might look like:
+
+            {<kfinance.kfinance.Company object>: 'Some City'}
+
+        Error Handling:
+            - If the result is a 404 HTTP error, the corresponding value
+            for that object in the dictionary will be set to None.
+            - For any other HTTP error, the error is raised and bubbles up.
+
+        Note:
+            This decorator requires [iterable_cls] to be an iterable of
+            instances of [singular_cls].
+        """
+
+        def process_in_thread_pool(
+            executor: ThreadPoolExecutor, method: Callable, obj: T, *args: Any, **kwargs: Any
+        ) -> Any:
+            with throttle:
+                future = executor.submit(method, obj, *args, **kwargs)
+                try:
+                    return future.result()
+                except HTTPError as http_err:
+                    error_code = http_err.response.status_code
+                    if error_code == 404:
+                        return None
+                    else:
+                        raise http_err
+
+        def process_in_batch(
+            method: Callable, self: IterableKfinanceClass, *args: Any, **kwargs: Any
+        ) -> dict:
+            results = {}
+            kfinance_api_client = self.kfinance_api_client
+            with kfinance_api_client.batch_request_header(batch_size=len(self)):
+                with kfinance_api_client.thread_pool as executor:
+                    results = dict(
+                        zip(
+                            self,
+                            [
+                                process_in_thread_pool(executor, method, obj, *args, **kwargs)
+                                for obj in self
+                            ],
+                        )
+                    )
+
+            return results
+
+        for method_name in dir(singular_cls):
+            method = getattr(singular_cls, method_name)
+            if method_name.startswith("__") or method_name.startswith("set_"):
+                continue
+            if callable(method):
+
+                def create_method_wrapper(method: Callable) -> Callable:
+                    @functools.wraps(method)
+                    def method_wrapper(
+                        self: IterableKfinanceClass, *args: Any, **kwargs: Any
+                    ) -> dict:
+                        return process_in_batch(method, self, *args, **kwargs)
+
+                    return method_wrapper
+
+                setattr(iterable_cls, method_name, create_method_wrapper(method))
+
+            elif isinstance(method, property):
+
+                def create_prop_wrapper(method: property) -> Callable:
+                    assert method.fget is not None
+
+                    @functools.wraps(method.fget)
+                    def prop_wrapper(self: IterableKfinanceClass) -> Any:
+                        assert method.fget is not None
+                        return process_in_batch(method.fget, self)
+
+                    return prop_wrapper
+
+                setattr(iterable_cls, method_name, property(create_prop_wrapper(method)))
+
+            elif isinstance(method, cached_property):
+
+                def create_cached_prop_wrapper(method: cached_property) -> cached_property:
+                    @functools.wraps(method.func)
+                    def cached_prop_wrapper(self: IterableKfinanceClass) -> Any:
+                        return process_in_batch(method.func, self)
+
+                    wrapped_cached_property = cached_property(cached_prop_wrapper)
+                    wrapped_cached_property.__set_name__(iterable_cls, method_name)
+                    return wrapped_cached_property
+
+                setattr(iterable_cls, method_name, create_cached_prop_wrapper(method))
+
+        return iterable_cls
+
+    return decorator

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kfinance/fetch.py

@@ -1,5 +1,8 @@
+from concurrent.futures import ThreadPoolExecutor
+from contextlib import contextmanager
 from time import time
-from typing import Callable, Optional
+from typing import Callable, Generator, Optional
+from uuid import uuid4
 
 import jwt
 import requests
@@ -19,6 +22,7 @@ DEFAULT_API_HOST: str = "https://kfinance.kensho.com"
 DEFAULT_API_VERSION: int = 1
 DEFAULT_OKTA_HOST: str = "https://kensho.okta.com"
 DEFAULT_OKTA_AUTH_SERVER: str = "default"
+DEFAULT_MAX_WORKERS: int = 10
 
 
 class KFinanceApiClient:
@@ -27,12 +31,33 @@ class KFinanceApiClient:
         refresh_token: Optional[str] = None,
         client_id: Optional[str] = None,
         private_key: Optional[str] = None,
+        thread_pool: Optional[ThreadPoolExecutor] = None,
         api_host: str = DEFAULT_API_HOST,
         api_version: int = DEFAULT_API_VERSION,
         okta_host: str = DEFAULT_OKTA_HOST,
         okta_auth_server: str = DEFAULT_OKTA_AUTH_SERVER,
     ):
-        """Configuration of KFinance Client."""
+        """Configuration of KFinance Client.
+
+        :param refresh_token: users refresh token
+        :type refresh_token: str, Optional
+        :param client_id: users client id will be provided by support@kensho.com
+        :type client_id: str, Optional
+        :param private_key: users private key that corresponds to the registered public sent to support@kensho.com
+        :type private_key: str, Optional
+        :param thread_pool: the thread pool used to execute batch requests. The number of concurrent requests is
+        capped at 10. If no thread pool is provided, a thread pool with 10 max workers will be created when batch
+        requests are made.
+        :type thread_pool: ThreadPoolExecutor, Optional
+        :param api_host: the api host URL
+        :type api_host: str
+        :param api_version: the api version number
+        :type api_version: int
+        :param okta_host: the okta host URL
+        :type okta_host: str
+        :param okta_auth_server: the okta route for authentication
+        :type okta_auth_server: str
+        """
         if refresh_token is not None:
             self.refresh_token = refresh_token
             self._access_token_refresh_func: Callable[..., str] = (
@@ -48,10 +73,40 @@ class KFinanceApiClient:
         self.api_version = api_version
         self.okta_host = okta_host
         self.okta_auth_server = okta_auth_server
+        self._thread_pool = thread_pool
         self.url_base = f"{self.api_host}/api/v{self.api_version}/"
         self._access_token_expiry = 0
         self._access_token: str | None = None
         self.user_agent_source = "object_oriented"
+        self._batch_id: str | None = None
+        self._batch_size: str | None = None
+
+    @contextmanager
+    def batch_request_header(self, batch_size: int) -> Generator:
+        """Set batch id and batch size for batch request request headers"""
+        batch_id = str(uuid4())
+
+        self._batch_id = batch_id
+        self._batch_size = str(batch_size)
+
+        try:
+            yield
+        finally:
+            self._batch_id = None
+            self._batch_size = None
+
+    @property
+    def thread_pool(self) -> ThreadPoolExecutor:
+        """Returns the thread pool used to execute batch requests.
+
+        If the thread pool is not set, a thread pool with 10 max workers will be created
+         and returned.
+        """
+
+        if self._thread_pool is None:
+            self._thread_pool = ThreadPoolExecutor(max_workers=DEFAULT_MAX_WORKERS)
+
+        return self._thread_pool
 
     @property
     def access_token(self) -> str:
@@ -110,13 +165,21 @@ class KFinanceApiClient:
 
     def fetch(self, url: str) -> dict:
         """Does the request and auth"""
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.access_token}",
+            "User-Agent": f"kfinance/{kfinance_version} {self.user_agent_source}",
+        }
+        if self._batch_id is not None:
+            assert self._batch_size is not None
+            headers.update(
+                {"Kfinance-Batch-Id": self._batch_id, "Kfinance-Batch-Size": self._batch_size}
+            )
+
         response = requests.get(
             url,
-            headers={
-                "Content-Type": "application/json",
-                "Authorization": f"Bearer {self.access_token}",
-                "User-Agent": f"kfinance/{kfinance_version} {self.user_agent_source}",
-            },
+            headers=headers,
             timeout=60,
         )
         response.raise_for_status()

{kensho_kfinance-1.1.0 → kensho_kfinance-1.2.0}/kfinance/kfinance.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from concurrent.futures import ThreadPoolExecutor
 from datetime import date, datetime, timezone
 from functools import cached_property
 from io import BytesIO
@@ -14,6 +15,7 @@ import numpy as np
 import pandas as pd
 from PIL.Image import Image, open as image_open
 
+from .batch_request_handling import add_methods_of_singular_class_to_iterable_class
 from .constants import HistoryMetadata, IdentificationTriple, LatestPeriods, YearAndQuarter
 from .fetch import (
     DEFAULT_API_HOST,
@@ -31,7 +33,10 @@ from .llm_tools import (
     langchain_tools,
     openai_tool_descriptions,
 )
-from .meta_classes import CompanyFunctionsMetaClass, DelegatedCompanyFunctionsMetaClass
+from .meta_classes import (
+    CompanyFunctionsMetaClass,
+    DelegatedCompanyFunctionsMetaClass,
+)
 from .prompt import PROMPT
 from .server_thread import ServerThread
 
@@ -917,6 +922,7 @@ class BusinessRelationships(NamedTuple):
         return f"{type(self).__module__}.{type(self).__qualname__} of {str(dictionary)}"
 
 
+@add_methods_of_singular_class_to_iterable_class(Company)
 class Companies(set):
     """Base class for representing a set of Companies"""
 
@@ -928,9 +934,11 @@ class Companies(set):
         :param company_ids: An iterable of S&P CIQ Company ids
         :type company_ids: Iterable[int]
         """
+        self.kfinance_api_client = kfinance_api_client
         super().__init__(Company(kfinance_api_client, company_id) for company_id in company_ids)
 
 
+@add_methods_of_singular_class_to_iterable_class(Security)
 class Securities(set):
     """Base class for representing a set of Securities"""
 
@@ -945,6 +953,7 @@ class Securities(set):
         super().__init__(Security(kfinance_api_client, security_id) for security_id in security_ids)
 
 
+@add_methods_of_singular_class_to_iterable_class(TradingItem)
 class TradingItems(set):
     """Base class for representing a set of Trading Items"""
 
@@ -958,14 +967,16 @@ class TradingItems(set):
         :param company_ids: An iterable of S&P CIQ Company ids
         :type company_ids: Iterable[int]
         """
+        self.kfinance_api_client = kfinance_api_client
         super().__init__(
             TradingItem(kfinance_api_client, trading_item_id)
             for trading_item_id in trading_item_ids
         )
 
 
+@add_methods_of_singular_class_to_iterable_class(Ticker)
 class Tickers(set):
-    """Base TickerSet class for representing a set of Tickers"""
+    """Base class for representing a set of Tickers"""
 
     def __init__(
         self,
@@ -1041,6 +1052,7 @@ class Client:
         refresh_token: Optional[str] = None,
         client_id: Optional[str] = None,
         private_key: Optional[str] = None,
+        thread_pool: Optional[ThreadPoolExecutor] = None,
         api_host: str = DEFAULT_API_HOST,
         api_version: int = DEFAULT_API_VERSION,
         okta_host: str = DEFAULT_OKTA_HOST,
@@ -1054,6 +1066,10 @@ class Client:
         :type client_id: str, Optional
         :param private_key: users private key that corresponds to the registered public sent to support@kensho.com
         :type private_key: str, Optional
+        :param thread_pool: the thread pool used to execute batch requests. The number of concurrent requests is
+        capped at 10. If no thread pool is provided, a thread pool with 10 max workers will be created when batch
+        requests are made.
+        :type thread_pool: ThreadPoolExecutor, Optional
         :param api_host: the api host URL
         :type api_host: str
         :param api_version: the api version number
@@ -1071,6 +1087,7 @@ class Client:
                 api_host=api_host,
                 api_version=api_version,
                 okta_host=okta_host,
+                thread_pool=thread_pool,
             )
         # method 2 keypair
         elif client_id is not None and private_key is not None:
@@ -1081,6 +1098,7 @@ class Client:
                 api_version=api_version,
                 okta_host=okta_host,
                 okta_auth_server=okta_auth_server,
+                thread_pool=thread_pool,
             )
         # method 3 automatic login getting a refresh token
         else:
@@ -1099,6 +1117,7 @@ class Client:
                 api_host=api_host,
                 api_version=api_version,
                 okta_host=okta_host,
+                thread_pool=thread_pool,
             )
             stdout.write("Login credentials received.\n")