fin-infra 0.1.69__py3-none-any.whl → 0.1.82__py3-none-any.whl

fin_infra/investments/add.py

@@ -7,7 +7,7 @@ transactions, accounts, allocation, and securities data.
 from __future__ import annotations
 
 from datetime import date
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING
 
 from fastapi import HTTPException
 from pydantic import BaseModel, Field
@@ -24,10 +24,10 @@ except ImportError:
 
 from .ease import easy_investments
 from .models import (
+    AssetAllocation,
     Holding,
-    InvestmentTransaction,
     InvestmentAccount,
-    AssetAllocation,
+    InvestmentTransaction,
     Security,
 )
 from .providers.base import InvestmentProvider
@@ -37,55 +37,55 @@ from .providers.base import InvestmentProvider
 class HoldingsRequest(BaseModel):
     """Request model for holdings endpoint."""
 
-    access_token: Optional[str] = Field(None, description="Plaid access token (Plaid only)")
-    user_id: Optional[str] = Field(None, description="SnapTrade user ID (SnapTrade only)")
-    user_secret: Optional[str] = Field(None, description="SnapTrade user secret (SnapTrade only)")
-    account_ids: Optional[list[str]] = Field(None, description="Filter by specific account IDs")
+    access_token: str | None = Field(None, description="Plaid access token (Plaid only)")
+    user_id: str | None = Field(None, description="SnapTrade user ID (SnapTrade only)")
+    user_secret: str | None = Field(None, description="SnapTrade user secret (SnapTrade only)")
+    account_ids: list[str] | None = Field(None, description="Filter by specific account IDs")
 
 
 class TransactionsRequest(BaseModel):
     """Request model for transactions endpoint."""
 
-    access_token: Optional[str] = Field(None, description="Plaid access token (Plaid only)")
-    user_id: Optional[str] = Field(None, description="SnapTrade user ID (SnapTrade only)")
-    user_secret: Optional[str] = Field(None, description="SnapTrade user secret (SnapTrade only)")
+    access_token: str | None = Field(None, description="Plaid access token (Plaid only)")
+    user_id: str | None = Field(None, description="SnapTrade user ID (SnapTrade only)")
+    user_secret: str | None = Field(None, description="SnapTrade user secret (SnapTrade only)")
     start_date: date = Field(..., description="Start date for transactions (YYYY-MM-DD)")
     end_date: date = Field(..., description="End date for transactions (YYYY-MM-DD)")
-    account_ids: Optional[list[str]] = Field(None, description="Filter by specific account IDs")
+    account_ids: list[str] | None = Field(None, description="Filter by specific account IDs")
 
 
 class AccountsRequest(BaseModel):
     """Request model for investment accounts endpoint."""
 
-    access_token: Optional[str] = Field(None, description="Plaid access token (Plaid only)")
-    user_id: Optional[str] = Field(None, description="SnapTrade user ID (SnapTrade only)")
-    user_secret: Optional[str] = Field(None, description="SnapTrade user secret (SnapTrade only)")
+    access_token: str | None = Field(None, description="Plaid access token (Plaid only)")
+    user_id: str | None = Field(None, description="SnapTrade user ID (SnapTrade only)")
+    user_secret: str | None = Field(None, description="SnapTrade user secret (SnapTrade only)")
 
 
 class AllocationRequest(BaseModel):
     """Request model for asset allocation endpoint."""
 
-    access_token: Optional[str] = Field(None, description="Plaid access token (Plaid only)")
-    user_id: Optional[str] = Field(None, description="SnapTrade user ID (SnapTrade only)")
-    user_secret: Optional[str] = Field(None, description="SnapTrade user secret (SnapTrade only)")
-    account_ids: Optional[list[str]] = Field(None, description="Filter by specific account IDs")
+    access_token: str | None = Field(None, description="Plaid access token (Plaid only)")
+    user_id: str | None = Field(None, description="SnapTrade user ID (SnapTrade only)")
+    user_secret: str | None = Field(None, description="SnapTrade user secret (SnapTrade only)")
+    account_ids: list[str] | None = Field(None, description="Filter by specific account IDs")
 
 
 class SecuritiesRequest(BaseModel):
     """Request model for securities endpoint."""
 
-    access_token: Optional[str] = Field(None, description="Plaid access token (Plaid only)")
-    user_id: Optional[str] = Field(None, description="SnapTrade user ID (SnapTrade only)")
-    user_secret: Optional[str] = Field(None, description="SnapTrade user secret (SnapTrade only)")
+    access_token: str | None = Field(None, description="Plaid access token (Plaid only)")
+    user_id: str | None = Field(None, description="SnapTrade user ID (SnapTrade only)")
+    user_secret: str | None = Field(None, description="SnapTrade user secret (SnapTrade only)")
     security_ids: list[str] = Field(..., description="List of security IDs to retrieve")
 
 
 def add_investments(
     app: FastAPI,
     prefix: str = "/investments",
-    provider: Optional[InvestmentProvider] = None,
+    provider: InvestmentProvider | None = None,
     include_in_schema: bool = True,
-    tags: Optional[list[str]] = None,
+    tags: list[str] | None = None,
 ) -> InvestmentProvider:
     """Add investment endpoints to FastAPI application.
 

fin_infra/investments/models.py

@@ -20,7 +20,7 @@ from __future__ import annotations
 from datetime import date
 from decimal import Decimal
 from enum import Enum
-from typing import TYPE_CHECKING, Dict, List, Optional
+from typing import TYPE_CHECKING, Optional
 
 from pydantic import BaseModel, ConfigDict, Field, computed_field
 
@@ -374,12 +374,12 @@ class InvestmentAccount(BaseModel):
     subtype: Optional[str] = Field(None, description="Account subtype (401k, ira, brokerage)")
 
     # Balances
-    balances: Dict[str, Optional[Decimal]] = Field(
+    balances: dict[str, Optional[Decimal]] = Field(
         ..., description="Current, available, and limit balances"
     )
 
     # Holdings
-    holdings: List[Holding] = Field(default_factory=list, description="List of holdings in account")
+    holdings: list[Holding] = Field(default_factory=list, description="List of holdings in account")
 
     if TYPE_CHECKING:
 
@@ -487,11 +487,11 @@ class AssetAllocation(BaseModel):
         },
     )
 
-    by_security_type: Dict[SecurityType, float] = Field(
+    by_security_type: dict[SecurityType, float] = Field(
         default_factory=dict,
         description="Percentage breakdown by security type (equity, bond, etc.)",
     )
-    by_sector: Dict[str, float] = Field(
+    by_sector: dict[str, float] = Field(
         default_factory=dict,
         description="Percentage breakdown by sector (Technology, Healthcare, etc.)",
     )

fin_infra/investments/providers/base.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 from abc import ABC, abstractmethod
 from datetime import date
-from typing import List, Optional
 
 # Import will work once models.py is fully implemented in Task 3
 # For now, using TYPE_CHECKING to avoid circular imports
@@ -30,8 +29,8 @@ class InvestmentProvider(ABC):
 
     @abstractmethod
     async def get_holdings(
-        self, access_token: str, account_ids: Optional[List[str]] = None
-    ) -> List[Holding]:
+        self, access_token: str, account_ids: list[str] | None = None
+    ) -> list[Holding]:
         """Fetch holdings for investment accounts.
 
         Args:
@@ -54,8 +53,8 @@ class InvestmentProvider(ABC):
         access_token: str,
         start_date: date,
         end_date: date,
-        account_ids: Optional[List[str]] = None,
-    ) -> List[InvestmentTransaction]:
+        account_ids: list[str] | None = None,
+    ) -> list[InvestmentTransaction]:
         """Fetch investment transactions within date range.
 
         Args:
@@ -77,7 +76,7 @@ class InvestmentProvider(ABC):
         pass
 
     @abstractmethod
-    async def get_securities(self, access_token: str, security_ids: List[str]) -> List[Security]:
+    async def get_securities(self, access_token: str, security_ids: list[str]) -> list[Security]:
         """Fetch security details (ticker, name, type, current price).
 
         Args:
@@ -95,7 +94,7 @@ class InvestmentProvider(ABC):
         pass
 
     @abstractmethod
-    async def get_investment_accounts(self, access_token: str) -> List[InvestmentAccount]:
+    async def get_investment_accounts(self, access_token: str) -> list[InvestmentAccount]:
         """Fetch investment accounts with aggregated holdings.
 
         Args:
@@ -113,7 +112,7 @@ class InvestmentProvider(ABC):
 
     # Helper methods (concrete - shared across all providers)
 
-    def calculate_allocation(self, holdings: List[Holding]) -> AssetAllocation:
+    def calculate_allocation(self, holdings: list[Holding]) -> AssetAllocation:
         """Calculate asset allocation by security type and sector.
 
         Groups holdings by security type (equity, bond, ETF, etc.) and calculates
@@ -183,7 +182,7 @@ class InvestmentProvider(ABC):
             cash_percent=cash_percent,
         )
 
-    def calculate_portfolio_metrics(self, holdings: List[Holding]) -> dict:
+    def calculate_portfolio_metrics(self, holdings: list[Holding]) -> dict:
         """Calculate total value, cost basis, unrealized gain/loss.
 
         Aggregates holdings to calculate portfolio-level metrics.

fin_infra/investments/providers/plaid.py

@@ -10,21 +10,35 @@ from __future__ import annotations
 
 from datetime import date
 from decimal import Decimal
-from typing import Any, Dict, List, Optional, cast
-
-from plaid.api import plaid_api
-from plaid.model.investments_holdings_get_request import InvestmentsHoldingsGetRequest
-from plaid.model.investments_transactions_get_request import (
-    InvestmentsTransactionsGetRequest,
-)
-from plaid.model.investments_holdings_get_response import InvestmentsHoldingsGetResponse
-from plaid.model.investments_transactions_get_response import (
-    InvestmentsTransactionsGetResponse,
-)
-from plaid.exceptions import ApiException
-import plaid
-from plaid.api_client import ApiClient
-from plaid.configuration import Configuration
+from typing import Any, cast
+
+try:
+    import plaid
+    from plaid.api import plaid_api
+    from plaid.api_client import ApiClient
+    from plaid.configuration import Configuration
+    from plaid.exceptions import ApiException
+    from plaid.model.investments_holdings_get_request import InvestmentsHoldingsGetRequest
+    from plaid.model.investments_holdings_get_response import InvestmentsHoldingsGetResponse
+    from plaid.model.investments_transactions_get_request import (
+        InvestmentsTransactionsGetRequest,
+    )
+    from plaid.model.investments_transactions_get_response import (
+        InvestmentsTransactionsGetResponse,
+    )
+
+    HAS_PLAID = True
+except ImportError:  # pragma: no cover
+    HAS_PLAID = False
+    plaid_api = None
+    plaid = None
+    ApiClient = None
+    Configuration = None
+    ApiException = Exception
+    InvestmentsHoldingsGetRequest = Any
+    InvestmentsTransactionsGetRequest = Any
+    InvestmentsHoldingsGetResponse = Any
+    InvestmentsTransactionsGetResponse = Any
 
 from ..models import (
     Holding,
@@ -36,6 +50,15 @@ from ..models import (
 from .base import InvestmentProvider
 
 
+def _require_plaid() -> None:
+    """Raise ImportError if plaid-python is not installed."""
+    if not HAS_PLAID:
+        raise ImportError(
+            "Plaid support requires the 'plaid-python' package. "
+            "Install with: pip install fin-infra[plaid] or pip install fin-infra[banking]"
+        )
+
+
 class PlaidInvestmentProvider(InvestmentProvider):
     """Plaid Investment API provider.
 
@@ -76,7 +99,10 @@ class PlaidInvestmentProvider(InvestmentProvider):
 
         Raises:
             ValueError: If client_id or secret is missing
+            ImportError: If plaid-python is not installed
         """
+        _require_plaid()
+
         if not client_id or not secret:
             raise ValueError("client_id and secret are required for Plaid provider")
 
@@ -102,11 +128,11 @@ class PlaidInvestmentProvider(InvestmentProvider):
             "development": plaid.Environment.Sandbox,  # Map development to sandbox
             "production": plaid.Environment.Production,
         }
-        return cast(str, hosts.get(environment.lower(), plaid.Environment.Sandbox))
+        return cast("str", hosts.get(environment.lower(), plaid.Environment.Sandbox))
 
     async def get_holdings(
-        self, access_token: str, account_ids: Optional[List[str]] = None
-    ) -> List[Holding]:
+        self, access_token: str, account_ids: list[str] | None = None
+    ) -> list[Holding]:
         """Fetch investment holdings from Plaid.
 
         Retrieves holdings with security details, quantity, cost basis, and current value.
@@ -163,8 +189,8 @@ class PlaidInvestmentProvider(InvestmentProvider):
         access_token: str,
         start_date: date,
         end_date: date,
-        account_ids: Optional[List[str]] = None,
-    ) -> List[InvestmentTransaction]:
+        account_ids: list[str] | None = None,
+    ) -> list[InvestmentTransaction]:
         """Fetch investment transactions from Plaid.
 
         Retrieves buy/sell/dividend transactions within the specified date range.
@@ -226,7 +252,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         except ApiException as e:
             raise self._transform_error(e)
 
-    async def get_securities(self, access_token: str, security_ids: List[str]) -> List[Security]:
+    async def get_securities(self, access_token: str, security_ids: list[str]) -> list[Security]:
         """Fetch security details from Plaid holdings.
 
         Note: Plaid doesn't have a dedicated securities endpoint.
@@ -264,7 +290,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         except ApiException as e:
             raise self._transform_error(e)
 
-    async def get_investment_accounts(self, access_token: str) -> List[InvestmentAccount]:
+    async def get_investment_accounts(self, access_token: str) -> list[InvestmentAccount]:
         """Fetch investment accounts with aggregated holdings.
 
         Returns accounts with total value, cost basis, and unrealized P&L.
@@ -295,7 +321,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
             }
 
             # Group holdings by account
-            accounts_map: Dict[str, Dict[str, Any]] = {}
+            accounts_map: dict[str, dict[str, Any]] = {}
             for plaid_holding in response.holdings:
                 holding_dict = plaid_holding.to_dict()
                 account_id = holding_dict["account_id"]
@@ -347,7 +373,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
 
     # Helper methods for data transformation
 
-    def _transform_security(self, plaid_security: Dict[str, Any]) -> Security:
+    def _transform_security(self, plaid_security: dict[str, Any]) -> Security:
         """Transform Plaid security data to Security model."""
         # Handle close_price - Plaid may return None for securities without recent pricing
         close_price_raw = plaid_security.get("close_price")
@@ -368,7 +394,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
             currency=plaid_security.get("iso_currency_code", "USD"),
         )
 
-    def _transform_holding(self, plaid_holding: Dict[str, Any], security: Security) -> Holding:
+    def _transform_holding(self, plaid_holding: dict[str, Any], security: Security) -> Holding:
         """Transform Plaid holding data to Holding model."""
         return Holding(
             account_id=plaid_holding["account_id"],
@@ -384,7 +410,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         )
 
     def _transform_transaction(
-        self, plaid_transaction: Dict[str, Any], security: Security
+        self, plaid_transaction: dict[str, Any], security: Security
     ) -> InvestmentTransaction:
         """Transform Plaid investment transaction to InvestmentTransaction model."""
         # Map Plaid transaction type to our enum

fin_infra/investments/providers/snaptrade.py

@@ -11,7 +11,7 @@ from __future__ import annotations
 
 from datetime import date
 from decimal import Decimal
-from typing import Any, Dict, List, Optional, cast
+from typing import Any, cast
 
 import httpx
 
@@ -93,7 +93,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             timeout=30.0,
         )
 
-    def _auth_headers(self, user_id: str, user_secret: str) -> Dict[str, str]:
+    def _auth_headers(self, user_id: str, user_secret: str) -> dict[str, str]:
         """Build authentication headers for SnapTrade API requests.
 
         SECURITY: User secrets are passed in headers, NOT URL params.
@@ -115,8 +115,8 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
     async def get_holdings(
         self,
         access_token: str,
-        account_ids: Optional[List[str]] = None,
-    ) -> List[Holding]:
+        account_ids: list[str] | None = None,
+    ) -> list[Holding]:
         """Fetch investment holdings from SnapTrade.
 
         Note: SnapTrade uses user_id + user_secret, passed as access_token in format "user_id:user_secret"
@@ -171,15 +171,15 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except httpx.HTTPStatusError as e:
             raise self._transform_error(e)
         except Exception as e:
-            raise ValueError(f"SnapTrade API error: {str(e)}")
+            raise ValueError(f"SnapTrade API error: {e!s}")
 
     async def get_transactions(
         self,
         access_token: str,
         start_date: date,
         end_date: date,
-        account_ids: Optional[List[str]] = None,
-    ) -> List[InvestmentTransaction]:
+        account_ids: list[str] | None = None,
+    ) -> list[InvestmentTransaction]:
         """Fetch investment transactions from SnapTrade.
 
         Args:
@@ -244,9 +244,9 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except httpx.HTTPStatusError as e:
             raise self._transform_error(e)
         except Exception as e:
-            raise ValueError(f"SnapTrade API error: {str(e)}")
+            raise ValueError(f"SnapTrade API error: {e!s}")
 
-    async def get_securities(self, access_token: str, security_ids: List[str]) -> List[Security]:
+    async def get_securities(self, access_token: str, security_ids: list[str]) -> list[Security]:
         """Fetch security details from SnapTrade positions.
 
         Note: SnapTrade doesn't have a dedicated securities endpoint.
@@ -267,7 +267,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             >>> for security in securities:
             ...     print(f"{security.ticker_symbol}: ${security.close_price}")
         """
-        user_id, user_secret = self._parse_access_token(access_token)
+        _user_id, _user_secret = self._parse_access_token(access_token)
 
         try:
             # Get all holdings to extract securities
@@ -282,9 +282,9 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             return list(securities_map.values())
 
         except Exception as e:
-            raise ValueError(f"SnapTrade API error: {str(e)}")
+            raise ValueError(f"SnapTrade API error: {e!s}")
 
-    async def get_investment_accounts(self, access_token: str) -> List[InvestmentAccount]:
+    async def get_investment_accounts(self, access_token: str) -> list[InvestmentAccount]:
         """Fetch investment accounts with aggregated holdings.
 
         Returns accounts with total value, cost basis, and unrealized P&L.
@@ -356,9 +356,9 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except httpx.HTTPStatusError as e:
             raise self._transform_error(e)
         except Exception as e:
-            raise ValueError(f"SnapTrade API error: {str(e)}")
+            raise ValueError(f"SnapTrade API error: {e!s}")
 
-    async def list_connections(self, access_token: str) -> List[Dict[str, Any]]:
+    async def list_connections(self, access_token: str) -> list[dict[str, Any]]:
         """List brokerage connections for a user.
 
         Returns which brokerages the user has connected (E*TRADE, Robinhood, etc.).
@@ -381,14 +381,14 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             url = f"{self.base_url}/connections"
             response = await self.client.get(url, headers=auth_headers)
             response.raise_for_status()
-            return cast(list[dict[str, Any]], await response.json())
+            return cast("list[dict[str, Any]]", await response.json())
 
         except httpx.HTTPStatusError as e:
             raise self._transform_error(e)
         except Exception as e:
-            raise ValueError(f"SnapTrade API error: {str(e)}")
+            raise ValueError(f"SnapTrade API error: {e!s}")
 
-    def get_brokerage_capabilities(self, brokerage_name: str) -> Dict[str, Any]:
+    def get_brokerage_capabilities(self, brokerage_name: str) -> dict[str, Any]:
         """Get capabilities for a specific brokerage.
 
         Important: Robinhood is READ-ONLY (no trading support).
@@ -480,7 +480,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except ValueError:
             raise ValueError("Invalid access_token format. Expected 'user_id:user_secret'")
 
-    def _transform_holding(self, snaptrade_position: Dict[str, Any], account_id: str) -> Holding:
+    def _transform_holding(self, snaptrade_position: dict[str, Any], account_id: str) -> Holding:
         """Transform SnapTrade position data to Holding model."""
         symbol_data = snaptrade_position.get("symbol", {})
 
@@ -510,7 +510,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         )
 
     def _transform_transaction(
-        self, snaptrade_tx: Dict[str, Any], account_id: str
+        self, snaptrade_tx: dict[str, Any], account_id: str
     ) -> InvestmentTransaction:
         """Transform SnapTrade transaction to InvestmentTransaction model."""
         symbol_data = snaptrade_tx.get("symbol", {})

fin_infra/markets/__init__.py

@@ -20,7 +20,10 @@ if TYPE_CHECKING:
     from fastapi import FastAPI
 
 from ..providers.base import MarketDataProvider
-from ..clients.base import MarketDataClient
+
+# Deprecated: MarketDataClient alias for backward compatibility
+# Use MarketDataProvider instead
+MarketDataClient = MarketDataProvider  # type: ignore[misc]
 
 
 def easy_market(
@@ -98,7 +101,7 @@ def easy_market(
 
 
 def add_market_data(
-    app: "FastAPI",
+    app: FastAPI,
     *,
     provider: str | MarketDataProvider | None = None,
     prefix: str = "/market",
@@ -178,7 +181,6 @@ def add_market_data(
     See Also:
         - easy_market(): For standalone provider usage without FastAPI
         - docs/market-data.md: API documentation and examples
-        - docs/adr/0004-market-data-integration.md: Architecture decisions
     """
     from fastapi import HTTPException, Query
 

fin_infra/models/__init__.py

@@ -1,21 +1,21 @@
 from .accounts import Account, AccountType
-from .transactions import Transaction
-from .quotes import Quote
-from .money import Money
-from .candle import Candle
-from .brokerage import Order, Position, PortfolioHistory
 from .brokerage import Account as BrokerageAccount  # Avoid name conflict
+from .brokerage import Order, PortfolioHistory, Position
+from .candle import Candle
+from .money import Money
+from .quotes import Quote
 from .tax import (
+    CryptoTaxReport,
+    CryptoTransaction,
     TaxDocument,
-    TaxFormW2,
-    TaxForm1099INT,
-    TaxForm1099DIV,
     TaxForm1099B,
+    TaxForm1099DIV,
+    TaxForm1099INT,
     TaxForm1099MISC,
-    CryptoTransaction,
-    CryptoTaxReport,
+    TaxFormW2,
     TaxLiability,
 )
+from .transactions import Transaction
 
 __all__ = [
     "Account",

fin_infra/models/brokerage.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 
-from decimal import Decimal
 from datetime import datetime
+from decimal import Decimal
 from typing import Literal
+
 from pydantic import BaseModel, Field
 
 

fin_infra/models/candle.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from decimal import Decimal
+
 from pydantic import BaseModel, field_validator
 
 

fin_infra/models/money.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from decimal import Decimal
+
 from pydantic import BaseModel, field_validator
 
 

fin_infra/models/quotes.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from decimal import Decimal
+
 from pydantic import BaseModel, field_validator
 
 
@@ -16,5 +17,5 @@ class Quote(BaseModel):
     def _ensure_tzaware(cls, v: datetime) -> datetime:
         # Normalize to timezone-aware (UTC) for consistency
         if v.tzinfo is None:
-            return v.replace(tzinfo=timezone.utc)
-        return v.astimezone(timezone.utc)
+            return v.replace(tzinfo=UTC)
+        return v.astimezone(UTC)

fin_infra/models/tax.py

@@ -37,7 +37,8 @@ Example:
 
 from datetime import date, datetime
 from decimal import Decimal
-from pydantic import BaseModel, Field, ConfigDict
+
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class TaxDocument(BaseModel):

fin_infra/models/transactions.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
-from datetime import date  # noqa: F401 - used in type annotation
+from datetime import date
 from decimal import Decimal
-from typing import Optional
 
 from pydantic import BaseModel, field_validator
 
@@ -19,8 +18,8 @@ class Transaction(BaseModel):
     date: date
     amount: Decimal
     currency: str = "USD"
-    description: Optional[str] = None
-    category: Optional[str] = None
+    description: str | None = None
+    category: str | None = None
 
     @field_validator("amount", mode="before")
     @classmethod

fin_infra/net_worth/__init__.py

@@ -4,6 +4,13 @@ Net Worth Tracking Module
 Calculates net worth by aggregating balances from multiple financial providers
 (banking, brokerage, crypto) with historical snapshots and change detection.
 
+**Feature Status**:
+    ✅ STABLE: Core calculation (works with provided data)
+    ✅ STABLE: Banking integration (Plaid, Teller)
+    ⚠️  INTEGRATION: Brokerage integration (requires provider setup)
+    ⚠️  INTEGRATION: Crypto integration (requires provider setup)
+    ⚠️  INTEGRATION: Currency conversion (pass exchange_rate manually)
+
 **Key Features**:
 - Multi-provider aggregation (banking + brokerage + crypto)
 - Currency normalization (all currencies → USD)