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

intentkit/skills/defillama/base.py

@@ -1,12 +1,11 @@
 """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.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill, SkillContext
+from intentkit.abstracts.graph import AgentContext
+from intentkit.skills.base import IntentKitSkill
 from intentkit.skills.defillama.config.chains import (
     get_chain_from_alias,
 )
@@ -26,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"
     )
@@ -39,7 +35,7 @@ class DefiLlamaBaseTool(IntentKitSkill):
         return "defillama"
 
     async def check_rate_limit(
-        self, context: SkillContext, max_requests: int = 30, interval: int = 5
+        self, context: AgentContext, max_requests: int = 30, interval: int = 5
     ) -> tuple[bool, str | None]:
         """Check if the rate limit has been exceeded.
 
@@ -51,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
@@ -66,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]:
@@ -123,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:
@@ -132,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.
 

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/schema.json

@@ -379,5 +379,9 @@
   "required": [
     "states",
     "enabled"
+  ],
+  "x-tags": [
+    "Analytics",
+    "DeFi"
   ]
-}
+}