fin-infra 0.1.66__py3-none-any.whl → 0.1.68__py3-none-any.whl

fin_infra/investments/models.py

@@ -27,7 +27,7 @@ from pydantic import BaseModel, ConfigDict, Field, computed_field
 
 class SecurityType(str, Enum):
     """Security type classification.
-    
+
     Categories:
     - equity: Common stock (AAPL, GOOGL, etc.)
     - etf: Exchange-traded fund (SPY, QQQ, etc.)
@@ -49,7 +49,7 @@ class SecurityType(str, Enum):
 
 class TransactionType(str, Enum):
     """Investment transaction type.
-    
+
     Categories:
     - buy: Purchase of security
     - sell: Sale of security
@@ -79,10 +79,10 @@ class TransactionType(str, Enum):
 
 class Security(BaseModel):
     """Security details (stock, bond, ETF, etc.).
-    
+
     Represents a tradable security with identifying information and current market data.
     Normalized across providers (Plaid, SnapTrade).
-    
+
     Example:
         >>> security = Security(
         ...     security_id="plaid_sec_123",
@@ -129,7 +129,9 @@ class Security(BaseModel):
     # Basic info
     name: str = Field(..., description="Security name")
     type: SecurityType = Field(..., description="Security type (equity, etf, bond, etc.)")
-    sector: Optional[str] = Field(None, description="Sector classification (Technology, Healthcare)")
+    sector: Optional[str] = Field(
+        None, description="Sector classification (Technology, Healthcare)"
+    )
 
     # Market data
     close_price: Optional[Decimal] = Field(None, ge=0, description="Latest closing price")
@@ -140,10 +142,10 @@ class Security(BaseModel):
 
 class Holding(BaseModel):
     """Investment holding with current value and cost basis.
-    
+
     Represents a position in a specific security within an investment account.
     Includes quantity, current value, cost basis, and calculated P&L.
-    
+
     Example:
         >>> holding = Holding(
         ...     account_id="acct_123",
@@ -193,8 +195,12 @@ class Holding(BaseModel):
     # Position data
     quantity: Decimal = Field(..., ge=0, description="Number of shares/units held")
     institution_price: Decimal = Field(..., ge=0, description="Current price per share")
-    institution_value: Decimal = Field(..., ge=0, description="Current market value (quantity × price)")
-    cost_basis: Optional[Decimal] = Field(None, ge=0, description="Total cost basis (original purchase price)")
+    institution_value: Decimal = Field(
+        ..., ge=0, description="Current market value (quantity × price)"
+    )
+    cost_basis: Optional[Decimal] = Field(
+        None, ge=0, description="Total cost basis (original purchase price)"
+    )
 
     # Additional data
     currency: str = Field("USD", description="Currency code")
@@ -240,10 +246,10 @@ class Holding(BaseModel):
 
 class InvestmentTransaction(BaseModel):
     """Investment transaction (buy, sell, dividend, etc.).
-    
+
     Represents a single transaction in an investment account.
     Used to calculate realized gains and track transaction history.
-    
+
     Example:
         >>> transaction = InvestmentTransaction(
         ...     transaction_id="tx_123",
@@ -295,7 +301,9 @@ class InvestmentTransaction(BaseModel):
     # Transaction details
     transaction_date: date = Field(..., alias="date", description="Transaction date")
     name: str = Field(..., description="Transaction description")
-    transaction_type: TransactionType = Field(..., alias="type", description="Transaction type (buy, sell, dividend)")
+    transaction_type: TransactionType = Field(
+        ..., alias="type", description="Transaction type (buy, sell, dividend)"
+    )
     subtype: Optional[str] = Field(None, description="Provider-specific subtype")
 
     # Amounts
@@ -311,10 +319,10 @@ class InvestmentTransaction(BaseModel):
 
 class InvestmentAccount(BaseModel):
     """Investment account with aggregated holdings and metrics.
-    
+
     Represents a complete investment account with all holdings, balances, and P&L.
     Includes calculated fields for total value, cost basis, and unrealized gains.
-    
+
     Example:
         >>> account = InvestmentAccount(
         ...     account_id="acct_123",
@@ -340,7 +348,11 @@ class InvestmentAccount(BaseModel):
                 "holdings": [
                     {
                         "account_id": "acct_abc123",
-                        "security": {"ticker_symbol": "AAPL", "name": "Apple Inc.", "type": "equity"},
+                        "security": {
+                            "ticker_symbol": "AAPL",
+                            "name": "Apple Inc.",
+                            "type": "equity",
+                        },
                         "quantity": 10.5,
                         "institution_price": 150.25,
                         "institution_value": 1577.63,
@@ -436,10 +448,10 @@ class InvestmentAccount(BaseModel):
 
 class AssetAllocation(BaseModel):
     """Asset allocation breakdown by security type and sector.
-    
+
     Provides percentage breakdown of portfolio by security type and sector.
     Used for diversification analysis and portfolio visualization.
-    
+
     Example:
         >>> allocation = AssetAllocation(
         ...     by_security_type={

fin_infra/investments/providers/base.py

@@ -77,9 +77,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:
@@ -97,9 +95,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:
@@ -176,8 +172,7 @@ class InvestmentProvider(ABC):
         }
 
         by_sector_percent = {
-            sector: round((value / total_value) * 100, 2)
-            for sector, value in sector_values.items()
+            sector: round((value / total_value) * 100, 2) for sector, value in sector_values.items()
         }
 
         cash_percent = round((cash_value / total_value) * 100, 2)

fin_infra/investments/providers/plaid.py

@@ -135,9 +135,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
             if account_ids:
                 request.options = {"account_ids": account_ids}
 
-            response: InvestmentsHoldingsGetResponse = (
-                self.client.investments_holdings_get(request)
-            )
+            response: InvestmentsHoldingsGetResponse = self.client.investments_holdings_get(request)
 
             # Build security lookup map
             securities_map = {
@@ -203,8 +201,8 @@ class PlaidInvestmentProvider(InvestmentProvider):
             if account_ids:
                 request.options = {"account_ids": account_ids}
 
-            response: InvestmentsTransactionsGetResponse = (
-                self.client.investments_transactions_get(request)
+            response: InvestmentsTransactionsGetResponse = self.client.investments_transactions_get(
+                request
             )
 
             # Build security lookup map
@@ -228,9 +226,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.
@@ -253,9 +249,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         """
         try:
             request = InvestmentsHoldingsGetRequest(access_token=access_token)
-            response: InvestmentsHoldingsGetResponse = (
-                self.client.investments_holdings_get(request)
-            )
+            response: InvestmentsHoldingsGetResponse = self.client.investments_holdings_get(request)
 
             # Filter securities by requested IDs
             securities = []
@@ -270,9 +264,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.
@@ -294,9 +286,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         """
         try:
             request = InvestmentsHoldingsGetRequest(access_token=access_token)
-            response: InvestmentsHoldingsGetResponse = (
-                self.client.investments_holdings_get(request)
-            )
+            response: InvestmentsHoldingsGetResponse = self.client.investments_holdings_get(request)
 
             # Build security lookup map
             securities_map = {
@@ -317,11 +307,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
                     if account_id not in accounts_map:
                         # Find account metadata
                         plaid_account = next(
-                            (
-                                acc
-                                for acc in response.accounts
-                                if acc.account_id == account_id
-                            ),
+                            (acc for acc in response.accounts if acc.account_id == account_id),
                             None,
                         )
                         accounts_map[account_id] = {
@@ -339,12 +325,16 @@ class PlaidInvestmentProvider(InvestmentProvider):
 
                 investment_account = InvestmentAccount(
                     account_id=account_id,
-                    name=account_dict.get("name", account_dict.get("official_name", "Unknown Account")),
+                    name=account_dict.get(
+                        "name", account_dict.get("official_name", "Unknown Account")
+                    ),
                     type=account_dict.get("type", "investment"),
                     subtype=account_dict.get("subtype"),
                     balances={
                         "current": Decimal(str(account_dict.get("balances", {}).get("current", 0))),
-                        "available": Decimal(str(account_dict.get("balances", {}).get("available") or 0)),
+                        "available": Decimal(
+                            str(account_dict.get("balances", {}).get("available") or 0)
+                        ),
                     },
                     holdings=holdings,
                 )
@@ -362,7 +352,7 @@ class PlaidInvestmentProvider(InvestmentProvider):
         # Handle close_price - Plaid may return None for securities without recent pricing
         close_price_raw = plaid_security.get("close_price")
         close_price = Decimal(str(close_price_raw)) if close_price_raw is not None else Decimal("0")
-        
+
         return Security(
             security_id=plaid_security["security_id"],
             cusip=plaid_security.get("cusip"),
@@ -378,9 +368,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"],
@@ -388,7 +376,9 @@ class PlaidInvestmentProvider(InvestmentProvider):
             quantity=Decimal(str(plaid_holding.get("quantity", 0))),
             institution_price=Decimal(str(plaid_holding.get("institution_price", 0))),
             institution_value=Decimal(str(plaid_holding.get("institution_value", 0))),
-            cost_basis=Decimal(str(plaid_holding.get("cost_basis"))) if plaid_holding.get("cost_basis") else None,
+            cost_basis=Decimal(str(plaid_holding.get("cost_basis")))
+            if plaid_holding.get("cost_basis")
+            else None,
             currency=plaid_holding.get("iso_currency_code", "USD"),
             unofficial_currency_code=plaid_holding.get("unofficial_currency_code"),
         )

fin_infra/investments/providers/snaptrade.py

@@ -77,9 +77,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             ValueError: If client_id or consumer_key is missing
         """
         if not client_id or not consumer_key:
-            raise ValueError(
-                "client_id and consumer_key are required for SnapTrade provider"
-            )
+            raise ValueError("client_id and consumer_key are required for SnapTrade provider")
 
         self.client_id = client_id
         self.consumer_key = consumer_key
@@ -97,15 +95,15 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
 
     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.
         URL params are logged in access logs, browser history, and proxy logs.
         Headers are not logged by default in most web servers.
-        
+
         Args:
             user_id: SnapTrade user ID
             user_secret: SnapTrade user secret (sensitive!)
-            
+
         Returns:
             Dict with authentication headers
         """
@@ -158,9 +156,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             all_holdings = []
             for account in accounts:
                 account_id = account["id"]
-                positions_url = (
-                    f"{self.base_url}/accounts/{account_id}/positions"
-                )
+                positions_url = f"{self.base_url}/accounts/{account_id}/positions"
                 pos_response = await self.client.get(positions_url, headers=auth_headers)
                 pos_response.raise_for_status()
                 positions = await pos_response.json()
@@ -226,15 +222,15 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             all_transactions = []
             for account in accounts:
                 account_id = account["id"]
-                transactions_url = (
-                    f"{self.base_url}/accounts/{account_id}/transactions"
-                )
+                transactions_url = f"{self.base_url}/accounts/{account_id}/transactions"
                 # Date params are non-sensitive, only auth goes in headers
                 tx_params = {
                     "startDate": start_date.isoformat(),
                     "endDate": end_date.isoformat(),
                 }
-                tx_response = await self.client.get(transactions_url, params=tx_params, headers=auth_headers)
+                tx_response = await self.client.get(
+                    transactions_url, params=tx_params, headers=auth_headers
+                )
                 tx_response.raise_for_status()
                 transactions = await tx_response.json()
 
@@ -250,9 +246,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except Exception as e:
             raise ValueError(f"SnapTrade API error: {str(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 SnapTrade positions.
 
         Note: SnapTrade doesn't have a dedicated securities endpoint.
@@ -290,9 +284,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except Exception as e:
             raise ValueError(f"SnapTrade API error: {str(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.
@@ -328,9 +320,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
                 account_id = account["id"]
 
                 # Get positions for this account
-                positions_url = (
-                    f"{self.base_url}/accounts/{account_id}/positions"
-                )
+                positions_url = f"{self.base_url}/accounts/{account_id}/positions"
                 pos_response = await self.client.get(positions_url, headers=auth_headers)
                 pos_response.raise_for_status()
                 positions = await pos_response.json()
@@ -368,9 +358,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         except Exception as e:
             raise ValueError(f"SnapTrade API error: {str(e)}")
 
-    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.).
@@ -490,16 +478,12 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
             user_id, user_secret = access_token.split(":", 1)
             return user_id, user_secret
         except ValueError:
-            raise ValueError(
-                "Invalid access_token format. Expected 'user_id:user_secret'"
-            )
+            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", {})
-        
+
         # Create Security from symbol data
         security = Security(
             security_id=symbol_data.get("id", symbol_data.get("symbol", "")),
@@ -513,9 +497,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
         # SnapTrade uses "average_purchase_price" for cost basis
         avg_price = snaptrade_position.get("average_purchase_price")
         quantity = Decimal(str(snaptrade_position.get("units", 0)))
-        cost_basis = (
-            Decimal(str(avg_price)) * quantity if avg_price is not None else None
-        )
+        cost_basis = Decimal(str(avg_price)) * quantity if avg_price is not None else None
 
         return Holding(
             account_id=account_id,
@@ -532,7 +514,7 @@ class SnapTradeInvestmentProvider(InvestmentProvider):
     ) -> InvestmentTransaction:
         """Transform SnapTrade transaction to InvestmentTransaction model."""
         symbol_data = snaptrade_tx.get("symbol", {})
-        
+
         # Create Security from symbol data
         security = Security(
             security_id=symbol_data.get("id", symbol_data.get("symbol", "")),

fin_infra/markets/__init__.py

@@ -93,7 +93,7 @@ def easy_market(
 
     else:
         raise ValueError(
-            f"Unknown market data provider: {provider_name}. " f"Supported: alphavantage, yahoo"
+            f"Unknown market data provider: {provider_name}. Supported: alphavantage, yahoo"
         )
 
 
@@ -231,7 +231,9 @@ def add_market_data(
                     candles_list.append(candle.dict())
                 else:
                     # Cast to dict for type compatibility
-                    candles_list.append(dict(candle) if hasattr(candle, "__iter__") else {"data": candle})
+                    candles_list.append(
+                        dict(candle) if hasattr(candle, "__iter__") else {"data": candle}
+                    )
             return {"candles": candles_list}
         except Exception as e:
             raise HTTPException(status_code=400, detail=str(e))

fin_infra/models/accounts.py

@@ -18,10 +18,11 @@ class AccountType(str, Enum):
 
 class Account(BaseModel):
     """Financial account model.
-    
+
     Uses Decimal for balance fields to prevent floating-point precision errors
     in financial calculations (e.g., $0.01 + $0.02 != $0.03 with float).
     """
+
     id: str
     name: str
     type: AccountType

fin_infra/models/transactions.py

@@ -9,10 +9,11 @@ from pydantic import BaseModel, field_validator
 
 class Transaction(BaseModel):
     """Financial transaction model.
-    
+
     Uses Decimal for amount to prevent floating-point precision errors
     in financial calculations (e.g., $0.01 + $0.02 != $0.03 with float).
     """
+
     id: str
     account_id: str
     date: date

fin_infra/net_worth/calculator.py

@@ -30,7 +30,6 @@ print(f"Net Worth: ${net_worth:,.2f}")  # $55,000.00
 ```
 """
 
-
 from fin_infra.net_worth.models import (
     AssetAllocation,
     AssetCategory,
@@ -100,18 +99,19 @@ def calculate_net_worth(
 
     Returns:
         Net worth in base currency
-        
+
     Raises:
         ValueError: If assets or liabilities contain non-base currencies and no
             exchange rate conversion is available. This prevents silent data loss.
     """
     import logging
+
     logger = logging.getLogger(__name__)
-    
+
     # Collect any non-base currency items for error reporting
     non_base_assets: list[tuple[str, str, float]] = []
     non_base_liabilities: list[tuple[str, str, float]] = []
-    
+
     # Sum all assets (use market_value if available, otherwise balance)
     total_assets = 0.0
     for asset in assets:
@@ -130,7 +130,9 @@ def calculate_net_worth(
     for liability in liabilities:
         # Check for non-base currency
         if liability.currency != base_currency:
-            non_base_liabilities.append((liability.name or liability.account_id, liability.currency, liability.balance))
+            non_base_liabilities.append(
+                (liability.name or liability.account_id, liability.currency, liability.balance)
+            )
             continue
 
         total_liabilities += liability.balance
@@ -143,7 +145,7 @@ def calculate_net_worth(
             items_msg.append(f"Assets: {non_base_assets}")
         if non_base_liabilities:
             items_msg.append(f"Liabilities: {non_base_liabilities}")
-        
+
         error_msg = (
             f"Cannot calculate net worth: found accounts in non-{base_currency} currencies. "
             f"Currency conversion not yet implemented. {'; '.join(items_msg)}. "

fin_infra/net_worth/ease.py

@@ -380,7 +380,7 @@ def easy_net_worth(
             from ai_infra.llm.llm import LLM
         except ImportError:
             raise ImportError(
-                "LLM features require ai-infra package. " "Install with: pip install ai-infra"
+                "LLM features require ai-infra package. Install with: pip install ai-infra"
             )
 
         cache = None
@@ -401,7 +401,7 @@ def easy_net_worth(
 
         if not model_name:
             raise ValueError(
-                f"Unknown llm_provider: {llm_provider}. " f"Use 'google', 'openai', or 'anthropic'"
+                f"Unknown llm_provider: {llm_provider}. Use 'google', 'openai', or 'anthropic'"
             )
 
         # Create shared LLM instance

fin_infra/net_worth/insights.py

@@ -479,13 +479,13 @@ class NetWorthInsightsGenerator:
 
         user_prompt = f"""Analyze wealth trends:
 
-Current net worth: ${current['total_net_worth']:,.0f}
-Previous net worth: ${previous['total_net_worth']:,.0f}
+Current net worth: ${current["total_net_worth"]:,.0f}
+Previous net worth: ${previous["total_net_worth"]:,.0f}
 Period: {period}
 Change: ${change_amount:,.0f} ({change_percent:.1%})
 
-Assets: ${current['total_assets']:,.0f}
-Liabilities: ${current['total_liabilities']:,.0f}
+Assets: ${current["total_assets"]:,.0f}
+Liabilities: ${current["total_liabilities"]:,.0f}
 
 Identify drivers of change, risk factors, and recommendations."""
 

fin_infra/normalization/__init__.py

@@ -145,7 +145,9 @@ def add_normalization(
     @router.get("/convert")
     async def convert_currency(
         amount: float = Query(..., description="Amount to convert"),
-        from_currency: str = Query(..., alias="from", description="Source currency code (e.g., USD)"),
+        from_currency: str = Query(
+            ..., alias="from", description="Source currency code (e.g., USD)"
+        ),
         to_currency: str = Query(..., alias="to", description="Target currency code (e.g., EUR)"),
     ):
         """Convert amount between currencies."""