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

intentkit/skills/defillama/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.defillama.base import DefiLlamaBaseTool
 from intentkit.skills.defillama.coins.fetch_batch_historical_prices import (
@@ -124,7 +123,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[DefiLlamaBaseTool]:
     """Get all DeFi Llama skills."""
@@ -140,7 +138,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_defillama_skill(name, store)
+        skill = get_defillama_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -148,13 +146,11 @@ async def get_skills(
 
 def get_defillama_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> DefiLlamaBaseTool:
     """Get a DeFi Llama skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested DeFi Llama skill
@@ -167,145 +163,101 @@ def get_defillama_skill(
     # TVL Skills
     if name == "fetch_protocols":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchProtocols(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchProtocols()
         return _cache[name]
     elif name == "fetch_protocol":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchProtocol(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchProtocol()
         return _cache[name]
     elif name == "fetch_historical_tvl":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchHistoricalTvl(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchHistoricalTvl()
         return _cache[name]
     elif name == "fetch_chain_historical_tvl":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchChainHistoricalTvl(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchChainHistoricalTvl()
         return _cache[name]
     elif name == "fetch_protocol_current_tvl":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchProtocolCurrentTvl(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchProtocolCurrentTvl()
         return _cache[name]
     elif name == "fetch_chains":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchChains(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchChains()
         return _cache[name]
 
     # Coins Skills
     elif name == "fetch_current_prices":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchCurrentPrices(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchCurrentPrices()
         return _cache[name]
     elif name == "fetch_historical_prices":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchHistoricalPrices(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchHistoricalPrices()
         return _cache[name]
     elif name == "fetch_batch_historical_prices":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchBatchHistoricalPrices(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchBatchHistoricalPrices()
         return _cache[name]
     elif name == "fetch_price_chart":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchPriceChart(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchPriceChart()
         return _cache[name]
     elif name == "fetch_price_percentage":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchPricePercentage(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchPricePercentage()
         return _cache[name]
     elif name == "fetch_first_price":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchFirstPrice(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchFirstPrice()
         return _cache[name]
     elif name == "fetch_block":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchBlock(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchBlock()
         return _cache[name]
 
     # Stablecoins Skills
     elif name == "fetch_stablecoins":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchStablecoins(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchStablecoins()
         return _cache[name]
     elif name == "fetch_stablecoin_charts":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchStablecoinCharts(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchStablecoinCharts()
         return _cache[name]
     elif name == "fetch_stablecoin_chains":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchStablecoinChains(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchStablecoinChains()
         return _cache[name]
     elif name == "fetch_stablecoin_prices":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchStablecoinPrices(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchStablecoinPrices()
         return _cache[name]
 
     # Yields Skills
     elif name == "fetch_pools":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchPools(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchPools()
         return _cache[name]
     elif name == "fetch_pool_chart":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchPoolChart(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchPoolChart()
         return _cache[name]
 
     # Volumes Skills
     elif name == "fetch_dex_overview":  # Handles both base and chain-specific overviews
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchDexOverview(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchDexOverview()
         return _cache[name]
     elif name == "fetch_dex_summary":
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchDexSummary(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchDexSummary()
         return _cache[name]
     elif (
         name == "fetch_options_overview"
     ):  # Handles both base and chain-specific overviews
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchOptionsOverview(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchOptionsOverview()
         return _cache[name]
 
     # Fees Skills
@@ -313,9 +265,7 @@ def get_defillama_skill(
         name == "fetch_fees_overview"
     ):  # Handles both base and chain-specific overviews
         if name not in _cache:
-            _cache[name] = DefiLlamaFetchFeesOverview(
-                skill_store=store,
-            )
+            _cache[name] = DefiLlamaFetchFeesOverview()
         return _cache[name]
 
     else:

intentkit/skills/defillama/api.py

@@ -1,7 +1,6 @@
 """DeFi Llama API implementation and shared schemas."""
 
 from datetime import datetime
-from typing import List, Optional
 
 import httpx
 
@@ -75,7 +74,7 @@ async def fetch_chains() -> dict:
 
 
 # Coins API Functions
-async def fetch_current_prices(coins: List[str]) -> dict:
+async def fetch_current_prices(coins: list[str]) -> dict:
     """Get current prices of tokens by contract address using a 4-hour search window."""
     coins_str = ",".join(coins)
     url = f"{DEFILLAMA_COINS_BASE_URL}/prices/current/{coins_str}?searchWidth=4h"
@@ -87,7 +86,7 @@ async def fetch_current_prices(coins: List[str]) -> dict:
     return response.json()
 
 
-async def fetch_historical_prices(timestamp: int, coins: List[str]) -> dict:
+async def fetch_historical_prices(timestamp: int, coins: list[str]) -> dict:
     """Get historical prices of tokens by contract address using a 4-hour search window."""
     coins_str = ",".join(coins)
     url = f"{DEFILLAMA_COINS_BASE_URL}/prices/historical/{timestamp}/{coins_str}?searchWidth=4h"
@@ -112,7 +111,7 @@ async def fetch_batch_historical_prices(coins_timestamps: dict) -> dict:
     return response.json()
 
 
-async def fetch_price_chart(coins: List[str]) -> dict:
+async def fetch_price_chart(coins: list[str]) -> dict:
     """Get historical price chart data from the past day for multiple tokens."""
     coins_str = ",".join(coins)
     start_time = int(datetime.now().timestamp()) - 86400  # now - 1 day
@@ -127,7 +126,7 @@ async def fetch_price_chart(coins: List[str]) -> dict:
     return response.json()
 
 
-async def fetch_price_percentage(coins: List[str]) -> dict:
+async def fetch_price_percentage(coins: list[str]) -> dict:
     """Get price percentage changes for multiple tokens over a 24h period."""
     coins_str = ",".join(coins)
     current_timestamp = int(datetime.now().timestamp())
@@ -142,7 +141,7 @@ async def fetch_price_percentage(coins: List[str]) -> dict:
     return response.json()
 
 
-async def fetch_first_price(coins: List[str]) -> dict:
+async def fetch_first_price(coins: list[str]) -> dict:
     """Get first recorded price data for multiple tokens."""
     coins_str = ",".join(coins)
     url = f"{DEFILLAMA_COINS_BASE_URL}/prices/first/{coins_str}"
@@ -179,9 +178,7 @@ async def fetch_stablecoins() -> dict:
     return response.json()
 
 
-async def fetch_stablecoin_charts(
-    stablecoin_id: str, chain: Optional[str] = None
-) -> dict:
+async def fetch_stablecoin_charts(stablecoin_id: str, chain: str | None = None) -> dict:
     """Get historical circulating supply data for a stablecoin."""
     base_url = f"{DEFILLAMA_STABLECOINS_BASE_URL}/stablecoincharts"
 

intentkit/skills/defillama/base.py

@@ -1,12 +1,10 @@
 """Base class for all DeFi Llama tools."""
 
-from datetime import datetime, timedelta, timezone
-from typing import Type
+from datetime import UTC, datetime, timedelta
 
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.graph import AgentContext
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import IntentKitSkill
 from intentkit.skills.defillama.config.chains import (
     get_chain_from_alias,
@@ -27,10 +25,7 @@ class DefiLlamaBaseTool(IntentKitSkill):
 
     name: str = Field(description="The name of the tool")
     description: str = Field(description="A description of what the tool does")
-    args_schema: Type[BaseModel]
-    skill_store: SkillStoreABC = Field(
-        description="The skill store for persisting data"
-    )
+    args_schema: type[BaseModel]
     base_url: str = Field(
         default=DEFILLAMA_BASE_URL, description="Base URL for DeFi Llama API"
     )
@@ -52,10 +47,8 @@ class DefiLlamaBaseTool(IntentKitSkill):
         Returns:
             Rate limit status and error message if limited
         """
-        rate_limit = await self.skill_store.get_agent_skill_data(
-            context.agent_id, self.name, "rate_limit"
-        )
-        current_time = datetime.now(tz=timezone.utc)
+        rate_limit = await self.get_agent_skill_data("rate_limit")
+        current_time = datetime.now(tz=UTC)
 
         if (
             rate_limit
@@ -67,18 +60,14 @@ class DefiLlamaBaseTool(IntentKitSkill):
                 return True, "Rate limit exceeded"
 
             rate_limit["count"] += 1
-            await self.skill_store.save_agent_skill_data(
-                context.agent_id, self.name, "rate_limit", rate_limit
-            )
+            await self.save_agent_skill_data("rate_limit", rate_limit)
             return False, None
 
         new_rate_limit = {
             "count": 1,
             "reset_time": (current_time + timedelta(minutes=interval)).isoformat(),
         }
-        await self.skill_store.save_agent_skill_data(
-            context.agent_id, self.name, "rate_limit", new_rate_limit
-        )
+        await self.save_agent_skill_data("rate_limit", new_rate_limit)
         return False, None
 
     async def validate_chain(self, chain: str | None) -> tuple[bool, str | None]:
@@ -124,7 +113,7 @@ class DefiLlamaBaseTool(IntentKitSkill):
             "error": True,
             "status_code": status_code,
             "message": message,
-            "timestamp": datetime.now(tz=timezone.utc).isoformat(),
+            "timestamp": datetime.now(tz=UTC).isoformat(),
         }
 
     def get_current_timestamp(self) -> int:
@@ -133,4 +122,4 @@ class DefiLlamaBaseTool(IntentKitSkill):
         Returns:
             Current Unix timestamp
         """
-        return int(datetime.now(tz=timezone.utc).timestamp())
+        return int(datetime.now(tz=UTC).timestamp())

intentkit/skills/defillama/coins/fetch_batch_historical_prices.py

@@ -1,7 +1,5 @@
 """Tool for fetching batch historical token prices via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_batch_historical_prices
@@ -33,7 +31,7 @@ class TokenPriceHistory(BaseModel):
     """Model representing historical price data for a single token."""
 
     symbol: str = Field(..., description="Token symbol")
-    prices: List[HistoricalPricePoint] = Field(
+    prices: list[HistoricalPricePoint] = Field(
         ..., description="List of historical price points"
     )
 
@@ -41,7 +39,7 @@ class TokenPriceHistory(BaseModel):
 class FetchBatchHistoricalPricesInput(BaseModel):
     """Input schema for fetching batch historical token prices."""
 
-    coins_timestamps: Dict[str, List[int]] = Field(
+    coins_timestamps: dict[str, list[int]] = Field(
         ..., description="Dictionary mapping token identifiers to lists of timestamps"
     )
 
@@ -49,11 +47,11 @@ class FetchBatchHistoricalPricesInput(BaseModel):
 class FetchBatchHistoricalPricesResponse(BaseModel):
     """Response schema for batch historical token prices."""
 
-    coins: Dict[str, TokenPriceHistory] = Field(
+    coins: dict[str, TokenPriceHistory] = Field(
         default_factory=dict,
         description="Historical token prices 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 DefiLlamaFetchBatchHistoricalPrices(DefiLlamaBaseTool):
@@ -64,9 +62,9 @@ class DefiLlamaFetchBatchHistoricalPrices(DefiLlamaBaseTool):
 
     Example:
         prices_tool = DefiLlamaFetchBatchHistoricalPrices(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await prices_tool._arun(
             coins_timestamps={
@@ -78,10 +76,10 @@ class DefiLlamaFetchBatchHistoricalPrices(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_batch_historical_prices"
     description: str = FETCH_BATCH_HISTORICAL_PRICES_PROMPT
-    args_schema: Type[BaseModel] = FetchBatchHistoricalPricesInput
+    args_schema: type[BaseModel] = FetchBatchHistoricalPricesInput
 
     async def _arun(
-        self, coins_timestamps: Dict[str, List[int]]
+        self, coins_timestamps: dict[str, list[int]]
     ) -> FetchBatchHistoricalPricesResponse:
         """Fetch historical prices for the given tokens at specified timestamps.
 

intentkit/skills/defillama/coins/fetch_block.py

@@ -1,7 +1,5 @@
 """Tool for fetching current block data via DeFi Llama API."""
 
-from typing import Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_block
@@ -34,9 +32,9 @@ class FetchBlockResponse(BaseModel):
     """Response schema for block data."""
 
     chain: str = Field(..., description="Normalized chain name")
-    height: Optional[int] = Field(None, description="Block height number")
-    timestamp: Optional[int] = Field(None, description="Unix timestamp of the block")
-    error: Optional[str] = Field(None, description="Error message if any")
+    height: int | None = Field(None, description="Block height number")
+    timestamp: int | None = Field(None, description="Unix timestamp of the block")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchBlock(DefiLlamaBaseTool):
@@ -46,16 +44,16 @@ class DefiLlamaFetchBlock(DefiLlamaBaseTool):
 
     Example:
         block_tool = DefiLlamaFetchBlock(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await block_tool._arun(chain="ethereum")
     """
 
     name: str = "defillama_fetch_block"
     description: str = FETCH_BLOCK_PROMPT
-    args_schema: Type[BaseModel] = FetchBlockInput
+    args_schema: type[BaseModel] = FetchBlockInput
 
     async def _arun(self, chain: str) -> FetchBlockResponse:
         """Fetch current block data for the given chain.

intentkit/skills/defillama/coins/fetch_current_prices.py

@@ -1,7 +1,5 @@
 """Tool for fetching token prices via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_current_prices
@@ -29,13 +27,13 @@ class TokenPrice(BaseModel):
     symbol: str = Field(..., description="Token symbol")
     timestamp: int = Field(..., description="Unix timestamp of last price update")
     confidence: float = Field(..., description="Confidence score for the price data")
-    decimals: Optional[int] = Field(None, description="Token decimals, if available")
+    decimals: int | None = Field(None, description="Token decimals, if available")
 
 
 class FetchCurrentPricesInput(BaseModel):
     """Input schema for fetching current token prices with a 4-hour search window."""
 
-    coins: List[str] = Field(
+    coins: list[str] = Field(
         ...,
         description="List of token identifiers (e.g. 'ethereum:0x...', 'coingecko:ethereum')",
     )
@@ -44,10 +42,10 @@ class FetchCurrentPricesInput(BaseModel):
 class FetchCurrentPricesResponse(BaseModel):
     """Response schema for current token prices."""
 
-    coins: Dict[str, TokenPrice] = Field(
+    coins: dict[str, TokenPrice] = Field(
         default_factory=dict, description="Token prices 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 DefiLlamaFetchCurrentPrices(DefiLlamaBaseTool):
@@ -58,9 +56,9 @@ class DefiLlamaFetchCurrentPrices(DefiLlamaBaseTool):
 
     Example:
         prices_tool = DefiLlamaFetchCurrentPrices(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await prices_tool._arun(
             coins=["ethereum:0x...", "coingecko:bitcoin"]
@@ -69,9 +67,9 @@ class DefiLlamaFetchCurrentPrices(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_current_prices"
     description: str = FETCH_PRICES_PROMPT
-    args_schema: Type[BaseModel] = FetchCurrentPricesInput
+    args_schema: type[BaseModel] = FetchCurrentPricesInput
 
-    async def _arun(self, coins: List[str]) -> FetchCurrentPricesResponse:
+    async def _arun(self, coins: list[str]) -> FetchCurrentPricesResponse:
         """Fetch current prices for the given tokens.
 
         Args:

intentkit/skills/defillama/coins/fetch_first_price.py

@@ -1,7 +1,5 @@
 """Tool for fetching first recorded token prices via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_first_price
@@ -31,7 +29,7 @@ class FirstPriceData(BaseModel):
 class FetchFirstPriceInput(BaseModel):
     """Input schema for fetching first token prices."""
 
-    coins: List[str] = Field(
+    coins: list[str] = Field(
         ..., description="List of token identifiers to fetch first prices for"
     )
 
@@ -39,10 +37,10 @@ class FetchFirstPriceInput(BaseModel):
 class FetchFirstPriceResponse(BaseModel):
     """Response schema for first token prices."""
 
-    coins: Dict[str, FirstPriceData] = Field(
+    coins: dict[str, FirstPriceData] = Field(
         default_factory=dict, description="First price 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 DefiLlamaFetchFirstPrice(DefiLlamaBaseTool):
@@ -53,9 +51,9 @@ class DefiLlamaFetchFirstPrice(DefiLlamaBaseTool):
 
     Example:
         first_price_tool = DefiLlamaFetchFirstPrice(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await first_price_tool._arun(
             coins=["ethereum:0x...", "coingecko:ethereum"]
@@ -64,9 +62,9 @@ class DefiLlamaFetchFirstPrice(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_first_price"
     description: str = FETCH_FIRST_PRICE_PROMPT
-    args_schema: Type[BaseModel] = FetchFirstPriceInput
+    args_schema: type[BaseModel] = FetchFirstPriceInput
 
-    async def _arun(self, coins: List[str]) -> FetchFirstPriceResponse:
+    async def _arun(self, coins: list[str]) -> FetchFirstPriceResponse:
         """Fetch first recorded prices for the given tokens.
 
         Args:

intentkit/skills/defillama/coins/fetch_historical_prices.py

@@ -1,7 +1,5 @@
 """Tool for fetching historical token prices via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_historical_prices
@@ -26,9 +24,9 @@ class HistoricalTokenPrice(BaseModel):
     """Model representing historical token price data."""
 
     price: float = Field(..., description="Token price in USD at the specified time")
-    symbol: Optional[str] = Field(None, description="Token symbol")
+    symbol: str | None = Field(None, description="Token symbol")
     timestamp: int = Field(..., description="Unix timestamp of the price data")
-    decimals: Optional[int] = Field(None, description="Token decimals, if available")
+    decimals: int | None = Field(None, description="Token decimals, if available")
 
 
 class FetchHistoricalPricesInput(BaseModel):
@@ -37,7 +35,7 @@ class FetchHistoricalPricesInput(BaseModel):
     timestamp: int = Field(
         ..., description="Unix timestamp for historical price lookup"
     )
-    coins: List[str] = Field(
+    coins: list[str] = Field(
         ...,
         description="List of token identifiers (e.g. 'ethereum:0x...', 'coingecko:ethereum')",
     )
@@ -46,11 +44,11 @@ class FetchHistoricalPricesInput(BaseModel):
 class FetchHistoricalPricesResponse(BaseModel):
     """Response schema for historical token prices."""
 
-    coins: Dict[str, HistoricalTokenPrice] = Field(
+    coins: dict[str, HistoricalTokenPrice] = Field(
         default_factory=dict,
         description="Historical token prices 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 DefiLlamaFetchHistoricalPrices(DefiLlamaBaseTool):
@@ -61,9 +59,9 @@ class DefiLlamaFetchHistoricalPrices(DefiLlamaBaseTool):
 
     Example:
         prices_tool = DefiLlamaFetchHistoricalPrices(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await prices_tool._arun(
             timestamp=1640995200,  # Jan 1, 2022
@@ -73,10 +71,10 @@ class DefiLlamaFetchHistoricalPrices(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_historical_prices"
     description: str = FETCH_HISTORICAL_PRICES_PROMPT
-    args_schema: Type[BaseModel] = FetchHistoricalPricesInput
+    args_schema: type[BaseModel] = FetchHistoricalPricesInput
 
     async def _arun(
-        self, timestamp: int, coins: List[str]
+        self, timestamp: int, coins: list[str]
     ) -> FetchHistoricalPricesResponse:
         """Fetch historical prices for the given tokens at the specified time.