kensho-kfinance 2.7.0__py3-none-any.whl → 2.9.0__py3-none-any.whl

kfinance/tests/test_models/test_price_models.py

@@ -0,0 +1,58 @@
+from decimal import Decimal
+
+from kfinance.decimal_with_unit import Money, Shares
+from kfinance.models.price_models import PriceHistory, Prices
+
+
+class TestPriceHistory:
+    api_resp = {
+        "currency": "USD",
+        "prices": [
+            {
+                "date": "2024-06-25",
+                "open": "445.790000",
+                "high": "449.240000",
+                "low": "442.770000",
+                "close": "448.780000",
+                "volume": "999134",
+            },
+            {
+                "date": "2024-06-26",
+                "open": "446.320000",
+                "high": "449.120000",
+                "low": "443.560000",
+                "close": "448.360000",
+                "volume": "1630769",
+            },
+        ],
+    }
+
+    def test_price_history_deserialization(self) -> None:
+        """
+        GIVEN a price history API response
+        WHEN we deserialize the response into a PriceHistory object
+        THEN the deserialization succeeds and returns the expected value.
+        """
+        expected_price_history = PriceHistory.model_construct(
+            prices=[
+                Prices(
+                    date="2024-06-25",
+                    open=Money(value=Decimal("445.79"), unit="USD", conventional_decimals=2),
+                    high=Money(value=Decimal("449.24"), unit="USD", conventional_decimals=2),
+                    low=Money(value=Decimal("442.77"), unit="USD", conventional_decimals=2),
+                    close=Money(value=Decimal("448.78"), unit="USD", conventional_decimals=2),
+                    volume=Shares(value=Decimal("999134"), unit="Shares", conventional_decimals=0),
+                ),
+                Prices(
+                    date="2024-06-26",
+                    open=Money(value=Decimal("446.32"), unit="USD", conventional_decimals=2),
+                    high=Money(value=Decimal("449.12"), unit="USD", conventional_decimals=2),
+                    low=Money(value=Decimal("443.56"), unit="USD", conventional_decimals=2),
+                    close=Money(value=Decimal("448.36"), unit="USD", conventional_decimals=2),
+                    volume=Shares(value=Decimal("1630769"), unit="Shares", conventional_decimals=0),
+                ),
+            ]
+        )
+
+        price_history = PriceHistory.model_validate(self.api_resp)
+        assert price_history == expected_price_history

kfinance/tests/test_objects.py

@@ -10,18 +10,19 @@ import pandas as pd
 from PIL.Image import open as image_open
 import time_machine
 
-from kfinance.constants import BusinessRelationshipType
 from kfinance.kfinance import (
-    AdvisedCompany,
     BusinessRelationships,
     Company,
     Earnings,
     MergerOrAcquisition,
+    ParticipantInMerger,
     Security,
     Ticker,
     TradingItem,
     Transcript,
 )
+from kfinance.models.business_relationship_models import BusinessRelationshipType
+from kfinance.models.capitalization_models import Capitalizations
 from kfinance.pydantic_models import CompanyIdAndName, RelationshipResponse
 
 
@@ -124,16 +125,40 @@ MOCK_COMPANY_DB = {
         },
         "mergers": {
             "target": [
-                {"transaction_id": 10998717, "merger_title": "Closed M/A of Microsoft Corporation"},
-                {"transaction_id": 28237969, "merger_title": "Closed M/A of Microsoft Corporation"},
+                {
+                    "transaction_id": 10998717,
+                    "merger_title": "Closed M/A of Microsoft Corporation",
+                    "closed_date": "2021-01-01",
+                },
+                {
+                    "transaction_id": 28237969,
+                    "merger_title": "Closed M/A of Microsoft Corporation",
+                    "closed_date": "2022-01-01",
+                },
             ],
             "buyer": [
-                {"transaction_id": 517414, "merger_title": "Closed M/A of MongoMusic, Inc."},
-                {"transaction_id": 596722, "merger_title": "Closed M/A of Digital Anvil, Inc."},
+                {
+                    "transaction_id": 517414,
+                    "merger_title": "Closed M/A of MongoMusic, Inc.",
+                    "closed_date": "2023-01-01",
+                },
+                {
+                    "transaction_id": 596722,
+                    "merger_title": "Closed M/A of Digital Anvil, Inc.",
+                    "closed_date": "2023-01-01",
+                },
             ],
             "seller": [
-                {"transaction_id": 455551, "merger_title": "Closed M/A of VacationSpot.com, Inc."},
-                {"transaction_id": 456045, "merger_title": "Closed M/A of TransPoint, LLC"},
+                {
+                    "transaction_id": 455551,
+                    "merger_title": "Closed M/A of VacationSpot.com, Inc.",
+                    "closed_date": "2024-01-01",
+                },
+                {
+                    "transaction_id": 456045,
+                    "merger_title": "Closed M/A of TransPoint, LLC",
+                    "closed_date": "2025-01-01",
+                },
             ],
         },
         "advisors": {
@@ -329,23 +354,26 @@ class MockKFinanceApiClient:
         company_id: int,
         start_date: Optional[str] = None,
         end_date: Optional[str] = None,
-    ) -> dict:
-        return {
-            "market_caps": [
-                {
-                    "date": "2025-01-01",
-                    "market_cap": "3133802247084.000000",
-                    "tev": "3152211247084.000000",
-                    "shares_outstanding": 7434880776,
-                },
-                {
-                    "date": "2025-01-02",
-                    "market_cap": "3112092395218.000000",
-                    "tev": "3130501395218.000000",
-                    "shares_outstanding": 7434880776,
-                },
-            ]
-        }
+    ) -> Capitalizations:
+        return Capitalizations.model_validate(
+            {
+                "currency": "USD",
+                "market_caps": [
+                    {
+                        "date": "2025-01-01",
+                        "market_cap": "3133802247084.000000",
+                        "tev": "3152211247084.000000",
+                        "shares_outstanding": 7434880776,
+                    },
+                    {
+                        "date": "2025-01-02",
+                        "market_cap": "3112092395218.000000",
+                        "tev": "3130501395218.000000",
+                        "shares_outstanding": 7434880776,
+                    },
+                ],
+            }
+        )
 
     def fetch_segments(
         self,
@@ -376,8 +404,8 @@ class MockKFinanceApiClient:
     def fetch_mergers_for_company(self, company_id):
         return copy.deepcopy(MOCK_COMPANY_DB[company_id]["mergers"])
 
-    def fetch_merger_info(self, transaction_id):
-        return copy.deepcopy(MOCK_MERGERS_DB[transaction_id])
+    def fetch_merger_info(self, transaction_id: int):
+        return copy.deepcopy(MOCK_MERGERS_DB[str(transaction_id)])
 
     def fetch_advisors_for_company_in_merger(self, transaction_id, advised_company_id):
         return copy.deepcopy(MOCK_COMPANY_DB[advised_company_id]["advisors"][transaction_id])
@@ -440,26 +468,31 @@ class TestCompany(TestCase):
     def setUp(self):
         """setup tests"""
         self.kfinance_api_client = MockKFinanceApiClient()
-        self.msft_company = AdvisedCompany(
-            self.kfinance_api_client, company_id=msft_company_id, transaction_id=msft_buys_mongo
+        self.msft_company = ParticipantInMerger(
+            kfinance_api_client=self.kfinance_api_client,
+            transaction_id=msft_buys_mongo,
+            company=Company(
+                kfinance_api_client=self.kfinance_api_client,
+                company_id=msft_company_id,
+            ),
         )
 
     def test_company_id(self) -> None:
         """test company id"""
         expected_company_id = msft_company_id
-        company_id = self.msft_company.company_id
+        company_id = self.msft_company.company.company_id
         self.assertEqual(expected_company_id, company_id)
 
     def test_info(self) -> None:
         """test info"""
         expected_info = MOCK_COMPANY_DB[msft_company_id]["info"]
-        info = self.msft_company.info
+        info = self.msft_company.company.info
         self.assertEqual(expected_info, info)
 
     def test_name(self) -> None:
         """test name"""
         expected_name = MOCK_COMPANY_DB[msft_company_id]["info"]["name"]
-        name = self.msft_company.name
+        name = self.msft_company.company.name
         self.assertEqual(expected_name, name)
 
     def test_founding_date(self) -> None:
@@ -467,7 +500,7 @@ class TestCompany(TestCase):
         expected_founding_date = datetime.strptime(
             MOCK_COMPANY_DB[msft_company_id]["info"]["founding_date"], "%Y-%m-%d"
         ).date()
-        founding_date = self.msft_company.founding_date
+        founding_date = self.msft_company.company.founding_date
         self.assertEqual(expected_founding_date, founding_date)
 
     def test_earnings_call_datetimes(self) -> None:
@@ -477,7 +510,7 @@ class TestCompany(TestCase):
                 MOCK_COMPANY_DB[msft_company_id]["earnings_call_dates"]["earnings"][0]
             ).replace(tzinfo=timezone.utc)
         ]
-        earnings_call_datetimes = self.msft_company.earnings_call_datetimes
+        earnings_call_datetimes = self.msft_company.company.earnings_call_datetimes
         self.assertEqual(expected_earnings_call_datetimes, earnings_call_datetimes)
 
     def test_income_statement(self) -> None:
@@ -489,7 +522,7 @@ class TestCompany(TestCase):
             .apply(pd.to_numeric)
             .replace(np.nan, None)
         )
-        income_statement = self.msft_company.income_statement()
+        income_statement = self.msft_company.company.income_statement()
         pd.testing.assert_frame_equal(expected_income_statement, income_statement)
 
     def test_revenue(self) -> None:
@@ -501,14 +534,14 @@ class TestCompany(TestCase):
             .replace(np.nan, None)
             .set_index(pd.Index(["revenue"]))
         )
-        revenue = self.msft_company.revenue()
+        revenue = self.msft_company.company.revenue()
         pd.testing.assert_frame_equal(expected_revenue, revenue)
 
     def test_business_segments(self) -> None:
         """test business statement"""
         expected_segments = MOCK_COMPANY_DB[msft_company_id]["segments"]
 
-        business_segment = self.msft_company.business_segments()
+        business_segment = self.msft_company.company.business_segments()
         self.assertEqual(expected_segments, business_segment)
 
     def test_relationships(self) -> None:
@@ -519,7 +552,9 @@ class TestCompany(TestCase):
 
         expected_suppliers = MOCK_COMPANY_DB[msft_company_id][BusinessRelationshipType.supplier]
 
-        suppliers_via_method = self.msft_company.relationships(BusinessRelationshipType.supplier)
+        suppliers_via_method = self.msft_company.company.relationships(
+            BusinessRelationshipType.supplier
+        )
         self.assertIsInstance(suppliers_via_method, BusinessRelationships)
         # Company ids should match
         self.assertEqual(
@@ -532,23 +567,35 @@ class TestCompany(TestCase):
         )
 
         # Fetching via property should return the same result
-        suppliers_via_property = self.msft_company.supplier
+        suppliers_via_property = self.msft_company.company.supplier
         self.assertEqual(suppliers_via_property, suppliers_via_method)
 
     def test_mergers(self) -> None:
         expected_mergers = MOCK_COMPANY_DB[msft_company_id]["mergers"]
-        mergers = self.msft_company.mergers_and_acquisitions
+        mergers = self.msft_company.company.mergers_and_acquisitions
         mergers_json = {
             "target": [
-                {"transaction_id": merger.transaction_id, "merger_title": merger.merger_title}
+                {
+                    "transaction_id": merger.transaction_id,
+                    "merger_title": merger.merger_title,
+                    "closed_date": merger.closed_date,
+                }
                 for merger in mergers["target"]
             ],
             "buyer": [
-                {"transaction_id": merger.transaction_id, "merger_title": merger.merger_title}
+                {
+                    "transaction_id": merger.transaction_id,
+                    "merger_title": merger.merger_title,
+                    "closed_date": merger.closed_date,
+                }
                 for merger in mergers["buyer"]
             ],
             "seller": [
-                {"transaction_id": merger.transaction_id, "merger_title": merger.merger_title}
+                {
+                    "transaction_id": merger.transaction_id,
+                    "merger_title": merger.merger_title,
+                    "closed_date": merger.closed_date,
+                }
                 for merger in mergers["seller"]
             ],
         }
@@ -567,7 +614,7 @@ class TestCompany(TestCase):
         company_ids: list[int] = []
         advisor_type_names: list[str] = []
         for advisor in advisors:
-            company_ids.append(advisor.company_id)
+            company_ids.append(advisor.company.company_id)
             advisor_type_names.append(advisor.advisor_type_name)
         self.assertListEqual(expected_company_ids, company_ids)
         self.assertListEqual(expected_advisor_type_names, advisor_type_names)
@@ -843,12 +890,14 @@ class TestTicker(TestCase):
         THEN the Ticker object can correctly extract market caps from the dict.
         """
 
-        expected_dataframe = pd.DataFrame(
-            {"market_cap": {"2025-01-01": 3133802247084.0, "2025-01-02": 3112092395218.0}}
-        )
-        expected_dataframe.index.name = "date"
+        expected_response = {
+            "market_cap": [
+                {"2025-01-01": {"unit": "USD", "value": "3133802247084.00"}},
+                {"2025-01-02": {"unit": "USD", "value": "3112092395218.00"}},
+            ]
+        }
         market_caps = self.msft_ticker_from_ticker.market_cap()
-        pd.testing.assert_frame_equal(expected_dataframe, market_caps)
+        assert market_caps == expected_response
 
 
 class TestTranscript(TestCase):
@@ -964,9 +1013,10 @@ class TestMerger(TestCase):
     def setUp(self):
         self.kfinance_api_client = MockKFinanceApiClient()
         self.merger = MergerOrAcquisition(
-            self.kfinance_api_client,
-            transaction_id=msft_buys_mongo,
+            kfinance_api_client=self.kfinance_api_client,
+            transaction_id=int(msft_buys_mongo),
             merger_title="Closed M/A of MongoMusic, Inc.",
+            closed_date=date(2021, 1, 1),
         )
 
     def test_merger_info(self) -> None:
@@ -978,16 +1028,16 @@ class TestMerger(TestCase):
             ],
             "participants": {
                 "target": {
-                    "company_id": self.merger.get_participants["target"].company_id,
-                    "company_name": self.merger.get_participants["target"].name,
+                    "company_id": self.merger.get_participants["target"].company.company_id,
+                    "company_name": self.merger.get_participants["target"].company.name,
                 },
                 "buyers": [
-                    {"company_id": company.company_id, "company_name": company.name}
-                    for company in self.merger.get_participants["buyers"]
+                    {"company_id": buyer.company.company_id, "company_name": buyer.company.name}
+                    for buyer in self.merger.get_participants["buyers"]
                 ],
                 "sellers": [
-                    {"company_id": company.company_id, "company_name": company.name}
-                    for company in self.merger.get_participants["sellers"]
+                    {"company_id": seller.company.company_id, "company_name": seller.company.name}
+                    for seller in self.merger.get_participants["sellers"]
                 ],
             },
             "consideration": {

kfinance/tests/test_tools.py

@@ -9,14 +9,12 @@ from pytest import raises
 from requests_mock import Mocker
 import time_machine
 
-from kfinance.constants import (
-    BusinessRelationshipType,
-    Capitalization,
-    CompetitorSource,
-    SegmentType,
-    StatementType,
-)
 from kfinance.kfinance import Client, NoEarningsDataError
+from kfinance.models.business_relationship_models import BusinessRelationshipType
+from kfinance.models.capitalization_models import Capitalization
+from kfinance.models.competitor_models import CompetitorSource
+from kfinance.models.segment_models import SegmentType
+from kfinance.models.statement_models import StatementType
 from kfinance.tests.conftest import SPGI_COMPANY_ID, SPGI_SECURITY_ID, SPGI_TRADING_ITEM_ID
 from kfinance.tests.test_objects import MOCK_COMPANY_DB, MOCK_MERGERS_DB, ordered
 from kfinance.tool_calling import (
@@ -164,6 +162,7 @@ class TestGetCapitalizationFromIdentifier:
         requests_mock.get(
             url=f"https://kfinance.kensho.com/api/v1/market_cap/{SPGI_COMPANY_ID}/none/none",
             json={
+                "currency": "USD",
                 "market_caps": [
                     {
                         "date": "2024-04-10",
@@ -177,11 +176,16 @@ class TestGetCapitalizationFromIdentifier:
                         "tev": "147105066761.000000",
                         "shares_outstanding": 313099562,
                     },
-                ]
+                ],
             },
         )
 
-        expected_response = "| date       |   market_cap |\n|:-----------|-------------:|\n| 2024-04-10 |  1.32767e+11 |\n| 2024-04-11 |  1.32416e+11 |"
+        expected_response = {
+            "market_cap": [
+                {"2024-04-10": {"unit": "USD", "value": "132766738270.00"}},
+                {"2024-04-11": {"unit": "USD", "value": "132416066761.00"}},
+            ]
+        }
 
         tool = GetCapitalizationFromIdentifier(kfinance_client=mock_client)
         args = GetCapitalizationFromIdentifierArgs(
@@ -442,6 +446,7 @@ class TestPricesFromIdentifier:
             url=f"https://kfinance.kensho.com/api/v1/pricing/{SPGI_TRADING_ITEM_ID}/none/none/day/adjusted",
             # truncated response
             json={
+                "currency": "USD",
                 "prices": [
                     {
                         "date": "2024-04-11",
@@ -459,10 +464,29 @@ class TestPricesFromIdentifier:
                         "close": "417.810000",
                         "volume": "1182229",
                     },
-                ]
+                ],
             },
         )
-        expected_response = "| date       |   open |   high |    low |   close |      volume |\n|:-----------|-------:|-------:|-------:|--------:|------------:|\n| 2024-04-11 | 424.26 | 425.99 | 422.04 |  422.92 | 1.12916e+06 |\n| 2024-04-12 | 419.23 | 421.94 | 416.45 |  417.81 | 1.18223e+06 |"
+        expected_response = {
+            "prices": [
+                {
+                    "date": "2024-04-11",
+                    "open": {"value": "424.26", "unit": "USD"},
+                    "high": {"value": "425.99", "unit": "USD"},
+                    "low": {"value": "422.04", "unit": "USD"},
+                    "close": {"value": "422.92", "unit": "USD"},
+                    "volume": {"value": "1129158", "unit": "Shares"},
+                },
+                {
+                    "date": "2024-04-12",
+                    "open": {"value": "419.23", "unit": "USD"},
+                    "high": {"value": "421.94", "unit": "USD"},
+                    "low": {"value": "416.45", "unit": "USD"},
+                    "close": {"value": "417.81", "unit": "USD"},
+                    "volume": {"value": "1182229", "unit": "Shares"},
+                },
+            ]
+        }
 
         tool = GetPricesFromIdentifier(kfinance_client=mock_client)
         response = tool.run(GetPricesFromIdentifierArgs(identifier="SPGI").model_dump(mode="json"))

kfinance/tool_calling/get_advisors_for_company_in_transaction_from_identifier.py

@@ -2,8 +2,8 @@ from typing import Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import Permission
-from kfinance.kfinance import AdvisedCompany
+from kfinance.kfinance import Company, ParticipantInMerger
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 
@@ -19,18 +19,21 @@ class GetAdvisorsForCompanyInTransactionFromIdentifier(KfinanceTool):
 
     def _run(self, identifier: str, transaction_id: int) -> list:
         ticker = self.kfinance_client.ticker(identifier)
-        advised_company = AdvisedCompany(
+        participant_in_merger = ParticipantInMerger(
             kfinance_api_client=ticker.kfinance_api_client,
-            company_id=ticker.company.company_id,
             transaction_id=transaction_id,
+            company=Company(
+                kfinance_api_client=ticker.kfinance_api_client,
+                company_id=ticker.company.company_id,
+            ),
         )
-        advisors = advised_company.advisors
+        advisors = participant_in_merger.advisors
 
         if advisors:
             return [
                 {
-                    "advisor_company_id": advisor.company_id,
-                    "advisor_company_name": advisor.name,
+                    "advisor_company_id": advisor.company.company_id,
+                    "advisor_company_name": advisor.company.name,
                     "advisor_type_name": advisor.advisor_type_name,
                 }
                 for advisor in advisors

kfinance/tool_calling/get_business_relationship_from_identifier.py

@@ -2,8 +2,9 @@ from typing import Type
 
 from pydantic import BaseModel
 
-from kfinance.constants import BusinessRelationshipType, Permission
 from kfinance.kfinance import BusinessRelationships
+from kfinance.models.business_relationship_models import BusinessRelationshipType
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 

kfinance/tool_calling/get_capitalization_from_identifier.py

@@ -2,7 +2,8 @@ from datetime import date
 
 from pydantic import Field
 
-from kfinance.constants import Capitalization, Permission
+from kfinance.models.capitalization_models import Capitalization
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 
@@ -29,8 +30,6 @@ class GetCapitalizationFromIdentifier(KfinanceTool):
         capitalization: Capitalization,
         start_date: str | None = None,
         end_date: str | None = None,
-    ) -> str:
+    ) -> dict:
         ticker = self.kfinance_client.ticker(identifier)
-        return getattr(ticker, capitalization.value)(
-            start_date=start_date, end_date=end_date
-        ).to_markdown()
+        return getattr(ticker, capitalization.value)(start_date=start_date, end_date=end_date)

kfinance/tool_calling/get_competitors_from_identifier.py

@@ -1,4 +1,5 @@
-from kfinance.constants import CompetitorSource, Permission
+from kfinance.models.competitor_models import CompetitorSource
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 

kfinance/tool_calling/get_cusip_from_ticker.py

@@ -2,7 +2,7 @@ from typing import Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import Permission
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool
 
 

kfinance/tool_calling/get_earnings.py

@@ -2,8 +2,8 @@ from typing import Type
 
 from pydantic import BaseModel
 
-from kfinance.constants import Permission
 from kfinance.kfinance import NoEarningsDataError
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 

kfinance/tool_calling/get_financial_line_item_from_identifier.py

@@ -2,7 +2,9 @@ from typing import Literal, Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import LINE_ITEM_NAMES_AND_ALIASES, PeriodType, Permission
+from kfinance.models.date_and_period_models import PeriodType
+from kfinance.models.line_item_models import LINE_ITEM_NAMES_AND_ALIASES
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier, ValidQuarter
 
 

kfinance/tool_calling/get_financial_statement_from_identifier.py

@@ -2,7 +2,9 @@ from typing import Literal, Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import PeriodType, Permission, StatementType
+from kfinance.models.date_and_period_models import PeriodType
+from kfinance.models.permission_models import Permission
+from kfinance.models.statement_models import StatementType
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier, ValidQuarter
 
 

kfinance/tool_calling/get_history_metadata_from_identifier.py

@@ -2,7 +2,8 @@ from typing import Type
 
 from pydantic import BaseModel
 
-from kfinance.constants import HistoryMetadata, Permission
+from kfinance.models.permission_models import Permission
+from kfinance.models.price_models import HistoryMetadata
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 

kfinance/tool_calling/get_info_from_identifier.py

@@ -2,7 +2,7 @@ from typing import Type
 
 from pydantic import BaseModel
 
-from kfinance.constants import Permission
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier
 
 

kfinance/tool_calling/get_isin_from_ticker.py

@@ -2,7 +2,7 @@ from typing import Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import Permission
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool
 
 

kfinance/tool_calling/get_latest.py

@@ -2,7 +2,8 @@ from typing import Type
 
 from pydantic import BaseModel, Field
 
-from kfinance.constants import LatestPeriods, Permission
+from kfinance.models.date_and_period_models import LatestPeriods
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool
 
 

kfinance/tool_calling/get_latest_earnings.py

@@ -2,8 +2,8 @@ from typing import Type
 
 from pydantic import BaseModel
 
-from kfinance.constants import Permission
 from kfinance.kfinance import NoEarningsDataError
+from kfinance.models.permission_models import Permission
 from kfinance.tool_calling.shared_models import KfinanceTool, ToolArgsWithIdentifier