kensho-kfinance 2.9.0__tar.gz → 3.0.0__tar.gz

{kensho_kfinance-2.9.0/kensho_kfinance.egg-info → kensho_kfinance-3.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 2.9.0
+Version: 3.0.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.9.0 → kensho_kfinance-3.0.0}/docs/build_tool_calling_documentation.py

@@ -5,8 +5,9 @@ from pathlib import Path
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from pydantic._internal._repr import display_as_type
 
-from kfinance.kfinance import Client
-from kfinance.tool_calling import ALL_TOOLS, KfinanceTool
+from kfinance.client.kfinance import Client
+from kfinance.integrations.tool_calling.all_tools import ALL_TOOLS
+from kfinance.integrations.tool_calling.tool_calling_models import KfinanceTool
 
 
 def add_tool_calling_docs_for_all_tools() -> None:

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/basic_usage.ipynb

@@ -58,7 +58,8 @@
    "outputs": [],
    "source": [
     "# import the kfinance client\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
+    "\n",
     "# import standard libraries\n",
     "import functools\n",
     "import types\n",
@@ -98,8 +99,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "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",
     "# market caps, statements, and line items.\n",
@@ -150,6 +149,8 @@
    "execution_count": null,
    "outputs": [],
    "source": [
+    "from kfinance.client.models.date_and_period_models import PeriodType\n",
+    "\n",
     "# However, it's also possible to set parameters to fetch, \n",
     "# for example, annual balance sheets from the 2010s.\n",
     "spgi.balance_sheet(period_type=PeriodType.annual, start_year=2010, end_year=2019)"
@@ -220,7 +221,7 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "from kfinance.models.date_and_period_models import Periodicity\n",
+    "from kfinance.client.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.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/langchain/anthropic_langchain_tool_calling.ipynb

@@ -83,7 +83,7 @@
    "source": [
     "# import the kfinance client\n",
     "import sys\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "\n",
     "# check if the current environment is a Google Colab\n",
     "try:\n",
@@ -130,7 +130,7 @@
    "outputs": [],
    "source": [
     "from pydantic import SecretStr\n",
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "from langchain.agents import AgentExecutor, create_tool_calling_agent\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
     "from langchain_anthropic import ChatAnthropic\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/langchain/google_gemini_langchain_tool_calling.ipynb

@@ -84,7 +84,7 @@
    "source": [
     "# import the kfinance client\n",
     "import sys\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "# check if the current environment is a Google Colab\n",
     "try:\n",
     "    import google.colab\n",
@@ -128,7 +128,7 @@
    },
    "outputs": [],
    "source": [
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "# import langchain\n",
     "from langchain.agents import AgentExecutor, create_tool_calling_agent\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/langchain/openai_langchain_tool_calling.ipynb

@@ -84,7 +84,7 @@
    "source": [
     "# import the kfinance client\n",
     "import sys\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "# check if the current environment is a Google Colab\n",
     "try:\n",
     "    import google.colab\n",
@@ -125,7 +125,8 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
+    "\n",
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "from pydantic import SecretStr\n",
     "from langchain.agents import AgentExecutor, create_tool_calling_agent\n",
     "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/non-langchain/anthropic_tool_calling.ipynb

@@ -83,7 +83,7 @@
     "import json\n",
     "\n",
     "# import the kfinance client\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "\n",
     "# check if the current environment is a Google Colab\n",
     "try:\n",
@@ -129,12 +129,13 @@
    },
    "outputs": [],
    "source": [
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "from typing import Any, cast\n",
-    "from anthropic.types import ToolParam, Message, MessageParam, ToolResultBlockParam\n",
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
+    "\n",
     "# import Anthropic\n",
     "from anthropic import Anthropic\n",
     "from anthropic.types import TextBlock, ToolUseBlock\n",
+    "from anthropic.types import ToolParam, Message, MessageParam, ToolResultBlockParam\n",
     "\n",
     "\n",
     "def text_from_response_message(response_message: Any) -> str | None:\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/non-langchain/google_gemini_tool_calling.ipynb

@@ -81,7 +81,7 @@
    "outputs": [],
    "source": [
     "# import the kfinance client\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "import json\n",
     "\n",
     "# check if the current environment is a Google Colab\n",
@@ -128,7 +128,7 @@
    },
    "outputs": [],
    "source": [
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "# import Google's generativeai package ('Gemini')\n",
     "import google.generativeai as genai\n",
     "\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/example_notebooks/tool_calling/non-langchain/openai_langchain_tool_calling.ipynb

@@ -83,7 +83,7 @@
     "import json\n",
     "\n",
     "# import the kfinance client\n",
-    "from kfinance.kfinance import Client\n",
+    "from kfinance.client.kfinance import Client\n",
     "\n",
     "# check if the current environment is a Google Colab\n",
     "try:\n",
@@ -129,10 +129,10 @@
    },
    "outputs": [],
    "source": [
+    "from kfinance.integrations.tool_calling.prompts import BASE_PROMPT\n",
     "from typing import cast\n",
     "from openai.types.chat import ChatCompletionMessage, ChatCompletionMessageParam, \\\n",
     "    ChatCompletionAssistantMessageParam, ChatCompletionSystemMessageParam, ChatCompletionToolParam\n",
-    "from kfinance.tool_calling.prompts import BASE_PROMPT\n",
     "# import OpenAI\n",
     "from openai import OpenAI\n",
     "\n",

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0/kensho_kfinance.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 2.9.0
+Version: 3.0.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-3.0.0/kensho_kfinance.egg-info/SOURCES.txt

@@ -0,0 +1,142 @@
+.coveragerc
+.gitignore
+.readthedocs.yaml
+AUTHORS.md
+CODE_OF_CONDUCT.md
+CONTRIBUTING.md
+LICENSE
+README.md
+justfile
+pyproject.toml
+setup.py
+.github/workflows/ci-lint.yml
+.github/workflows/ci-test.yml
+.github/workflows/python-publish.yml
+docs/build_tool_calling_documentation.py
+docs/conf.py
+docs/index.rst
+docs/kfinance.rst
+docs/requirements.txt
+docs/tool_calling.rst
+docs/templates/apidoc/package.rst_t
+docs/templates/apidoc/toc.rst_t
+example_notebooks/basic_usage.ipynb
+example_notebooks/tool_calling/langchain/anthropic_langchain_tool_calling.ipynb
+example_notebooks/tool_calling/langchain/google_gemini_langchain_tool_calling.ipynb
+example_notebooks/tool_calling/langchain/openai_langchain_tool_calling.ipynb
+example_notebooks/tool_calling/non-langchain/anthropic_tool_calling.ipynb
+example_notebooks/tool_calling/non-langchain/google_gemini_tool_calling.ipynb
+example_notebooks/tool_calling/non-langchain/openai_langchain_tool_calling.ipynb
+images/colab_logo_32px.png
+kensho_kfinance.egg-info/PKG-INFO
+kensho_kfinance.egg-info/SOURCES.txt
+kensho_kfinance.egg-info/dependency_links.txt
+kensho_kfinance.egg-info/requires.txt
+kensho_kfinance.egg-info/top_level.txt
+kfinance/CHANGELOG.md
+kfinance/__init__.py
+kfinance/conftest.py
+kfinance/py.typed
+kfinance/version.py
+kfinance/client/README.md
+kfinance/client/__init__.py
+kfinance/client/batch_request_handling.py
+kfinance/client/fetch.py
+kfinance/client/industry_models.py
+kfinance/client/kfinance.py
+kfinance/client/meta_classes.py
+kfinance/client/permission_models.py
+kfinance/client/server_thread.py
+kfinance/client/models/__init__.py
+kfinance/client/models/currency_models.py
+kfinance/client/models/date_and_period_models.py
+kfinance/client/models/decimal_with_unit.py
+kfinance/client/models/tests/__init__.py
+kfinance/client/models/tests/test_decimal_with_unit.py
+kfinance/client/tests/__init__.py
+kfinance/client/tests/test_batch_requests.py
+kfinance/client/tests/test_client.py
+kfinance/client/tests/test_fetch.py
+kfinance/client/tests/test_group_objects.py
+kfinance/client/tests/test_objects.py
+kfinance/domains/README.md
+kfinance/domains/__init__.py
+kfinance/domains/business_relationships/__init__.py
+kfinance/domains/business_relationships/business_relationship_models.py
+kfinance/domains/business_relationships/business_relationship_tools.py
+kfinance/domains/business_relationships/tests/__init__.py
+kfinance/domains/business_relationships/tests/test_business_relationship_tools.py
+kfinance/domains/capitalizations/__init__.py
+kfinance/domains/capitalizations/capitalization_models.py
+kfinance/domains/capitalizations/capitalization_tools.py
+kfinance/domains/capitalizations/tests/__init__.py
+kfinance/domains/capitalizations/tests/test_capitalization_models.py
+kfinance/domains/capitalizations/tests/test_capitalization_tools.py
+kfinance/domains/companies/__init__.py
+kfinance/domains/companies/company_identifiers.py
+kfinance/domains/companies/company_models.py
+kfinance/domains/companies/company_tools.py
+kfinance/domains/companies/tests/__init__.py
+kfinance/domains/companies/tests/test_company_tools.py
+kfinance/domains/competitors/__init__.py
+kfinance/domains/competitors/competitor_models.py
+kfinance/domains/competitors/competitor_tools.py
+kfinance/domains/competitors/tests/__init__.py
+kfinance/domains/competitors/tests/test_competitor_tools.py
+kfinance/domains/cusip_and_isin/__init__.py
+kfinance/domains/cusip_and_isin/cusip_and_isin_tools.py
+kfinance/domains/cusip_and_isin/tests/__init__.py
+kfinance/domains/cusip_and_isin/tests/test_cusip_and_isin_tools.py
+kfinance/domains/earnings/__init__.py
+kfinance/domains/earnings/earning_models.py
+kfinance/domains/earnings/earning_tools.py
+kfinance/domains/earnings/tests/__init__.py
+kfinance/domains/earnings/tests/test_earnings_tools.py
+kfinance/domains/line_items/__init__.py
+kfinance/domains/line_items/line_item_models.py
+kfinance/domains/line_items/line_item_tools.py
+kfinance/domains/line_items/tests/__init__.py
+kfinance/domains/line_items/tests/test_line_item_tools.py
+kfinance/domains/mergers_and_acquisitions/__init__.py
+kfinance/domains/mergers_and_acquisitions/merger_and_acquisition_tools.py
+kfinance/domains/mergers_and_acquisitions/tests/__init__.py
+kfinance/domains/mergers_and_acquisitions/tests/test_merger_and_acquisition_tools.py
+kfinance/domains/prices/__init__.py
+kfinance/domains/prices/price_models.py
+kfinance/domains/prices/price_tools.py
+kfinance/domains/prices/tests/__init__.py
+kfinance/domains/prices/tests/test_price_models.py
+kfinance/domains/prices/tests/test_price_tools.py
+kfinance/domains/segments/__init__.py
+kfinance/domains/segments/segment_models.py
+kfinance/domains/segments/segment_tools.py
+kfinance/domains/segments/tests/__init__.py
+kfinance/domains/segments/tests/test_segment_tools.py
+kfinance/domains/statements/__init__.py
+kfinance/domains/statements/statement_models.py
+kfinance/domains/statements/statement_tools.py
+kfinance/domains/statements/tests/__init__.py
+kfinance/domains/statements/tests/test_statement_tools.py
+kfinance/integrations/README.md
+kfinance/integrations/__init__.py
+kfinance/integrations/mcp/__init__.py
+kfinance/integrations/mcp/mcp.py
+kfinance/integrations/tests/__init__.py
+kfinance/integrations/tests/test_example_notebook.py
+kfinance/integrations/tool_calling/README.md
+kfinance/integrations/tool_calling/__init__.py
+kfinance/integrations/tool_calling/all_tools.py
+kfinance/integrations/tool_calling/prompts.py
+kfinance/integrations/tool_calling/tool_calling_models.py
+kfinance/integrations/tool_calling/static_tools/README.md
+kfinance/integrations/tool_calling/static_tools/__init__.py
+kfinance/integrations/tool_calling/static_tools/get_latest.py
+kfinance/integrations/tool_calling/static_tools/get_n_quarters_ago.py
+kfinance/integrations/tool_calling/static_tools/tests/__init__.py
+kfinance/integrations/tool_calling/static_tools/tests/test_get_lastest.py
+kfinance/integrations/tool_calling/static_tools/tests/test_get_n_quarters_ago.py
+kfinance/integrations/tool_calling/tests/__init__.py
+kfinance/integrations/tool_calling/tests/test_tool_calling_models.py
+scripts/copyright_line_check.sh
+scripts/lint.sh
+scripts/test.sh

{kensho_kfinance-2.9.0 → kensho_kfinance-3.0.0}/kfinance/CHANGELOG.md

@@ -1,5 +1,8 @@
 # Changelog
 
+## v3.0.0
+- Add tool calls that accept lists of identifiers.
+
 ## v2.9.0
 - Limit FastMCP version
 

kensho_kfinance-3.0.0/kfinance/__init__.py

@@ -0,0 +1 @@
+from kfinance.integrations.mcp.mcp import run_mcp  # noqa:F401

kensho_kfinance-3.0.0/kfinance/client/README.md

@@ -0,0 +1,9 @@
+# Client
+
+Client contains the main Kfinance `Client` class from `kfinance.py`
+and the lower level `KFinanceApiClient` from `fetch.py`. It also
+includes the objects like `Company` on which the `Client` operates 
+and related helpers like batch processing functionality. 
+
+It may at some point make sense to factor out objects and fetch 
+functions into their related domains.

{kensho_kfinance-2.9.0/kfinance → kensho_kfinance-3.0.0/kfinance/client}/batch_request_handling.py

@@ -1,11 +1,12 @@
 from concurrent.futures import Future
+from dataclasses import dataclass, field
 import functools
 import threading
-from typing import Any, Callable, Iterable, Protocol, Sized, Type, TypeVar
+from typing import Any, Callable, Hashable, Iterable, Protocol, Sized, Type, TypeVar
 
 from requests.exceptions import HTTPError
 
-from .fetch import KFinanceApiClient
+from kfinance.client.fetch import KFinanceApiClient
 
 
 T = TypeVar("T")
@@ -54,29 +55,6 @@ def add_methods_of_singular_class_to_iterable_class(singular_cls: Type[T]) -> Ca
             instances of [singular_cls].
         """
 
-        def process_in_batch(
-            method: Callable, self: IterableKfinanceClass, *args: Any, **kwargs: Any
-        ) -> dict:
-            with self.kfinance_api_client.batch_request_header(batch_size=len(self)):
-                futures = []
-                for obj in self:
-                    # Acquire throttle before submitting the task
-                    throttle.acquire()
-                    future = self.kfinance_api_client.thread_pool.submit(
-                        method, obj, *args, **kwargs
-                    )
-                    # On success or failure, release the throttle.
-                    # This releases the throttle before the
-                    # `resolve_future_with_error_handling` call.
-                    future.add_done_callback(lambda f: throttle.release())
-                    futures.append(future)
-
-                results = {}
-                for obj, future in zip(self, futures):
-                    results[obj] = resolve_future_with_error_handling(future)
-
-            return results
-
         for method_name in dir(singular_cls):
             method = getattr(singular_cls, method_name)
             if method_name.startswith("__") or method_name.startswith("set_"):
@@ -88,7 +66,13 @@ def add_methods_of_singular_class_to_iterable_class(singular_cls: Type[T]) -> Ca
                     def method_wrapper(
                         self: IterableKfinanceClass, *args: Any, **kwargs: Any
                     ) -> dict:
-                        return process_in_batch(method, self, *args, **kwargs)
+                        return process_tasks_in_thread_pool_executor(
+                            api_client=self.kfinance_api_client,
+                            tasks=[
+                                Task(func=method, args=(obj, *args), kwargs=kwargs, result_key=obj)
+                                for obj in self
+                            ],
+                        )
 
                     return method_wrapper
 
@@ -102,7 +86,12 @@ def add_methods_of_singular_class_to_iterable_class(singular_cls: Type[T]) -> Ca
                     @functools.wraps(method.fget)
                     def prop_wrapper(self: IterableKfinanceClass) -> Any:
                         assert method.fget is not None
-                        return process_in_batch(method.fget, self)
+                        return process_tasks_in_thread_pool_executor(
+                            api_client=self.kfinance_api_client,
+                            tasks=[
+                                Task(func=method.fget, args=(obj,), result_key=obj) for obj in self
+                            ],
+                        )
 
                     return prop_wrapper
 
@@ -113,6 +102,53 @@ def add_methods_of_singular_class_to_iterable_class(singular_cls: Type[T]) -> Ca
     return decorator
 
 
+@dataclass(kw_only=True)
+class Task:
+    """A task for batch processing.
+
+    - args and kwargs are intended to be passed into the func as func(*args, **kwargs)
+    - results from batch processing are returned as dicts. The result_key is usually
+        something that either is an id (e.g. company_id) or has an id (e.g. Company
+        or CompanyIdentifier), used to map from that key to the corresponding result.
+    - The future is used to store the batch processing future. It should not be modified
+        directly outside of process_tasks_in_thread_pool_executor.
+    """
+
+    func: Callable
+    args: Any = field(default_factory=tuple)
+    kwargs: Any = field(default_factory=dict)
+    result_key: Hashable
+    future: Future | None = field(init=False, default=None)
+
+
+def process_tasks_in_thread_pool_executor(api_client: KFinanceApiClient, tasks: list[Task]) -> dict:
+    """Execute a list of tasks in the api client's thread pool executor and return the results.
+
+    Returns a dict mapping from each task's key to the corresponding result.
+    """
+
+    # Update access if necessary before submitting batch job.
+    # If the batch job starts without a valid token, each thread may try to refresh it.
+    assert api_client.access_token
+    with api_client.batch_request_header(batch_size=len(tasks)):
+        for task in tasks:
+            # Acquire throttle before submitting the task
+            throttle.acquire()
+            future = api_client.thread_pool.submit(task.func, *task.args, **task.kwargs)
+            # On success or failure, release the throttle.
+            # This releases the throttle before the
+            # `resolve_future_with_error_handling` call.
+            future.add_done_callback(lambda f: throttle.release())
+            task.future = future
+
+        results = {}
+        for task in tasks:
+            assert task.future
+            results[task.result_key] = resolve_future_with_error_handling(task.future)
+
+    return results
+
+
 def resolve_future_with_error_handling(future: Future) -> Any:
     """Return the result of a future with error handling for non-200 status codes.
 

{kensho_kfinance-2.9.0/kfinance → kensho_kfinance-3.0.0/kfinance/client}/fetch.py

@@ -7,19 +7,21 @@ from typing import Callable, Generator, Optional
 from uuid import uuid4
 
 import jwt
-from pydantic import ValidationError
 import requests
 
-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
+from kfinance.client.industry_models import IndustryClassification
+from kfinance.client.models.date_and_period_models import Periodicity, PeriodType
+from kfinance.client.permission_models import Permission
+from kfinance.domains.business_relationships.business_relationship_models import (
+    BusinessRelationshipType,
+    RelationshipResponse,
+)
+from kfinance.domains.capitalizations.capitalization_models import Capitalizations
+from kfinance.domains.companies.company_models import IdentificationTriple
+from kfinance.domains.competitors.competitor_models import CompetitorResponse, CompetitorSource
+from kfinance.domains.earnings.earning_models import EarningsCallResp
+from kfinance.domains.prices.price_models import PriceHistory
+from kfinance.domains.segments.segment_models import SegmentType
 
 
 # version.py gets autogenerated by setuptools-scm and is not available
@@ -528,30 +530,20 @@ class KFinanceApiClient:
 
     def fetch_companies_from_business_relationship(
         self, company_id: int, relationship_type: BusinessRelationshipType
-    ) -> RelationshipResponse | RelationshipResponseNoName:
+    ) -> RelationshipResponse:
         """Fetches a dictionary of current and previous company IDs and names associated with a given company ID based on the specified relationship type.
 
         Example: fetch_companies_from_business_relationship(company_id=1234, relationship_type="distributor") returns a dictionary of company 1234's current and previous distributors.
 
-        As of 2024-05-28, we are changing the response on the backend from
-        RelationshipResponseNoName to RelationshipResponse. This function can handle both response
-        types.
-
         :param company_id: The ID of the company for which associated companies are being fetched.
         :type company_id: int
         :param relationship_type: The type of relationship to filter by. Valid relationship types are defined in the BusinessRelationshipType class.
         :type relationship_type: BusinessRelationshipType
-        :return: A dictionary containing lists of current and previous company IDs that have the specified relationship with the given company_id.
-        :rtype: RelationshipResponse | RelationshipResponseNoName
+        :return: A dictionary containing lists of current and previous company IDs and names that have the specified relationship with the given company_id.
+        :rtype: RelationshipResponse
         """
         url = f"{self.url_base}relationship/{company_id}/{relationship_type}"
-        result = self.fetch(url)
-        # Try to parse as the newer RelationshipResponse and fall back to
-        # RelationshipResponseNoName if that fails.
-        try:
-            return RelationshipResponse.model_validate(result)
-        except ValidationError:
-            return RelationshipResponseNoName.model_validate(result)
+        return RelationshipResponse.model_validate(self.fetch(url))
 
     def fetch_ticker_from_industry_code(
         self,
@@ -589,22 +581,24 @@ class KFinanceApiClient:
         url = f"{self.url_base}company_groups/industry/{industry_classification}/{industry_code}"
         return self.fetch(url)
 
-    def fetch_earnings(self, company_id: int) -> dict:
+    def fetch_earnings(self, company_id: int) -> EarningsCallResp:
         """Get the earnings for a company."""
         url = f"{self.url_base}earnings/{company_id}"
-        return self.fetch(url)
+        return EarningsCallResp.model_validate(self.fetch(url))
 
     def fetch_transcript(self, key_dev_id: int) -> dict:
         """Get the transcript for an earnings item."""
         url = f"{self.url_base}transcript/{key_dev_id}"
         return self.fetch(url)
 
-    def fetch_competitors(self, company_id: int, competitor_source: CompetitorSource) -> dict:
+    def fetch_competitors(
+        self, company_id: int, competitor_source: CompetitorSource
+    ) -> CompetitorResponse:
         """Get the competitors for a company."""
         url = f"{self.url_base}competitors/{company_id}"
         if competitor_source is not CompetitorSource.all:
             url = url + f"/{competitor_source}"
-        return self.fetch(url)
+        return CompetitorResponse.model_validate(self.fetch(url))
 
     def fetch_mergers_for_company(
         self,