kensho-kfinance 3.2.14__tar.gz → 3.2.16__tar.gz

{kensho_kfinance-3.2.14/kensho_kfinance.egg-info → kensho_kfinance-3.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 3.2.14
+Version: 3.2.16
 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.2.14 → kensho_kfinance-3.2.16/kensho_kfinance.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kensho-kfinance
-Version: 3.2.14
+Version: 3.2.16
 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.2.14 → kensho_kfinance-3.2.16}/kfinance/CHANGELOG.md

@@ -1,4 +1,11 @@
 # Changelog
+
+## v3.2.16
+- Allow None and "NaN" for DecimalWithValue
+
+## v3.2.15
+- Add smart line item validation with intelligent suggestions and descriptions to reduce token consumption by 75%
+
 # 3.2.14
 - Fix Capitalizations validator to handle nulls
 

{kensho_kfinance-3.2.14 → kensho_kfinance-3.2.16}/kfinance/client/models/decimal_with_unit.py

@@ -2,7 +2,7 @@ from copy import deepcopy
 from decimal import Decimal
 from typing import Any
 
-from pydantic import BaseModel, Field, model_validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 from typing_extensions import Self
 
 from kfinance.client.models.currency_models import ISO_CODE_TO_CURRENCY
@@ -20,11 +20,23 @@ class DecimalWithUnit(BaseModel):
     existing subclass like `Money` or `Shares` or create a new one.
     """
 
-    value: Decimal
+    value: Decimal = Field(allow_inf_nan=True)
     unit: str
     # exclude conventional_decimals from serialization
     conventional_decimals: int = Field(exclude=True)
 
+    @field_validator("value", mode="before")
+    @classmethod
+    def convert_none_to_nan(cls, v: Any) -> Any:
+        """Convert None values to NaN.
+
+        Price data can include None for open prices.
+        https://kfinance.kensho.com/api/v1/pricing/37284793/2003-01-01/2024-12-31/month/adjusted
+        """
+        if v is None:
+            return Decimal("NaN")
+        return v
+
     @model_validator(mode="after")
     def quantize_value(self) -> Self:
         """Quantize the value at the end of the deserialization.

{kensho_kfinance-3.2.14 → kensho_kfinance-3.2.16}/kfinance/client/models/tests/test_decimal_with_unit.py

@@ -32,6 +32,15 @@ class TestDecimalWithUnit:
         )
         assert dwu.value == expected_value
 
+    @pytest.mark.parametrize("value", [None, "NaN"])
+    def test_null_nan_allowed(self, value: str | None):
+        """
+        WHEN a DecimalWithUnit gets deserialized with None or "NaN"
+        THEN the deserialized value is Decimal("NaN")
+        """
+        dwu = DecimalWithUnit.model_validate(dict(value=value, conventional_decimals=1, unit="foo"))
+        assert dwu.value.is_nan()
+
 
 class TestMoney:
     @pytest.mark.parametrize("currency, expected_conventional_decimals", [("USD", 2), ("BIF", 0)])

{kensho_kfinance-3.2.14 → kensho_kfinance-3.2.16}/kfinance/domains/competitors/competitor_tools.py

@@ -1,3 +1,5 @@
+from typing import Type
+
 from kfinance.client.batch_request_handling import Task, process_tasks_in_thread_pool_executor
 from kfinance.client.permission_models import Permission
 from kfinance.domains.competitors.competitor_models import CompetitorResponse, CompetitorSource
@@ -20,7 +22,7 @@ class GetCompetitorsFromIdentifiersResp(ToolRespWithErrors):
 class GetCompetitorsFromIdentifiers(KfinanceTool):
     name: str = "get_competitors_from_identifiers"
     description: str = "Retrieves a list of company_id and company_name that are competitors for a list of companies, optionally filtered by the source of the competitor information."
-    args_schema = GetCompetitorsFromIdentifiersArgs
+    args_schema: Type[GetCompetitorsFromIdentifiersArgs] = GetCompetitorsFromIdentifiersArgs
     accepted_permissions: set[Permission] | None = {Permission.CompetitorsPermission}
 
     def _run(