kensho-kfinance 2.6.5__tar.gz → 2.8.0__tar.gz

{kensho_kfinance-2.6.5/kensho_kfinance.egg-info → kensho_kfinance-2.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 2.6.5
+Version: 2.8.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

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/example_notebooks/basic_usage.ipynb

@@ -98,7 +98,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from kfinance.constants import PeriodType\n",
+    "from kfinance.models.date_and_period_models import PeriodType\n",
     "\n",
     "# You can create a Ticker object with client.ticker.\n",
     "# A Ticker has access to a wide variety of attributes, including prices, \n",
@@ -220,7 +220,7 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "from kfinance.constants import Periodicity\n",
+    "from kfinance.models.date_and_period_models import Periodicity\n",
     "\n",
     "# To fetch recent spgi prices, you can use the `history` function.\n",
     "# Without any further configuration `history()` will return adjusted\n",

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0/kensho_kfinance.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 2.6.5
+Version: 2.8.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

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/kensho_kfinance.egg-info/SOURCES.txt

@@ -36,7 +36,7 @@ kensho_kfinance.egg-info/top_level.txt
 kfinance/CHANGELOG.md
 kfinance/__init__.py
 kfinance/batch_request_handling.py
-kfinance/constants.py
+kfinance/decimal_with_unit.py
 kfinance/fetch.py
 kfinance/kfinance.py
 kfinance/mcp.py
@@ -46,16 +46,32 @@ kfinance/py.typed
 kfinance/pydantic_models.py
 kfinance/server_thread.py
 kfinance/version.py
+kfinance/models/__init__.py
+kfinance/models/business_relationship_models.py
+kfinance/models/capitalization_models.py
+kfinance/models/competitor_models.py
+kfinance/models/currency_models.py
+kfinance/models/date_and_period_models.py
+kfinance/models/id_models.py
+kfinance/models/industry_models.py
+kfinance/models/line_item_models.py
+kfinance/models/permission_models.py
+kfinance/models/price_models.py
+kfinance/models/segment_models.py
+kfinance/models/statement_models.py
 kfinance/tests/__init__.py
 kfinance/tests/conftest.py
-kfinance/tests/scratch.py
 kfinance/tests/test_batch_requests.py
 kfinance/tests/test_client.py
+kfinance/tests/test_decimal_with_unit.py
 kfinance/tests/test_example_notebook.py
 kfinance/tests/test_fetch.py
 kfinance/tests/test_group_objects.py
 kfinance/tests/test_objects.py
 kfinance/tests/test_tools.py
+kfinance/tests/test_models/__init__.py
+kfinance/tests/test_models/test_capitalization_models.py
+kfinance/tests/test_models/test_price_models.py
 kfinance/tool_calling/README.md
 kfinance/tool_calling/__init__.py
 kfinance/tool_calling/get_advisors_for_company_in_transaction_from_identifier.py

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/kfinance/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## v2.8.0
+- Add currency to get_capitalization and get_prices tools.
+
+## v2.7.0
+ - Expose a new list of LLM tools `grounding_tools` that return a list of all endpoint urls called
+
 ## v2.6.5
  - Add competitor company_name
 

kensho_kfinance-2.8.0/kfinance/decimal_with_unit.py

@@ -0,0 +1,78 @@
+from copy import deepcopy
+from decimal import Decimal
+from typing import Any
+
+from pydantic import BaseModel, Field, model_validator
+from typing_extensions import Self
+
+from kfinance.models.currency_models import ISO_CODE_TO_CURRENCY
+
+
+class DecimalWithUnit(BaseModel):
+    """DecimalWithUnit (DWU) represents a decimal with a corresponding unit like $100 or 20 shares.
+
+    In addition to a value and unit, each DWU has a `conventional_decimals` attribute,
+    which indicates the number of decimals that should be represented.
+    For example, for USD, conventional_decimals is 2, which will display as "1.00".
+    For shares, conventional_decimals is 0, which will display as "1"
+
+    Usually, rather than initializing a DWU directly, you'll likely want to use an
+    existing subclass like `Money` or `Shares` or create a new one.
+    """
+
+    value: Decimal
+    unit: str
+    # exclude conventional_decimals from serialization
+    conventional_decimals: int = Field(exclude=True)
+
+    @model_validator(mode="after")
+    def quantize_value(self) -> Self:
+        """Quantize the value at the end of the deserialization.
+
+        The value gets adjusted so that it always has the expected number of decimals defined in
+        conventional_decimals.
+        For USD with conventional_decimals=2, it will show values like "1.00"
+        For Shares with conventional_decimals=0, it will show values like "1"
+        """
+        exponent = Decimal("10") ** Decimal(-self.conventional_decimals)
+        self.value = self.value.quantize(exp=exponent)
+        return self
+
+
+class Money(DecimalWithUnit):
+    @model_validator(mode="before")
+    @classmethod
+    def inject_conventional_decimals_into_data(cls, data: Any) -> Any:
+        """Inject conventional_decimals into data dict.
+
+        Each currency has an associated conventional_decimals defined in the
+        CURRENCIES list. This validator fetches that number and injects it into the
+        data dict.
+        """
+
+        if isinstance(data, dict) and "conventional_decimals" not in data:
+            data = deepcopy(data)
+            currency = ISO_CODE_TO_CURRENCY[data["unit"]]
+            data["conventional_decimals"] = currency.conventional_decimals
+
+        return data
+
+
+class Shares(DecimalWithUnit):
+    unit: str = "Shares"
+    conventional_decimals: int = Field(exclude=True, default=0)
+
+    @model_validator(mode="before")
+    @classmethod
+    def convert_numbers_to_dicts(cls, data: Any) -> Any:
+        """Convert numbers into dicts.
+
+        The shares class can be built from a single number because unit and
+        conventional_decimals are always the same. However, the parser expects a
+        dict instead of a number, so we have to convert any number ("10") into a
+        dict {"value": "10"}.
+        """
+
+        if isinstance(data, (str, int, float, Decimal)):
+            data = {"value": data}
+        return data

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/kfinance/fetch.py

@@ -1,6 +1,7 @@
 from concurrent.futures import ThreadPoolExecutor
 from contextlib import contextmanager
 import logging
+from queue import Queue
 from time import time
 from typing import Callable, Generator, Optional
 from uuid import uuid4
@@ -9,23 +10,22 @@ import jwt
 from pydantic import ValidationError
 import requests
 
-from .constants import (
-    BusinessRelationshipType,
-    CompetitorSource,
-    IdentificationTriple,
-    IndustryClassification,
-    Periodicity,
-    PeriodType,
-    Permission,
-    SegmentType,
-)
-from .pydantic_models import RelationshipResponse, RelationshipResponseNoName
+from kfinance.models.business_relationship_models import BusinessRelationshipType
+from kfinance.models.capitalization_models import Capitalizations
+from kfinance.models.competitor_models import CompetitorSource
+from kfinance.models.date_and_period_models import Periodicity, PeriodType
+from kfinance.models.id_models import IdentificationTriple
+from kfinance.models.industry_models import IndustryClassification
+from kfinance.models.permission_models import Permission
+from kfinance.models.price_models import PriceHistory
+from kfinance.models.segment_models import SegmentType
+from kfinance.pydantic_models import RelationshipResponse, RelationshipResponseNoName
 
 
 # version.py gets autogenerated by setuptools-scm and is not available
 # during local development.
 try:
-    from .version import __version__ as kfinance_version
+    from kfinance.version import __version__ as kfinance_version
 except ImportError:
     kfinance_version = "dev"
 
@@ -95,6 +95,7 @@ class KFinanceApiClient:
         self._batch_id: str | None = None
         self._batch_size: str | None = None
         self._user_permissions: set[Permission] | None = None
+        self._endpoint_tracker_queue: Queue[str] | None = None
 
     @contextmanager
     def batch_request_header(self, batch_size: int) -> Generator:
@@ -208,9 +209,28 @@ class KFinanceApiClient:
                     permission_str,
                 )
 
+    @contextmanager
+    def endpoint_tracker(self) -> Generator:
+        """Context manager to track and return endpoint URLs in our thread-safe queue during execution.
+
+        endpoint_tracker yields a queue into which all endpoint URLs are written until the context manager gets exited.
+        It is up to the callers to dequeue the queue before the context manager gets exited and the queue gets wiped.
+        This functionality is currently used by `run_with_grounding` to collect and forward endpoint URLs.
+        """
+        self._endpoint_tracker_queue = Queue[str]()
+
+        try:
+            yield self._endpoint_tracker_queue
+        finally:
+            self._endpoint_tracker_queue = None
+
     def fetch(self, url: str) -> dict:
         """Does the request and auth"""
 
+        # _endpoint_tracker_queue will only be initialized if inside the endpoint_tracker context manager
+        if self._endpoint_tracker_queue:
+            self._endpoint_tracker_queue.put(url)
+
         headers = {
             "Content-Type": "application/json",
             "Authorization": f"Bearer {self.access_token}",
@@ -279,7 +299,7 @@ class KFinanceApiClient:
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
         periodicity: Optional[Periodicity] = None,
-    ) -> dict:
+    ) -> PriceHistory:
         """Get the pricing history."""
         url = (
             f"{self.url_base}pricing/{trading_item_id}/"
@@ -288,7 +308,7 @@ class KFinanceApiClient:
             f"{periodicity if periodicity else 'none'}/"
             f"{'adjusted' if is_adjusted else 'unadjusted'}"
         )
-        return self.fetch(url)
+        return PriceHistory.model_validate(self.fetch(url))
 
     def fetch_history_metadata(self, trading_item_id: int) -> dict[str, str]:
         """Get the pricing history metadata."""
@@ -300,14 +320,14 @@ class KFinanceApiClient:
         company_id: int,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> dict:
+    ) -> Capitalizations:
         """Get the market cap, TEV, and shares outstanding for a company."""
         url = (
             f"{self.url_base}market_cap/{company_id}/"
             f"{start_date if start_date is not None else 'none'}/"
             f"{end_date if end_date is not None else 'none'}"
         )
-        return self.fetch(url)
+        return Capitalizations.model_validate(self.fetch(url))
 
     def fetch_segments(
         self,

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/kfinance/kfinance.py

@@ -15,33 +15,28 @@ import webbrowser
 import google.ai.generativelanguage_v1beta.types as gapic
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from langchain_google_genai._function_utils import convert_to_genai_function_declarations
-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,
-    IndustryClassification,
-    LatestPeriods,
-    Periodicity,
-    YearAndQuarter,
-)
-from .fetch import (
+from kfinance.batch_request_handling import add_methods_of_singular_class_to_iterable_class
+from kfinance.fetch import (
     DEFAULT_API_HOST,
     DEFAULT_API_VERSION,
     DEFAULT_OKTA_AUTH_SERVER,
     DEFAULT_OKTA_HOST,
     KFinanceApiClient,
 )
-from .meta_classes import (
+from kfinance.meta_classes import (
     CompanyFunctionsMetaClass,
     DelegatedCompanyFunctionsMetaClass,
 )
-from .prompt import PROMPT
-from .pydantic_models import TranscriptComponent
-from .server_thread import ServerThread
+from kfinance.models.date_and_period_models import LatestPeriods, Periodicity, YearAndQuarter
+from kfinance.models.id_models import IdentificationTriple
+from kfinance.models.industry_models import IndustryClassification
+from kfinance.models.price_models import HistoryMetadata, PriceHistory
+from kfinance.prompt import PROMPT
+from kfinance.pydantic_models import TranscriptComponent
+from kfinance.server_thread import ServerThread
 
 
 if TYPE_CHECKING:
@@ -140,36 +135,29 @@ class TradingItem:
         adjusted: bool = True,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> pd.DataFrame:
+    ) -> PriceHistory:
         """Retrieves the historical price data for a given asset over a specified date range.
 
         :param periodicity: Determines the frequency of the historical data returned. Defaults to Periodicity.day.
         :param Optional[bool] adjusted: Whether to retrieve adjusted prices that account for corporate actions such as dividends and splits, it defaults True
         :param Optional[str] start_date: The start date for historical price retrieval in format "YYYY-MM-DD", default to None
         :param Optional[str] end_date: The end date for historical price retrieval in format "YYYY-MM-DD", default to None
-        :return: A pd.DataFrame containing historical price data with columns corresponding to the specified periodicity, with Date as the index, and columns "open", "high", "low", "close", "volume" in type decimal. The Date index is a string that depends on the periodicity. If Periodicity.day, the Date index is the day in format "YYYY-MM-DD", eg "2024-05-13" If Periodicity.week, the Date index is the week number of the year in format "YYYY Week ##", eg "2024 Week 2" If Periodicity.month, the Date index is the month name of the year in format "<Month> YYYY", eg "January 2024". If Periodicity.year, the Date index is the year in format "YYYY", eg "2024".
-        :rtype: pd.DataFrame
+        :return: A PriceHistory containing historical price data including "open", "high", "low", "close", "volume" in type Money. The date value is a string that depends on the periodicity. If Periodicity.day, the Date index is the day in format "YYYY-MM-DD", eg "2024-05-13" If Periodicity.week, the Date index is the week number of the year in format "YYYY Week ##", eg "2024 Week 2" If Periodicity.month, the Date index is the month name of the year in format "<Month> YYYY", eg "January 2024". If Periodicity.year, the Date index is the year in format "YYYY", eg "2024".
+        :rtype: PriceHistory
         """
         if start_date and end_date:
             if (
                 datetime.strptime(start_date, "%Y-%m-%d").date()
                 > datetime.strptime(end_date, "%Y-%m-%d").date()
             ):
-                return pd.DataFrame()
+                return PriceHistory(prices=[])
 
-        return (
-            pd.DataFrame(
-                self.kfinance_api_client.fetch_history(
-                    trading_item_id=self.trading_item_id,
-                    is_adjusted=adjusted,
-                    start_date=start_date,
-                    end_date=end_date,
-                    periodicity=periodicity,
-                )["prices"]
-            )
-            .set_index("date")
-            .apply(pd.to_numeric)
-            .replace(np.nan, None)
+        return self.kfinance_api_client.fetch_history(
+            trading_item_id=self.trading_item_id,
+            is_adjusted=adjusted,
+            start_date=start_date,
+            end_date=end_date,
+            periodicity=periodicity,
         )
 
     def price_chart(
@@ -815,7 +803,11 @@ class Security:
                 "trading_items"
             ]
             self._trading_items = TradingItems(
-                kfinance_api_client=self.kfinance_api_client, trading_item_ids=trading_item_ids
+                kfinance_api_client=self.kfinance_api_client,
+                trading_items=[
+                    TradingItem(kfinance_api_client=self.kfinance_api_client, trading_item_id=tii)
+                    for tii in trading_item_ids
+                ],
             )
         return self._trading_items
 
@@ -1194,7 +1186,7 @@ class Ticker(DelegatedCompanyFunctionsMetaClass):
         adjusted: bool = True,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> pd.DataFrame:
+    ) -> PriceHistory:
         """Retrieves the historical price data for a given asset over a specified date range.
 
         :param periodicity: Determines the frequency of the historical data returned. Defaults to Periodicity.day.
@@ -1205,8 +1197,8 @@ class Ticker(DelegatedCompanyFunctionsMetaClass):
         :type start_date: str, optional
         :param end_date: The end date for historical price retrieval in format "YYYY-MM-DD", default to None
         :type end_date: str, optional
-        :return: A pd.DataFrame containing historical price data with columns corresponding to the specified periodicity, with Date as the index, and columns "open", "high", "low", "close", "volume" in type decimal. The Date index is a string that depends on the periodicity. If Periodicity.day, the Date index is the day in format "YYYY-MM-DD", eg "2024-05-13" If Periodicity.week, the Date index is the week number of the year in format "YYYY Week ##", eg "2024 Week 2" If Periodicity.month, the Date index is the month name of the year in format "<Month> YYYY", eg "January 2024". If Periodicity.year, the Date index is the year in format "YYYY", eg "2024".
-        :rtype: pd.DataFrame
+        :return: A PriceHistory containing historical price data including "open", "high", "low", "close", "volume" in type Money. The date value is a string that depends on the periodicity. If Periodicity.day, the Date index is the day in format "YYYY-MM-DD", eg "2024-05-13" If Periodicity.week, the Date index is the week number of the year in format "YYYY Week ##", eg "2024 Week 2" If Periodicity.month, the Date index is the month name of the year in format "<Month> YYYY", eg "January 2024". If Periodicity.year, the Date index is the year in format "YYYY", eg "2024".
+        :rtype: PriceHistory
         """
         return self.primary_trading_item.history(
             periodicity,
@@ -1425,20 +1417,17 @@ class TradingItems(set):
     """Base class for representing a set of Trading Items"""
 
     def __init__(
-        self, kfinance_api_client: KFinanceApiClient, trading_item_ids: Iterable[int]
+        self, kfinance_api_client: KFinanceApiClient, trading_items: Iterable[TradingItem]
     ) -> None:
         """Initialize the Trading Items
 
         :param kfinance_api_client: The KFinanceApiClient used to fetch data
         :type kfinance_api_client: KFinanceApiClient
-        :param company_ids: An iterable of S&P CIQ Company ids
-        :type company_ids: Iterable[int]
+        :param trading_items: An iterable of TradingItem
+        :type trading_items: Iterable[TradingItem]
         """
         self.kfinance_api_client = kfinance_api_client
-        super().__init__(
-            TradingItem(kfinance_api_client, trading_item_id)
-            for trading_item_id in trading_item_ids
-        )
+        super().__init__(trading_items)
 
 
 @add_methods_of_singular_class_to_iterable_class(Ticker)
@@ -1515,7 +1504,14 @@ class Tickers(set):
         :rtype: TradingItems
         """
         return TradingItems(
-            self.kfinance_api_client, (ticker.trading_item_id for ticker in self.__iter__())
+            self.kfinance_api_client,
+            [
+                TradingItem(
+                    kfinance_api_client=self.kfinance_api_client,
+                    trading_item_id=ticker.trading_item_id,
+                )
+                for ticker in self.__iter__()
+            ],
         )
 
 
@@ -1663,6 +1659,11 @@ class Client:
         """
         return {t.name: t.run_without_langchain for t in self.langchain_tools}
 
+    @property
+    def grounding_tools(self) -> dict[str, Callable]:
+        """Return a mapping of tool calling function names to the corresponding functions for the grounding agent."""
+        return {t.name: t.run_with_grounding for t in self.langchain_tools}
+
     @property
     def anthropic_tool_descriptions(self) -> list[dict[str, Any]]:
         """Return tool descriptions for anthropic"""

{kensho_kfinance-2.6.5 → kensho_kfinance-2.8.0}/kfinance/meta_classes.py

@@ -1,20 +1,19 @@
 from abc import abstractmethod
 from datetime import datetime
 import logging
-from typing import TYPE_CHECKING, Any, Callable, Literal, Optional
+from typing import TYPE_CHECKING, Any, Callable, Optional
 
 from cachetools import LRUCache, cached
 import numpy as np
 import pandas as pd
 
-from .constants import (
-    LINE_ITEMS,
-    BusinessRelationshipType,
-    CompetitorSource,
-    PeriodType,
-    SegmentType,
-)
 from .fetch import KFinanceApiClient
+from .models.business_relationship_models import BusinessRelationshipType
+from .models.capitalization_models import Capitalization
+from .models.competitor_models import CompetitorSource
+from .models.date_and_period_models import PeriodType
+from .models.line_item_models import LINE_ITEMS
+from .models.segment_models import SegmentType
 from .pydantic_models import RelationshipResponse
 
 
@@ -262,73 +261,75 @@ class CompanyFunctionsMetaClass:
         self,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> pd.DataFrame:
+    ) -> dict:
         """Retrieves market caps for a company between start and end date.
 
         :param start_date: The start date in format "YYYY-MM-DD", default to None
         :type start_date: str, optional
         :param end_date: The end date in format "YYYY-MM-DD", default to None
         :type end_date: str, optional
-        :return: A DataFrame with a `market_cap` column. The dates are the index.
-        :rtype: pd.DataFrame
+        :return: A dict with market_cap
+        :rtype: dict
         """
 
         return self._fetch_market_cap_tev_or_shares_outstanding(
-            column_to_extract="market_cap", start_date=start_date, end_date=end_date
+            capitalization_to_extract=Capitalization.market_cap,
+            start_date=start_date,
+            end_date=end_date,
         )
 
     def tev(
         self,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> pd.DataFrame:
+    ) -> dict:
         """Retrieves TEV (total enterprise value) for a company between start and end date.
 
         :param start_date: The start date in format "YYYY-MM-DD", default to None
         :type start_date: str, optional
         :param end_date: The end date in format "YYYY-MM-DD", default to None
         :type end_date: str, optional
-        :return: A DataFrame with a `tev` column. The dates are the index.
-        :rtype: pd.DataFrame
+        :return: A dict with TEV
+        :rtype: dict
         """
 
         return self._fetch_market_cap_tev_or_shares_outstanding(
-            column_to_extract="tev", start_date=start_date, end_date=end_date
+            capitalization_to_extract=Capitalization.tev, start_date=start_date, end_date=end_date
         )
 
     def shares_outstanding(
         self,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> pd.DataFrame:
+    ) -> dict:
         """Retrieves shares outstanding for a company between start and end date.
 
         :param start_date: The start date in format "YYYY-MM-DD", default to None
         :type start_date: str, optional
         :param end_date: The end date in format "YYYY-MM-DD", default to None
         :type end_date: str, optional
-        :return: A DataFrame with a `shares_outstanding` column. The dates are the index.
-        :rtype: pd.DataFrame
+        :return: A dict with outstanding shares
+        :rtype: dict
         """
 
         return self._fetch_market_cap_tev_or_shares_outstanding(
-            column_to_extract="shares_outstanding", start_date=start_date, end_date=end_date
+            capitalization_to_extract=Capitalization.shares_outstanding,
+            start_date=start_date,
+            end_date=end_date,
         )
 
     def _fetch_market_cap_tev_or_shares_outstanding(
         self,
-        column_to_extract: Literal["market_cap", "tev", "shares_outstanding"],
+        capitalization_to_extract: Capitalization,
         start_date: str | None,
         end_date: str | None,
-    ) -> pd.DataFrame:
+    ) -> dict:
         """Helper function to fetch market cap, TEV, and shares outstanding."""
 
-        df = pd.DataFrame(
-            self.kfinance_api_client.fetch_market_caps_tevs_and_shares_outstanding(
-                company_id=self.company_id, start_date=start_date, end_date=end_date
-            )["market_caps"]
+        capitalizations = self.kfinance_api_client.fetch_market_caps_tevs_and_shares_outstanding(
+            company_id=self.company_id, start_date=start_date, end_date=end_date
         )
-        return df.set_index("date")[[column_to_extract]].apply(pd.to_numeric).replace(np.nan, None)
+        return capitalizations.jsonify_single_attribute(capitalization_to_extract)
 
     def _segments(
         self,

kensho_kfinance-2.8.0/kfinance/models/business_relationship_models.py

@@ -0,0 +1,26 @@
+from strenum import StrEnum
+
+
+class BusinessRelationshipType(StrEnum):
+    """The type of business relationship"""
+
+    supplier = "supplier"
+    customer = "customer"
+    distributor = "distributor"
+    franchisor = "franchisor"
+    franchisee = "franchisee"
+    landlord = "landlord"
+    tenant = "tenant"
+    licensor = "licensor"
+    licensee = "licensee"
+    creditor = "creditor"
+    borrower = "borrower"
+    lessor = "lessor"
+    lessee = "lessee"
+    strategic_alliance = "strategic_alliance"
+    investor_relations_firm = "investor_relations_firm"
+    investor_relations_client = "investor_relations_client"
+    transfer_agent = "transfer_agent"
+    transfer_agent_client = "transfer_agent_client"
+    vendor = "vendor"
+    client_services = "client_services"