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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (108) hide show
  1. fin_infra/__init__.py +53 -3
  2. fin_infra/analytics/__init__.py +13 -2
  3. fin_infra/analytics/add.py +21 -21
  4. fin_infra/analytics/ease.py +19 -20
  5. fin_infra/analytics/portfolio.py +6 -6
  6. fin_infra/analytics/projections.py +1 -3
  7. fin_infra/banking/__init__.py +27 -28
  8. fin_infra/banking/history.py +8 -9
  9. fin_infra/banking/utils.py +27 -26
  10. fin_infra/brokerage/__init__.py +22 -24
  11. fin_infra/budgets/__init__.py +3 -3
  12. fin_infra/budgets/add.py +16 -17
  13. fin_infra/budgets/alerts.py +3 -3
  14. fin_infra/budgets/tracker.py +2 -2
  15. fin_infra/cashflows/__init__.py +3 -3
  16. fin_infra/cashflows/core.py +1 -1
  17. fin_infra/categorization/add.py +2 -3
  18. fin_infra/categorization/engine.py +17 -14
  19. fin_infra/categorization/llm_layer.py +7 -6
  20. fin_infra/categorization/rules.py +2 -4
  21. fin_infra/categorization/taxonomy.py +2 -2
  22. fin_infra/chat/__init__.py +6 -6
  23. fin_infra/chat/planning.py +0 -1
  24. fin_infra/cli/cmds/scaffold_cmds.py +10 -11
  25. fin_infra/clients/__init__.py +23 -1
  26. fin_infra/clients/base.py +1 -1
  27. fin_infra/clients/plaid.py +2 -2
  28. fin_infra/compliance/__init__.py +5 -4
  29. fin_infra/credit/add.py +6 -7
  30. fin_infra/credit/experian/auth.py +2 -2
  31. fin_infra/credit/experian/client.py +1 -1
  32. fin_infra/credit/experian/provider.py +4 -4
  33. fin_infra/crypto/__init__.py +7 -9
  34. fin_infra/crypto/insights.py +4 -3
  35. fin_infra/documents/add.py +6 -8
  36. fin_infra/documents/analysis.py +9 -9
  37. fin_infra/documents/ease.py +14 -14
  38. fin_infra/documents/models.py +4 -4
  39. fin_infra/documents/ocr.py +7 -7
  40. fin_infra/documents/storage.py +21 -13
  41. fin_infra/exceptions.py +0 -1
  42. fin_infra/goals/__init__.py +8 -8
  43. fin_infra/goals/add.py +36 -36
  44. fin_infra/goals/funding.py +4 -6
  45. fin_infra/goals/management.py +2 -3
  46. fin_infra/goals/milestones.py +1 -2
  47. fin_infra/goals/models.py +7 -11
  48. fin_infra/insights/__init__.py +6 -3
  49. fin_infra/insights/aggregator.py +1 -1
  50. fin_infra/investments/__init__.py +1 -1
  51. fin_infra/investments/add.py +23 -23
  52. fin_infra/investments/models.py +5 -5
  53. fin_infra/investments/providers/base.py +8 -9
  54. fin_infra/investments/providers/plaid.py +52 -26
  55. fin_infra/investments/providers/snaptrade.py +19 -19
  56. fin_infra/markets/__init__.py +5 -3
  57. fin_infra/models/__init__.py +10 -10
  58. fin_infra/models/brokerage.py +2 -1
  59. fin_infra/models/candle.py +1 -0
  60. fin_infra/models/money.py +1 -0
  61. fin_infra/models/quotes.py +4 -3
  62. fin_infra/models/tax.py +2 -1
  63. fin_infra/models/transactions.py +3 -4
  64. fin_infra/net_worth/__init__.py +7 -0
  65. fin_infra/net_worth/aggregator.py +4 -2
  66. fin_infra/net_worth/insights.py +0 -1
  67. fin_infra/normalization/__init__.py +2 -2
  68. fin_infra/normalization/providers/exchangerate.py +5 -5
  69. fin_infra/obs/classifier.py +1 -1
  70. fin_infra/providers/banking/plaid_client.py +5 -5
  71. fin_infra/providers/banking/teller_client.py +7 -6
  72. fin_infra/providers/base.py +27 -2
  73. fin_infra/providers/brokerage/alpaca.py +3 -3
  74. fin_infra/providers/market/alphavantage.py +6 -11
  75. fin_infra/providers/market/ccxt_crypto.py +19 -3
  76. fin_infra/providers/market/coingecko.py +5 -6
  77. fin_infra/providers/market/yahoo.py +23 -8
  78. fin_infra/providers/tax/__init__.py +1 -1
  79. fin_infra/providers/tax/irs.py +1 -1
  80. fin_infra/providers/tax/mock.py +5 -5
  81. fin_infra/providers/tax/taxbit.py +1 -1
  82. fin_infra/recurring/__init__.py +6 -6
  83. fin_infra/recurring/add.py +5 -4
  84. fin_infra/recurring/detector.py +7 -7
  85. fin_infra/recurring/detectors_llm.py +6 -6
  86. fin_infra/recurring/ease.py +2 -4
  87. fin_infra/recurring/insights.py +13 -13
  88. fin_infra/recurring/normalizer.py +1 -1
  89. fin_infra/recurring/normalizers.py +4 -4
  90. fin_infra/recurring/summary.py +13 -15
  91. fin_infra/scaffold/budgets.py +9 -9
  92. fin_infra/scaffold/goals.py +5 -5
  93. fin_infra/security/__init__.py +8 -8
  94. fin_infra/security/encryption.py +6 -6
  95. fin_infra/security/models.py +7 -7
  96. fin_infra/security/pii_filter.py +6 -6
  97. fin_infra/settings.py +2 -1
  98. fin_infra/tax/__init__.py +1 -1
  99. fin_infra/tax/add.py +3 -2
  100. fin_infra/tax/tlh.py +5 -5
  101. fin_infra/utils/http.py +4 -3
  102. fin_infra/utils/retry.py +2 -1
  103. {fin_infra-0.1.69.dist-info → fin_infra-0.1.82.dist-info}/METADATA +14 -8
  104. fin_infra-0.1.82.dist-info/RECORD +180 -0
  105. fin_infra-0.1.69.dist-info/RECORD +0 -180
  106. {fin_infra-0.1.69.dist-info → fin_infra-0.1.82.dist-info}/LICENSE +0 -0
  107. {fin_infra-0.1.69.dist-info → fin_infra-0.1.82.dist-info}/WHEEL +0 -0
  108. {fin_infra-0.1.69.dist-info → fin_infra-0.1.82.dist-info}/entry_points.txt +0 -0
fin_infra/__init__.py CHANGED
@@ -1,9 +1,45 @@
1
- """fin_infra: Financial infrastructure toolkit.
1
+ """fin_infra: Financial Infrastructure Toolkit.
2
2
 
3
- Public surface is intentionally small at this stage. Import from submodules for
4
- specific domains (clients, models, markets, credit).
3
+ A comprehensive financial infrastructure library providing:
4
+ - Banking integration (Plaid, Teller, MX)
5
+ - Brokerage integration (Alpaca, Interactive Brokers)
6
+ - Market data (stocks, crypto, forex)
7
+ - Credit scores (Experian, Equifax, TransUnion)
8
+ - Financial calculations (NPV, IRR, PMT, FV, PV)
9
+ - Portfolio analytics (returns, allocation, benchmarking)
10
+ - Transaction categorization (rule-based + ML)
11
+ - Budget management and cash flow analysis
12
+ - Net worth tracking and goal management
13
+
14
+ Example:
15
+ from fin_infra.banking import easy_banking
16
+ from fin_infra.markets import easy_market
17
+
18
+ banking = easy_banking()
19
+ market = easy_market()
20
+ quote = market.quote("AAPL")
5
21
  """
6
22
 
23
+ from __future__ import annotations
24
+
25
+ # Core modules - can be imported as `from fin_infra import banking`
26
+ from . import (
27
+ analytics,
28
+ banking,
29
+ brokerage,
30
+ budgets,
31
+ cashflows,
32
+ categorization,
33
+ credit,
34
+ crypto,
35
+ investments,
36
+ markets,
37
+ net_worth,
38
+ recurring,
39
+ tax,
40
+ )
41
+
42
+ # Base exceptions
7
43
  from .exceptions import (
8
44
  FinInfraError,
9
45
  ProviderError,
@@ -14,6 +50,20 @@ from .version import __version__
14
50
 
15
51
  __all__ = [
16
52
  "__version__",
53
+ # Core modules
54
+ "analytics",
55
+ "banking",
56
+ "brokerage",
57
+ "budgets",
58
+ "cashflows",
59
+ "categorization",
60
+ "credit",
61
+ "crypto",
62
+ "investments",
63
+ "markets",
64
+ "net_worth",
65
+ "recurring",
66
+ "tax",
17
67
  # Base errors
18
68
  "FinInfraError",
19
69
  "ProviderError",
@@ -7,6 +7,16 @@ This module provides comprehensive financial analytics capabilities including:
7
7
  - Portfolio analytics (returns, allocation, benchmarking)
8
8
  - Growth projections (net worth forecasting with scenarios)
9
9
 
10
+ Feature Status:
11
+ ✅ STABLE: Core calculation functions (all analytics work with provided data)
12
+ ⚠️ INTEGRATION: Auto-fetching from providers requires setup:
13
+ - Banking provider for transaction data
14
+ - Brokerage provider for investment data
15
+ - Categorization for expense categorization
16
+
17
+ When providers aren't configured, functions accept data directly or return
18
+ sensible placeholder values for testing/development.
19
+
10
20
  Serves multiple use cases:
11
21
  - Personal finance apps (cash flow, savings tracking)
12
22
  - Wealth management platforms (portfolio analytics, projections)
@@ -45,10 +55,11 @@ Dependencies:
45
55
 
46
56
  from __future__ import annotations
47
57
 
48
- # Import actual implementations
49
- from .ease import easy_analytics, AnalyticsEngine
50
58
  from .add import add_analytics
51
59
 
60
+ # Import actual implementations
61
+ from .ease import AnalyticsEngine, easy_analytics
62
+
52
63
  __all__ = [
53
64
  "easy_analytics",
54
65
  "add_analytics",
@@ -7,7 +7,7 @@ MUST use svc-infra dual routers (user_router) - NEVER generic APIRouter.
7
7
  from __future__ import annotations
8
8
 
9
9
  from datetime import datetime
10
- from typing import TYPE_CHECKING, Optional
10
+ from typing import TYPE_CHECKING
11
11
 
12
12
  from fastapi import HTTPException, Query
13
13
  from pydantic import BaseModel, Field
@@ -15,15 +15,15 @@ from pydantic import BaseModel, Field
15
15
  if TYPE_CHECKING:
16
16
  from fastapi import FastAPI
17
17
 
18
- from .ease import easy_analytics, AnalyticsEngine
18
+ from .ease import AnalyticsEngine, easy_analytics
19
19
  from .models import (
20
+ BenchmarkComparison,
20
21
  CashFlowAnalysis,
21
- SavingsRateData,
22
- SpendingInsight,
22
+ GrowthProjection,
23
23
  PersonalizedSpendingAdvice,
24
24
  PortfolioMetrics,
25
- BenchmarkComparison,
26
- GrowthProjection,
25
+ SavingsRateData,
26
+ SpendingInsight,
27
27
  )
28
28
 
29
29
 
@@ -33,15 +33,15 @@ class NetWorthForecastRequest(BaseModel):
33
33
 
34
34
  user_id: str = Field(..., description="User identifier")
35
35
  years: int = Field(default=30, ge=1, le=50, description="Projection years (1-50)")
36
- initial_net_worth: Optional[float] = Field(None, description="Override initial net worth")
37
- annual_contribution: Optional[float] = Field(None, description="Annual savings contribution")
38
- conservative_return: Optional[float] = Field(
36
+ initial_net_worth: float | None = Field(None, description="Override initial net worth")
37
+ annual_contribution: float | None = Field(None, description="Annual savings contribution")
38
+ conservative_return: float | None = Field(
39
39
  None, description="Conservative return rate (e.g., 0.05 = 5%)"
40
40
  )
41
- moderate_return: Optional[float] = Field(
41
+ moderate_return: float | None = Field(
42
42
  None, description="Moderate return rate (e.g., 0.07 = 7%)"
43
43
  )
44
- aggressive_return: Optional[float] = Field(
44
+ aggressive_return: float | None = Field(
45
45
  None, description="Aggressive return rate (e.g., 0.10 = 10%)"
46
46
  )
47
47
 
@@ -49,7 +49,7 @@ class NetWorthForecastRequest(BaseModel):
49
49
  def add_analytics(
50
50
  app: FastAPI,
51
51
  prefix: str = "/analytics",
52
- provider: Optional[AnalyticsEngine] = None,
52
+ provider: AnalyticsEngine | None = None,
53
53
  include_in_schema: bool = True,
54
54
  ) -> AnalyticsEngine:
55
55
  """Add analytics endpoints to FastAPI application.
@@ -124,9 +124,9 @@ def add_analytics(
124
124
  )
125
125
  async def get_cash_flow(
126
126
  user_id: str,
127
- start_date: Optional[datetime] = None,
128
- end_date: Optional[datetime] = None,
129
- period_days: Optional[int] = None,
127
+ start_date: datetime | None = None,
128
+ end_date: datetime | None = None,
129
+ period_days: int | None = None,
130
130
  ) -> CashFlowAnalysis:
131
131
  """
132
132
  Calculate cash flow analysis for a user.
@@ -164,7 +164,7 @@ def add_analytics(
164
164
  )
165
165
  async def get_spending_insights(
166
166
  user_id: str,
167
- period_days: Optional[int] = None,
167
+ period_days: int | None = None,
168
168
  include_trends: bool = True,
169
169
  ) -> SpendingInsight:
170
170
  """
@@ -186,7 +186,7 @@ def add_analytics(
186
186
  )
187
187
  async def get_spending_advice(
188
188
  user_id: str,
189
- period_days: Optional[int] = None,
189
+ period_days: int | None = None,
190
190
  ) -> PersonalizedSpendingAdvice:
191
191
  """
192
192
  Generate personalized spending advice using AI.
@@ -206,11 +206,11 @@ def add_analytics(
206
206
  )
207
207
  async def get_portfolio_metrics(
208
208
  user_id: str,
209
- accounts: Optional[list[str]] = None,
209
+ accounts: list[str] | None = None,
210
210
  with_holdings: bool = Query(
211
211
  False, description="Use real holdings data from investment provider for accurate P/L"
212
212
  ),
213
- access_token: Optional[str] = Query(
213
+ access_token: str | None = Query(
214
214
  None, description="Investment provider access token (required if with_holdings=true)"
215
215
  ),
216
216
  ) -> PortfolioMetrics:
@@ -286,9 +286,9 @@ def add_analytics(
286
286
  )
287
287
  async def get_benchmark_comparison(
288
288
  user_id: str,
289
- benchmark: Optional[str] = None,
289
+ benchmark: str | None = None,
290
290
  period: str = "1y",
291
- accounts: Optional[list[str]] = None,
291
+ accounts: list[str] | None = None,
292
292
  ) -> BenchmarkComparison:
293
293
  """
294
294
  Compare portfolio to benchmark (e.g., SPY, VTI).
@@ -16,23 +16,22 @@ Typical usage:
16
16
  portfolio = await analytics.portfolio_metrics(user_id="user123")
17
17
  """
18
18
 
19
- from typing import Optional
20
19
  from datetime import datetime, timedelta
21
20
 
22
21
  from .cash_flow import calculate_cash_flow
23
- from .savings import calculate_savings_rate, SavingsDefinition
24
- from .spending import analyze_spending, generate_spending_insights
25
- from .portfolio import calculate_portfolio_metrics, compare_to_benchmark
26
- from .projections import project_net_worth, calculate_compound_interest
27
22
  from .models import (
23
+ BenchmarkComparison,
28
24
  CashFlowAnalysis,
29
- SavingsRateData,
30
- SpendingInsight,
25
+ GrowthProjection,
31
26
  PersonalizedSpendingAdvice,
32
27
  PortfolioMetrics,
33
- BenchmarkComparison,
34
- GrowthProjection,
28
+ SavingsRateData,
29
+ SpendingInsight,
35
30
  )
31
+ from .portfolio import calculate_portfolio_metrics, compare_to_benchmark
32
+ from .projections import calculate_compound_interest, project_net_worth
33
+ from .savings import SavingsDefinition, calculate_savings_rate
34
+ from .spending import analyze_spending, generate_spending_insights
36
35
 
37
36
 
38
37
  class AnalyticsEngine:
@@ -93,9 +92,9 @@ class AnalyticsEngine:
93
92
  self,
94
93
  user_id: str,
95
94
  *,
96
- start_date: Optional[datetime] = None,
97
- end_date: Optional[datetime] = None,
98
- period_days: Optional[int] = None,
95
+ start_date: datetime | None = None,
96
+ end_date: datetime | None = None,
97
+ period_days: int | None = None,
99
98
  ) -> CashFlowAnalysis:
100
99
  """Analyze cash flow (income vs expenses).
101
100
 
@@ -129,7 +128,7 @@ class AnalyticsEngine:
129
128
  self,
130
129
  user_id: str,
131
130
  *,
132
- definition: Optional[str | SavingsDefinition] = None,
131
+ definition: str | SavingsDefinition | None = None,
133
132
  period: str = "monthly",
134
133
  ) -> SavingsRateData:
135
134
  """Calculate savings rate.
@@ -163,7 +162,7 @@ class AnalyticsEngine:
163
162
  self,
164
163
  user_id: str,
165
164
  *,
166
- period_days: Optional[int] = None,
165
+ period_days: int | None = None,
167
166
  include_trends: bool = True,
168
167
  ) -> SpendingInsight:
169
168
  """Analyze spending patterns and generate insights.
@@ -193,8 +192,8 @@ class AnalyticsEngine:
193
192
  self,
194
193
  user_id: str,
195
194
  *,
196
- period_days: Optional[int] = None,
197
- user_context: Optional[dict] = None,
195
+ period_days: int | None = None,
196
+ user_context: dict | None = None,
198
197
  ) -> PersonalizedSpendingAdvice:
199
198
  """Generate AI-powered personalized spending advice.
200
199
 
@@ -228,7 +227,7 @@ class AnalyticsEngine:
228
227
  self,
229
228
  user_id: str,
230
229
  *,
231
- accounts: Optional[list[str]] = None,
230
+ accounts: list[str] | None = None,
232
231
  ) -> PortfolioMetrics:
233
232
  """Calculate portfolio performance metrics.
234
233
 
@@ -250,9 +249,9 @@ class AnalyticsEngine:
250
249
  self,
251
250
  user_id: str,
252
251
  *,
253
- benchmark: Optional[str] = None,
252
+ benchmark: str | None = None,
254
253
  period: str = "1y",
255
- accounts: Optional[list[str]] = None,
254
+ accounts: list[str] | None = None,
256
255
  ) -> BenchmarkComparison:
257
256
  """Compare portfolio to benchmark index.
258
257
 
@@ -282,7 +281,7 @@ class AnalyticsEngine:
282
281
  user_id: str,
283
282
  *,
284
283
  years: int = 30,
285
- assumptions: Optional[dict] = None,
284
+ assumptions: dict | None = None,
286
285
  ) -> GrowthProjection:
287
286
  """Project net worth growth with scenarios.
288
287
 
@@ -35,7 +35,6 @@ Examples:
35
35
  """
36
36
 
37
37
  from datetime import datetime
38
- from typing import Optional
39
38
 
40
39
  from fin_infra.analytics.models import (
41
40
  AssetAllocation,
@@ -47,7 +46,7 @@ from fin_infra.analytics.models import (
47
46
  async def calculate_portfolio_metrics(
48
47
  user_id: str,
49
48
  *,
50
- accounts: Optional[list[str]] = None,
49
+ accounts: list[str] | None = None,
51
50
  brokerage_provider=None,
52
51
  market_provider=None,
53
52
  ) -> PortfolioMetrics:
@@ -131,7 +130,7 @@ async def compare_to_benchmark(
131
130
  *,
132
131
  benchmark: str = "SPY",
133
132
  period: str = "1y",
134
- accounts: Optional[list[str]] = None,
133
+ accounts: list[str] | None = None,
135
134
  brokerage_provider=None,
136
135
  market_provider=None,
137
136
  ) -> BenchmarkComparison:
@@ -221,7 +220,7 @@ async def compare_to_benchmark(
221
220
 
222
221
  def _generate_mock_holdings(
223
222
  user_id: str,
224
- accounts: Optional[list[str]] = None,
223
+ accounts: list[str] | None = None,
225
224
  ) -> list[dict]:
226
225
  """Generate mock portfolio holdings for testing.
227
226
 
@@ -415,7 +414,7 @@ def _parse_benchmark_period(period: str) -> int:
415
414
  def _calculate_portfolio_return(
416
415
  user_id: str,
417
416
  period_days: int,
418
- accounts: Optional[list[str]] = None,
417
+ accounts: list[str] | None = None,
419
418
  ) -> tuple[float, float]:
420
419
  """Calculate portfolio return for specified period.
421
420
 
@@ -483,7 +482,7 @@ def _calculate_beta(
483
482
  user_id: str,
484
483
  benchmark: str,
485
484
  period_days: int,
486
- ) -> Optional[float]:
485
+ ) -> float | None:
487
486
  """Calculate portfolio beta (volatility relative to benchmark).
488
487
 
489
488
  Beta = Covariance(portfolio_returns, benchmark_returns) / Variance(benchmark_returns)
@@ -713,6 +712,7 @@ def _calculate_allocation_from_holdings(
713
712
  - other → Other
714
713
  """
715
714
  from collections import defaultdict
715
+
716
716
  from .models import AssetAllocation
717
717
 
718
718
  if total_value == 0:
@@ -19,12 +19,10 @@ Typical usage:
19
19
  )
20
20
  """
21
21
 
22
- from typing import Optional
23
22
  import math
24
23
 
25
24
  from fin_infra.analytics.models import GrowthProjection, Scenario
26
25
 
27
-
28
26
  # ============================================================================
29
27
  # Public API
30
28
  # ============================================================================
@@ -34,7 +32,7 @@ async def project_net_worth(
34
32
  user_id: str,
35
33
  *,
36
34
  years: int = 30,
37
- assumptions: Optional[dict] = None,
35
+ assumptions: dict | None = None,
38
36
  net_worth_provider=None,
39
37
  cash_flow_provider=None,
40
38
  ) -> GrowthProjection:
@@ -45,12 +45,12 @@ from __future__ import annotations
45
45
 
46
46
  import os
47
47
  from datetime import date
48
- from typing import TYPE_CHECKING, Optional, cast
48
+ from typing import TYPE_CHECKING, cast
49
49
 
50
50
  from pydantic import BaseModel, Field
51
51
 
52
- from ..providers.registry import resolve
53
52
  from ..providers.base import BankingProvider
53
+ from ..providers.registry import resolve
54
54
 
55
55
  if TYPE_CHECKING:
56
56
  from fastapi import FastAPI
@@ -99,7 +99,7 @@ class ExchangeTokenResponse(BaseModel):
99
99
  """Response model for token exchange."""
100
100
 
101
101
  access_token: str
102
- item_id: Optional[str] = None
102
+ item_id: str | None = None
103
103
 
104
104
 
105
105
  class BalanceHistoryStats(BaseModel):
@@ -174,7 +174,6 @@ def easy_banking(provider: str = "teller", **config) -> BankingProvider:
174
174
  See Also:
175
175
  - add_banking(): For FastAPI integration with routes
176
176
  - docs/banking.md: Comprehensive banking integration guide
177
- - docs/adr/0003-banking-integration.md: Architecture decisions
178
177
  """
179
178
  # Auto-detect provider config from environment if not explicitly provided
180
179
  # Only auto-detect if no config params were passed
@@ -199,11 +198,11 @@ def easy_banking(provider: str = "teller", **config) -> BankingProvider:
199
198
  }
200
199
 
201
200
  # Use provider registry to dynamically load and configure provider
202
- return cast(BankingProvider, resolve("banking", provider, **config))
201
+ return cast("BankingProvider", resolve("banking", provider, **config))
203
202
 
204
203
 
205
204
  def add_banking(
206
- app: "FastAPI",
205
+ app: FastAPI,
207
206
  *,
208
207
  provider: str | BankingProvider | None = None,
209
208
  prefix: str = "/banking",
@@ -350,25 +349,25 @@ def add_banking(
350
349
  @router.get("/transactions")
351
350
  async def get_transactions(
352
351
  access_token: str = Depends(get_access_token),
353
- start_date: Optional[date] = Query(None, description="Filter by start date (ISO format)"),
354
- end_date: Optional[date] = Query(None, description="Filter by end date (ISO format)"),
355
- merchant: Optional[str] = Query(
352
+ start_date: date | None = Query(None, description="Filter by start date (ISO format)"),
353
+ end_date: date | None = Query(None, description="Filter by end date (ISO format)"),
354
+ merchant: str | None = Query(
356
355
  None, description="Filter by merchant name (partial match, case-insensitive)"
357
356
  ),
358
- category: Optional[str] = Query(
357
+ category: str | None = Query(
359
358
  None, description="Filter by category (comma-separated list for multiple)"
360
359
  ),
361
- min_amount: Optional[float] = Query(
360
+ min_amount: float | None = Query(
362
361
  None, description="Minimum transaction amount (inclusive)"
363
362
  ),
364
- max_amount: Optional[float] = Query(
363
+ max_amount: float | None = Query(
365
364
  None, description="Maximum transaction amount (inclusive)"
366
365
  ),
367
- tags: Optional[str] = Query(None, description="Filter by tags (comma-separated list)"),
368
- account_id: Optional[str] = Query(None, description="Filter by specific account ID"),
369
- is_recurring: Optional[bool] = Query(None, description="Filter by recurring status"),
370
- sort_by: Optional[str] = Query("date", description="Sort field: date, amount, or merchant"),
371
- order: Optional[str] = Query("desc", description="Sort order: asc or desc"),
366
+ tags: str | None = Query(None, description="Filter by tags (comma-separated list)"),
367
+ account_id: str | None = Query(None, description="Filter by specific account ID"),
368
+ is_recurring: bool | None = Query(None, description="Filter by recurring status"),
369
+ sort_by: str | None = Query("date", description="Sort field: date, amount, or merchant"),
370
+ order: str | None = Query("desc", description="Sort order: asc or desc"),
372
371
  page: int = Query(1, ge=1, description="Page number (starts at 1)"),
373
372
  per_page: int = Query(50, ge=1, le=200, description="Items per page (max 200)"),
374
373
  ):
@@ -476,7 +475,7 @@ def add_banking(
476
475
  @router.get("/balances")
477
476
  async def get_balances(
478
477
  access_token: str = Depends(get_access_token),
479
- account_id: Optional[str] = Query(None),
478
+ account_id: str | None = Query(None),
480
479
  ):
481
480
  """Get current balances."""
482
481
  balances = banking.balances(
@@ -593,17 +592,17 @@ def add_banking(
593
592
 
594
593
  # Import utilities at end to avoid circular imports
595
594
  from .utils import ( # noqa: E402
596
- validate_plaid_token,
597
- validate_teller_token,
598
- validate_mx_token,
599
- validate_provider_token,
595
+ BankingConnectionInfo,
596
+ BankingConnectionStatus,
597
+ get_primary_access_token,
598
+ mark_connection_healthy,
599
+ mark_connection_unhealthy,
600
600
  parse_banking_providers,
601
601
  sanitize_connection_status,
602
- mark_connection_unhealthy,
603
- mark_connection_healthy,
604
- get_primary_access_token,
605
- test_connection_health,
606
602
  should_refresh_token,
607
- BankingConnectionInfo,
608
- BankingConnectionStatus,
603
+ test_connection_health,
604
+ validate_mx_token,
605
+ validate_plaid_token,
606
+ validate_provider_token,
607
+ validate_teller_token,
609
608
  )
@@ -42,9 +42,8 @@ from __future__ import annotations
42
42
  import logging
43
43
  import os
44
44
  from datetime import date, datetime, timedelta
45
- from typing import List, Optional
46
- from pydantic import BaseModel, Field, ConfigDict
47
45
 
46
+ from pydantic import BaseModel, ConfigDict, Field
48
47
 
49
48
  __all__ = [
50
49
  "BalanceSnapshot",
@@ -59,7 +58,7 @@ _logger = logging.getLogger(__name__)
59
58
 
60
59
  # In-memory storage for testing (will be replaced with SQL database in production)
61
60
  # ⚠️ WARNING: All data is LOST on restart when using in-memory storage!
62
- _balance_snapshots: List[BalanceSnapshot] = []
61
+ _balance_snapshots: list[BalanceSnapshot] = []
63
62
  _production_warning_logged = False
64
63
 
65
64
 
@@ -155,9 +154,9 @@ def record_balance_snapshot(
155
154
  def get_balance_history(
156
155
  account_id: str,
157
156
  days: int = 90,
158
- start_date: Optional[date] = None,
159
- end_date: Optional[date] = None,
160
- ) -> List[BalanceSnapshot]:
157
+ start_date: date | None = None,
158
+ end_date: date | None = None,
159
+ ) -> list[BalanceSnapshot]:
161
160
  """Get balance history for an account.
162
161
 
163
162
  Retrieves balance snapshots for the specified account within a date range.
@@ -216,8 +215,8 @@ def get_balance_history(
216
215
 
217
216
  def get_balance_snapshots(
218
217
  account_id: str,
219
- dates: List[date],
220
- ) -> List[BalanceSnapshot]:
218
+ dates: list[date],
219
+ ) -> list[BalanceSnapshot]:
221
220
  """Get balance snapshots for specific dates.
222
221
 
223
222
  Args:
@@ -248,7 +247,7 @@ def get_balance_snapshots(
248
247
 
249
248
  def delete_balance_history(
250
249
  account_id: str,
251
- before_date: Optional[date] = None,
250
+ before_date: date | None = None,
252
251
  ) -> int:
253
252
  """Delete balance history for an account.
254
253