intentkit 0.8.6.dev2__py3-none-any.whl → 0.8.17__py3-none-any.whl

intentkit/skills/defillama/coins/fetch_price_chart.py

@@ -1,7 +1,5 @@
 """Tool for fetching token price charts via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_price_chart
@@ -33,14 +31,14 @@ class TokenPriceChart(BaseModel):
 
     symbol: str = Field(..., description="Token symbol")
     confidence: float = Field(..., description="Confidence score for the price data")
-    decimals: Optional[int] = Field(None, description="Token decimals")
-    prices: List[PricePoint] = Field(..., description="List of historical price points")
+    decimals: int | None = Field(None, description="Token decimals")
+    prices: list[PricePoint] = Field(..., description="List of historical price points")
 
 
 class FetchPriceChartInput(BaseModel):
     """Input schema for fetching token price charts."""
 
-    coins: List[str] = Field(
+    coins: list[str] = Field(
         ..., description="List of token identifiers to fetch price charts for"
     )
 
@@ -48,10 +46,10 @@ class FetchPriceChartInput(BaseModel):
 class FetchPriceChartResponse(BaseModel):
     """Response schema for token price charts."""
 
-    coins: Dict[str, TokenPriceChart] = Field(
+    coins: dict[str, TokenPriceChart] = Field(
         default_factory=dict, description="Price chart data keyed by token identifier"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchPriceChart(DefiLlamaBaseTool):
@@ -62,9 +60,9 @@ class DefiLlamaFetchPriceChart(DefiLlamaBaseTool):
 
     Example:
         chart_tool = DefiLlamaFetchPriceChart(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await chart_tool._arun(
             coins=["ethereum:0x...", "coingecko:ethereum"]
@@ -73,9 +71,9 @@ class DefiLlamaFetchPriceChart(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_price_chart"
     description: str = FETCH_PRICE_CHART_PROMPT
-    args_schema: Type[BaseModel] = FetchPriceChartInput
+    args_schema: type[BaseModel] = FetchPriceChartInput
 
-    async def _arun(self, coins: List[str]) -> FetchPriceChartResponse:
+    async def _arun(self, coins: list[str]) -> FetchPriceChartResponse:
         """Fetch price charts for the given tokens.
 
         Args:

intentkit/skills/defillama/coins/fetch_price_percentage.py

@@ -1,7 +1,5 @@
 """Tool for fetching token price percentage changes via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_price_percentage
@@ -23,7 +21,7 @@ Returns price percentage changes:
 class FetchPricePercentageInput(BaseModel):
     """Input schema for fetching token price percentage changes."""
 
-    coins: List[str] = Field(
+    coins: list[str] = Field(
         ..., description="List of token identifiers to fetch price changes for"
     )
 
@@ -31,11 +29,11 @@ class FetchPricePercentageInput(BaseModel):
 class FetchPricePercentageResponse(BaseModel):
     """Response schema for token price percentage changes."""
 
-    coins: Dict[str, float] = Field(
+    coins: dict[str, float] = Field(
         default_factory=dict,
         description="Price percentage changes keyed by token identifier",
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchPricePercentage(DefiLlamaBaseTool):
@@ -46,9 +44,9 @@ class DefiLlamaFetchPricePercentage(DefiLlamaBaseTool):
 
     Example:
         percentage_tool = DefiLlamaFetchPricePercentage(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await percentage_tool._arun(
             coins=["ethereum:0x...", "coingecko:ethereum"]
@@ -57,9 +55,9 @@ class DefiLlamaFetchPricePercentage(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_price_percentage"
     description: str = FETCH_PRICE_PERCENTAGE_PROMPT
-    args_schema: Type[BaseModel] = FetchPricePercentageInput
+    args_schema: type[BaseModel] = FetchPricePercentageInput
 
-    async def _arun(self, coins: List[str]) -> FetchPricePercentageResponse:
+    async def _arun(self, coins: list[str]) -> FetchPricePercentageResponse:
         """Fetch price percentage changes for the given tokens.
 
         Args:

intentkit/skills/defillama/config/chains.py

@@ -4,10 +4,8 @@ This module contains the valid chains and their aliases for use with the DeFi Ll
 The VALID_CHAINS dictionary maps primary chain identifiers to their known aliases.
 """
 
-from typing import Dict, List
-
 # Chain configuration with aliases
-VALID_CHAINS: Dict[str, List[str]] = {
+VALID_CHAINS: dict[str, list[str]] = {
     "ethereum": ["eth", "eth1", "eth2"],
     "solana": ["sol"],
     "bitcoin": ["btc"],

intentkit/skills/defillama/fees/fetch_fees_overview.py

@@ -1,7 +1,5 @@
 """Tool for fetching fees overview data via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_fees_overview
@@ -20,16 +18,16 @@ Returns detailed metrics including:
 class ProtocolMethodology(BaseModel):
     """Model representing protocol methodology data."""
 
-    UserFees: Optional[str] = Field(None, description="Description of user fees")
-    Fees: Optional[str] = Field(None, description="Description of fees")
-    Revenue: Optional[str] = Field(None, description="Description of revenue")
-    ProtocolRevenue: Optional[str] = Field(
+    UserFees: str | None = Field(None, description="Description of user fees")
+    Fees: str | None = Field(None, description="Description of fees")
+    Revenue: str | None = Field(None, description="Description of revenue")
+    ProtocolRevenue: str | None = Field(
         None, description="Description of protocol revenue"
     )
-    HoldersRevenue: Optional[str] = Field(
+    HoldersRevenue: str | None = Field(
         None, description="Description of holders revenue"
     )
-    SupplySideRevenue: Optional[str] = Field(
+    SupplySideRevenue: str | None = Field(
         None, description="Description of supply side revenue"
     )
 
@@ -41,23 +39,23 @@ class Protocol(BaseModel):
     displayName: str = Field(..., description="Display name of protocol")
     category: str = Field(..., description="Protocol category")
     logo: str = Field(..., description="Logo URL")
-    chains: List[str] = Field(..., description="Supported chains")
+    chains: list[str] = Field(..., description="Supported chains")
     module: str = Field(..., description="Protocol module")
-    total24h: Optional[float] = Field(None, description="24-hour total fees")
-    total7d: Optional[float] = Field(None, description="7-day total fees")
-    total30d: Optional[float] = Field(None, description="30-day total fees")
-    total1y: Optional[float] = Field(None, description="1-year total fees")
-    totalAllTime: Optional[float] = Field(None, description="All-time total fees")
-    change_1d: Optional[float] = Field(None, description="24-hour change percentage")
-    change_7d: Optional[float] = Field(None, description="7-day change percentage")
-    change_1m: Optional[float] = Field(None, description="30-day change percentage")
-    methodology: Optional[ProtocolMethodology] = Field(
+    total24h: float | None = Field(None, description="24-hour total fees")
+    total7d: float | None = Field(None, description="7-day total fees")
+    total30d: float | None = Field(None, description="30-day total fees")
+    total1y: float | None = Field(None, description="1-year total fees")
+    totalAllTime: float | None = Field(None, description="All-time total fees")
+    change_1d: float | None = Field(None, description="24-hour change percentage")
+    change_7d: float | None = Field(None, description="7-day change percentage")
+    change_1m: float | None = Field(None, description="30-day change percentage")
+    methodology: ProtocolMethodology | None = Field(
         None, description="Protocol methodology"
     )
-    breakdown24h: Optional[Dict[str, Dict[str, float]]] = Field(
+    breakdown24h: dict[str, dict[str, float]] | None = Field(
         None, description="24-hour breakdown by chain"
     )
-    breakdown30d: Optional[Dict[str, Dict[str, float]]] = Field(
+    breakdown30d: dict[str, dict[str, float]] | None = Field(
         None, description="30-day breakdown by chain"
     )
 
@@ -72,9 +70,9 @@ class FetchFeesOverviewResponse(BaseModel):
     change_1d: float = Field(..., description="24-hour change percentage")
     change_7d: float = Field(..., description="7-day change percentage")
     change_1m: float = Field(..., description="30-day change percentage")
-    allChains: List[str] = Field(..., description="List of all chains")
-    protocols: List[Protocol] = Field(..., description="List of protocols")
-    error: Optional[str] = Field(None, description="Error message if any")
+    allChains: list[str] = Field(..., description="List of all chains")
+    protocols: list[Protocol] = Field(..., description="List of protocols")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchFeesOverview(DefiLlamaBaseTool):
@@ -85,9 +83,9 @@ class DefiLlamaFetchFeesOverview(DefiLlamaBaseTool):
 
     Example:
         overview_tool = DefiLlamaFetchFeesOverview(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await overview_tool._arun()
     """
@@ -100,7 +98,7 @@ class DefiLlamaFetchFeesOverview(DefiLlamaBaseTool):
 
         pass
 
-    args_schema: Type[BaseModel] = EmptyArgsSchema
+    args_schema: type[BaseModel] = EmptyArgsSchema
 
     async def _arun(self, **kwargs) -> FetchFeesOverviewResponse:
         """Fetch overview data for protocol fees.

intentkit/skills/defillama/stablecoins/fetch_stablecoin_chains.py

@@ -1,7 +1,5 @@
 """Tool for fetching stablecoin chains data via DeFi Llama API."""
 
-from typing import List, Optional
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_chains
@@ -19,16 +17,16 @@ Returns:
 class CirculatingUSD(BaseModel):
     """Model representing circulating amounts in different pegs."""
 
-    peggedUSD: Optional[float] = Field(None, description="Amount pegged to USD")
-    peggedEUR: Optional[float] = Field(None, description="Amount pegged to EUR")
-    peggedVAR: Optional[float] = Field(None, description="Amount in variable pegs")
-    peggedJPY: Optional[float] = Field(None, description="Amount pegged to JPY")
-    peggedCHF: Optional[float] = Field(None, description="Amount pegged to CHF")
-    peggedCAD: Optional[float] = Field(None, description="Amount pegged to CAD")
-    peggedGBP: Optional[float] = Field(None, description="Amount pegged to GBP")
-    peggedAUD: Optional[float] = Field(None, description="Amount pegged to AUD")
-    peggedCNY: Optional[float] = Field(None, description="Amount pegged to CNY")
-    peggedREAL: Optional[float] = Field(
+    peggedUSD: float | None = Field(None, description="Amount pegged to USD")
+    peggedEUR: float | None = Field(None, description="Amount pegged to EUR")
+    peggedVAR: float | None = Field(None, description="Amount in variable pegs")
+    peggedJPY: float | None = Field(None, description="Amount pegged to JPY")
+    peggedCHF: float | None = Field(None, description="Amount pegged to CHF")
+    peggedCAD: float | None = Field(None, description="Amount pegged to CAD")
+    peggedGBP: float | None = Field(None, description="Amount pegged to GBP")
+    peggedAUD: float | None = Field(None, description="Amount pegged to AUD")
+    peggedCNY: float | None = Field(None, description="Amount pegged to CNY")
+    peggedREAL: float | None = Field(
         None, description="Amount pegged to Brazilian Real"
     )
 
@@ -36,21 +34,21 @@ class CirculatingUSD(BaseModel):
 class ChainData(BaseModel):
     """Model representing stablecoin data for a single chain."""
 
-    gecko_id: Optional[str] = Field(None, description="CoinGecko ID of the chain")
+    gecko_id: str | None = Field(None, description="CoinGecko ID of the chain")
     totalCirculatingUSD: CirculatingUSD = Field(
         ..., description="Total circulating amounts in different pegs"
     )
-    tokenSymbol: Optional[str] = Field(None, description="Native token symbol")
+    tokenSymbol: str | None = Field(None, description="Native token symbol")
     name: str = Field(..., description="Chain name")
 
 
 class FetchStablecoinChainsResponse(BaseModel):
     """Response schema for stablecoin chains data."""
 
-    chains: List[ChainData] = Field(
+    chains: list[ChainData] = Field(
         default_factory=list, description="List of chains with their stablecoin data"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchStablecoinChains(DefiLlamaBaseTool):
@@ -61,9 +59,9 @@ class DefiLlamaFetchStablecoinChains(DefiLlamaBaseTool):
 
     Example:
         chains_tool = DefiLlamaFetchStablecoinChains(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await chains_tool._arun()
     """

intentkit/skills/defillama/stablecoins/fetch_stablecoin_charts.py

@@ -1,7 +1,5 @@
 """Tool for fetching stablecoin charts via DeFi Llama API."""
 
-from typing import List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_charts
@@ -44,7 +42,7 @@ class FetchStablecoinChartsInput(BaseModel):
     stablecoin_id: str = Field(
         ..., description="ID of the stablecoin to fetch data for"
     )
-    chain: Optional[str] = Field(
+    chain: str | None = Field(
         None, description="Optional chain name for chain-specific data"
     )
 
@@ -52,13 +50,13 @@ class FetchStablecoinChartsInput(BaseModel):
 class FetchStablecoinChartsResponse(BaseModel):
     """Response schema for stablecoin chart data."""
 
-    data: List[StablecoinDataPoint] = Field(
+    data: list[StablecoinDataPoint] = Field(
         default_factory=list, description="List of historical data points"
     )
-    chain: Optional[str] = Field(
+    chain: str | None = Field(
         None, description="Chain name if chain-specific data was requested"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchStablecoinCharts(DefiLlamaBaseTool):
@@ -69,9 +67,9 @@ class DefiLlamaFetchStablecoinCharts(DefiLlamaBaseTool):
 
     Example:
         charts_tool = DefiLlamaFetchStablecoinCharts(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         # Get all chains data
         result = await charts_tool._arun(stablecoin_id="1")
@@ -81,10 +79,10 @@ class DefiLlamaFetchStablecoinCharts(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_stablecoin_charts"
     description: str = FETCH_STABLECOIN_CHARTS_PROMPT
-    args_schema: Type[BaseModel] = FetchStablecoinChartsInput
+    args_schema: type[BaseModel] = FetchStablecoinChartsInput
 
     async def _arun(
-        self, stablecoin_id: str, chain: Optional[str] = None
+        self, stablecoin_id: str, chain: str | None = None
     ) -> FetchStablecoinChartsResponse:
         """Fetch historical chart data for the given stablecoin.
 

intentkit/skills/defillama/stablecoins/fetch_stablecoin_prices.py

@@ -1,7 +1,5 @@
 """Tool for fetching stablecoin prices via DeFi Llama API."""
 
-from typing import Dict, List, Optional
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_prices
@@ -20,7 +18,7 @@ class PriceDataPoint(BaseModel):
     """Model representing a price data point."""
 
     date: str = Field(..., description="Unix timestamp for the price data")
-    prices: Dict[str, float] = Field(
+    prices: dict[str, float] = Field(
         ..., description="Dictionary of stablecoin prices indexed by identifier"
     )
 
@@ -28,10 +26,10 @@ class PriceDataPoint(BaseModel):
 class FetchStablecoinPricesResponse(BaseModel):
     """Response schema for stablecoin prices data."""
 
-    data: List[PriceDataPoint] = Field(
+    data: list[PriceDataPoint] = Field(
         default_factory=list, description="List of price data points"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchStablecoinPrices(DefiLlamaBaseTool):
@@ -42,9 +40,9 @@ class DefiLlamaFetchStablecoinPrices(DefiLlamaBaseTool):
 
     Example:
         prices_tool = DefiLlamaFetchStablecoinPrices(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await prices_tool._arun()
     """

intentkit/skills/defillama/stablecoins/fetch_stablecoins.py

@@ -1,7 +1,5 @@
 """Tool for fetching stablecoin data via DeFi Llama API."""
 
-from typing import Dict, List, Optional
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoins
@@ -45,7 +43,7 @@ class Stablecoin(BaseModel):
     id: str = Field(..., description="Unique identifier")
     name: str = Field(..., description="Stablecoin name")
     symbol: str = Field(..., description="Token symbol")
-    gecko_id: Optional[str] = Field(None, description="CoinGecko ID if available")
+    gecko_id: str | None = Field(None, description="CoinGecko ID if available")
     pegType: str = Field(..., description="Type of peg (e.g. peggedUSD)")
     priceSource: str = Field(..., description="Source of price data")
     pegMechanism: str = Field(..., description="Mechanism maintaining the peg")
@@ -61,10 +59,10 @@ class Stablecoin(BaseModel):
     circulatingPrevMonth: CirculatingAmount = Field(
         ..., description="Total circulating amount from previous month"
     )
-    chainCirculating: Dict[str, ChainCirculating] = Field(
+    chainCirculating: dict[str, ChainCirculating] = Field(
         ..., description="Circulating amounts per chain"
     )
-    chains: List[str] = Field(
+    chains: list[str] = Field(
         ..., description="List of chains where the stablecoin is present"
     )
     price: float = Field(..., description="Current price in USD")
@@ -73,10 +71,10 @@ class Stablecoin(BaseModel):
 class FetchStablecoinsResponse(BaseModel):
     """Response schema for stablecoin data."""
 
-    peggedAssets: List[Stablecoin] = Field(
+    peggedAssets: list[Stablecoin] = Field(
         default_factory=list, description="List of stablecoins with their data"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchStablecoins(DefiLlamaBaseTool):
@@ -87,9 +85,9 @@ class DefiLlamaFetchStablecoins(DefiLlamaBaseTool):
 
     Example:
         stablecoins_tool = DefiLlamaFetchStablecoins(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await stablecoins_tool._arun()
     """

intentkit/skills/defillama/tests/api_integration.test.py

@@ -84,7 +84,7 @@ class TestDefiLlamaAPI(unittest.TestCase):
             return self.loop.run_until_complete(
                 asyncio.wait_for(coro, timeout=self.timeout)
             )
-        except asyncio.TimeoutError:
+        except TimeoutError:
             raise AssertionError(f"Test timed out after {self.timeout} seconds")
         except Exception as e:
             raise AssertionError(f"Test failed with exception: {str(e)}")

intentkit/skills/defillama/tvl/fetch_chain_historical_tvl.py

@@ -1,7 +1,5 @@
 """Tool for fetching chain historical TVL via DeFiLlama API."""
 
-from typing import List, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_chain_historical_tvl
@@ -33,7 +31,7 @@ class FetchChainHistoricalTVLResponse(BaseModel):
     """Response schema for chain-specific historical TVL data."""
 
     chain: str = Field(..., description="Normalized chain name")
-    data: List[HistoricalTVLDataPoint] = Field(
+    data: list[HistoricalTVLDataPoint] = Field(
         default_factory=list, description="List of historical TVL data points"
     )
     error: str | None = Field(default=None, description="Error message if any")
@@ -48,16 +46,16 @@ class DefiLlamaFetchChainHistoricalTvl(DefiLlamaBaseTool):
 
     Example:
         tvl_tool = DefiLlamaFetchChainHistoricalTvl(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await tvl_tool._arun(chain="ethereum")
     """
 
     name: str = "defillama_fetch_chain_historical_tvl"
     description: str = FETCH_HISTORICAL_TVL_PROMPT
-    args_schema: Type[BaseModel] = FetchChainHistoricalTVLInput
+    args_schema: type[BaseModel] = FetchChainHistoricalTVLInput
 
     async def _arun(self, chain: str) -> FetchChainHistoricalTVLResponse:
         """Fetch historical TVL data for the given chain.

intentkit/skills/defillama/tvl/fetch_chains.py

@@ -1,7 +1,5 @@
 """Tool for fetching chain TVL data via DeFi Llama API."""
 
-from typing import List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_chains
@@ -23,14 +21,14 @@ class ChainTVLData(BaseModel):
 
     name: str = Field(..., description="Chain name")
     tvl: float = Field(..., description="Total Value Locked in USD")
-    gecko_id: Optional[str] = Field(None, description="CoinGecko identifier")
-    token_symbol: Optional[str] = Field(
+    gecko_id: str | None = Field(None, description="CoinGecko identifier")
+    token_symbol: str | None = Field(
         None, alias="tokenSymbol", description="Native token symbol"
     )
-    cmc_id: Optional[str] = Field(
+    cmc_id: str | None = Field(
         None, alias="cmcId", description="CoinMarketCap identifier"
     )
-    chain_id: Optional[int | str] = Field(
+    chain_id: int | str | None = Field(
         None, alias="chainId", description="Chain identifier"
     )
 
@@ -48,11 +46,11 @@ class FetchChainsInput(BaseModel):
 class FetchChainsResponse(BaseModel):
     """Response schema for all chains' TVL data."""
 
-    chains: List[ChainTVLData] = Field(
+    chains: list[ChainTVLData] = Field(
         default_factory=list, description="List of chains with their TVL data"
     )
     total_tvl: float = Field(..., description="Total TVL across all chains in USD")
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchChains(DefiLlamaBaseTool):
@@ -63,16 +61,16 @@ class DefiLlamaFetchChains(DefiLlamaBaseTool):
 
     Example:
         chains_tool = DefiLlamaFetchChains(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await chains_tool._arun()
     """
 
     name: str = "defillama_fetch_chains"
     description: str = FETCH_CHAINS_PROMPT
-    args_schema: Type[BaseModel] = FetchChainsInput
+    args_schema: type[BaseModel] = FetchChainsInput
 
     async def _arun(self, **kwargs) -> FetchChainsResponse:
         """Fetch TVL data for all chains.

intentkit/skills/defillama/tvl/fetch_historical_tvl.py

@@ -1,7 +1,5 @@
 """Tool for fetching total historical TVL via DeFiLlama API."""
 
-from typing import List, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_historical_tvl
@@ -34,7 +32,7 @@ class FetchHistoricalTVLInput(BaseModel):
 class FetchHistoricalTVLResponse(BaseModel):
     """Response schema for historical TVL data."""
 
-    data: List[HistoricalTVLDataPoint] = Field(
+    data: list[HistoricalTVLDataPoint] = Field(
         default_factory=list,
         description="List of historical TVL data points across all chains",
     )
@@ -50,16 +48,16 @@ class DefiLlamaFetchHistoricalTvl(DefiLlamaBaseTool):
 
     Example:
         tvl_tool = DefiLlamaFetchHistoricalTvl(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await tvl_tool._arun()
     """
 
     name: str = "defillama_fetch_total_historical_tvl"
     description: str = FETCH_TOTAL_HISTORICAL_TVL_PROMPT
-    args_schema: Type[BaseModel] = FetchHistoricalTVLInput
+    args_schema: type[BaseModel] = FetchHistoricalTVLInput
 
     async def _arun(self, **kwargs) -> FetchHistoricalTVLResponse:
         """Fetch historical TVL data across all chains.