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

intentkit/skills/defillama/tvl/fetch_protocols.py

@@ -1,7 +1,5 @@
 """Tool for fetching all protocols via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type, Union
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocols
@@ -35,89 +33,83 @@ class Protocol(BaseModel):
     # Basic Info
     id: str = Field(..., description="Protocol unique identifier")
     name: str = Field(..., description="Protocol name")
-    address: Optional[str] = Field(None, description="Protocol's main contract address")
+    address: str | None = Field(None, description="Protocol's main contract address")
     symbol: str = Field(..., description="Protocol token symbol")
-    url: Optional[str] = Field(None, description="Protocol website")
-    description: Optional[str] = Field(None, description="Protocol description")
-    chain: Optional[str] = Field(None, description="Main chain of the protocol")
-    logo: Optional[str] = Field(None, description="URL to protocol logo")
+    url: str | None = Field(None, description="Protocol website")
+    description: str | None = Field(None, description="Protocol description")
+    chain: str | None = Field(None, description="Main chain of the protocol")
+    logo: str | None = Field(None, description="URL to protocol logo")
 
     # Audit Information
-    audits: Union[str, int] = Field("0", description="Number of audits")
-    audit_note: Optional[str] = Field(None, description="Additional audit information")
-    audit_links: Optional[List[str]] = Field(None, description="Links to audit reports")
+    audits: str | int = Field("0", description="Number of audits")
+    audit_note: str | None = Field(None, description="Additional audit information")
+    audit_links: list[str] | None = Field(None, description="Links to audit reports")
 
     # External IDs
-    gecko_id: Optional[str] = Field(None, description="CoinGecko ID")
-    cmcId: Optional[Union[str, int]] = Field(None, description="CoinMarketCap ID")
+    gecko_id: str | None = Field(None, description="CoinGecko ID")
+    cmcId: str | int | None = Field(None, description="CoinMarketCap ID")
 
     # Classification
     category: str = Field(..., description="Protocol category")
-    chains: List[str] = Field(
+    chains: list[str] = Field(
         default_factory=list, description="Chains the protocol operates on"
     )
 
     # Module and Related Info
     module: str = Field(..., description="Module name in DefiLlama")
-    parentProtocol: Optional[str] = Field(
-        None, description="Parent protocol identifier"
-    )
+    parentProtocol: str | None = Field(None, description="Parent protocol identifier")
 
     # Social and Development
-    twitter: Optional[str] = Field(None, description="Twitter handle")
-    github: Optional[List[str]] = Field(None, description="GitHub organization names")
+    twitter: str | None = Field(None, description="Twitter handle")
+    github: list[str] | None = Field(None, description="GitHub organization names")
 
     # Protocol Relationships
-    oracles: List[str] = Field(default_factory=list, description="Oracle services used")
-    forkedFrom: List[str] = Field(
+    oracles: list[str] = Field(default_factory=list, description="Oracle services used")
+    forkedFrom: list[str] = Field(
         default_factory=list, description="Protocols this one was forked from"
     )
 
     # Additional Metadata
-    methodology: Optional[str] = Field(None, description="TVL calculation methodology")
-    listedAt: Optional[int] = Field(
-        None, description="Timestamp when protocol was listed"
-    )
-    openSource: Optional[bool] = Field(
-        None, description="Whether protocol is open source"
-    )
-    treasury: Optional[str] = Field(None, description="Treasury information")
-    misrepresentedTokens: Optional[bool] = Field(
+    methodology: str | None = Field(None, description="TVL calculation methodology")
+    listedAt: int | None = Field(None, description="Timestamp when protocol was listed")
+    openSource: bool | None = Field(None, description="Whether protocol is open source")
+    treasury: str | None = Field(None, description="Treasury information")
+    misrepresentedTokens: bool | None = Field(
         None, description="Whether tokens are misrepresented"
     )
-    hallmarks: Optional[List[Hallmark]] = Field(
+    hallmarks: list[Hallmark] | None = Field(
         None, description="Significant protocol events"
     )
 
     # TVL Related Data
-    tvl: Optional[float] = Field(None, description="Total Value Locked in USD")
-    chainTvls: Dict[str, float] = Field(
+    tvl: float | None = Field(None, description="Total Value Locked in USD")
+    chainTvls: dict[str, float] = Field(
         default_factory=dict,
         description="TVL breakdown by chain including special types (staking, borrowed, etc.)",
     )
-    change_1h: Optional[float] = Field(None, description="1 hour TVL change percentage")
-    change_1d: Optional[float] = Field(None, description="1 day TVL change percentage")
-    change_7d: Optional[float] = Field(None, description="7 day TVL change percentage")
+    change_1h: float | None = Field(None, description="1 hour TVL change percentage")
+    change_1d: float | None = Field(None, description="1 day TVL change percentage")
+    change_7d: float | None = Field(None, description="7 day TVL change percentage")
 
     # Additional TVL Components
-    staking: Optional[float] = Field(None, description="Value in staking")
-    pool2: Optional[float] = Field(None, description="Value in pool2")
-    borrowed: Optional[float] = Field(None, description="Value borrowed")
+    staking: float | None = Field(None, description="Value in staking")
+    pool2: float | None = Field(None, description="Value in pool2")
+    borrowed: float | None = Field(None, description="Value borrowed")
 
     # Token Information
-    tokenBreakdowns: Dict[str, float] = Field(
+    tokenBreakdowns: dict[str, float] = Field(
         default_factory=dict, description="TVL breakdown by token"
     )
-    mcap: Optional[float] = Field(None, description="Market capitalization")
+    mcap: float | None = Field(None, description="Market capitalization")
 
 
 class DefiLlamaProtocolsOutput(BaseModel):
     """Output model for the protocols fetching tool."""
 
-    protocols: List[Protocol] = Field(
+    protocols: list[Protocol] = Field(
         default_factory=list, description="List of fetched protocols"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchProtocols(DefiLlamaBaseTool):
@@ -128,9 +120,9 @@ class DefiLlamaFetchProtocols(DefiLlamaBaseTool):
 
     Example:
         protocols_tool = DefiLlamaFetchProtocols(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await protocols_tool._arun()
     """
@@ -143,7 +135,7 @@ class DefiLlamaFetchProtocols(DefiLlamaBaseTool):
 
         pass
 
-    args_schema: Type[BaseModel] = EmptyArgsSchema
+    args_schema: type[BaseModel] = EmptyArgsSchema
 
     async def _arun(self, **kwargs) -> DefiLlamaProtocolsOutput:
         """Fetch information about all protocols.

intentkit/skills/defillama/volumes/fetch_dex_overview.py

@@ -1,7 +1,5 @@
 """Tool for fetching DEX overview data via DeFi Llama API."""
 
-from typing import Dict, List, Optional
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_dex_overview
@@ -20,78 +18,74 @@ Returns:
 class MethodologyInfo(BaseModel):
     """Model representing methodology information."""
 
-    UserFees: Optional[str] = Field(None, description="User fee information")
-    Fees: Optional[str] = Field(None, description="Fee structure")
-    Revenue: Optional[str] = Field(None, description="Revenue model")
-    ProtocolRevenue: Optional[str] = Field(None, description="Protocol revenue info")
-    HoldersRevenue: Optional[str] = Field(None, description="Holder revenue info")
-    SupplySideRevenue: Optional[str] = Field(
-        None, description="Supply side revenue info"
-    )
+    UserFees: str | None = Field(None, description="User fee information")
+    Fees: str | None = Field(None, description="Fee structure")
+    Revenue: str | None = Field(None, description="Revenue model")
+    ProtocolRevenue: str | None = Field(None, description="Protocol revenue info")
+    HoldersRevenue: str | None = Field(None, description="Holder revenue info")
+    SupplySideRevenue: str | None = Field(None, description="Supply side revenue info")
 
 
 class ProtocolInfo(BaseModel):
     """Model representing individual protocol data."""
 
-    total24h: Optional[float] = Field(None, description="24h total")
-    total48hto24h: Optional[float] = Field(None, description="48h to 24h total")
-    total7d: Optional[float] = Field(None, description="7d total")
-    total14dto7d: Optional[float] = Field(None, description="14d to 7d total")
-    total60dto30d: Optional[float] = Field(None, description="60d to 30d total")
-    total30d: Optional[float] = Field(None, description="30d total")
-    total1y: Optional[float] = Field(None, description="1y total")
-    totalAllTime: Optional[float] = Field(None, description="All time total")
-    average1y: Optional[float] = Field(None, description="1y average")
-    change_1d: Optional[float] = Field(None, description="1d change")
-    change_7d: Optional[float] = Field(None, description="7d change")
-    change_1m: Optional[float] = Field(None, description="1m change")
-    change_7dover7d: Optional[float] = Field(None, description="7d over 7d change")
-    change_30dover30d: Optional[float] = Field(None, description="30d over 30d change")
-    breakdown24h: Optional[Dict[str, Dict[str, float]]] = Field(
+    total24h: float | None = Field(None, description="24h total")
+    total48hto24h: float | None = Field(None, description="48h to 24h total")
+    total7d: float | None = Field(None, description="7d total")
+    total14dto7d: float | None = Field(None, description="14d to 7d total")
+    total60dto30d: float | None = Field(None, description="60d to 30d total")
+    total30d: float | None = Field(None, description="30d total")
+    total1y: float | None = Field(None, description="1y total")
+    totalAllTime: float | None = Field(None, description="All time total")
+    average1y: float | None = Field(None, description="1y average")
+    change_1d: float | None = Field(None, description="1d change")
+    change_7d: float | None = Field(None, description="7d change")
+    change_1m: float | None = Field(None, description="1m change")
+    change_7dover7d: float | None = Field(None, description="7d over 7d change")
+    change_30dover30d: float | None = Field(None, description="30d over 30d change")
+    breakdown24h: dict[str, dict[str, float]] | None = Field(
         None, description="24h breakdown by chain"
     )
-    breakdown30d: Optional[Dict[str, Dict[str, float]]] = Field(
+    breakdown30d: dict[str, dict[str, float]] | None = Field(
         None, description="30d breakdown by chain"
     )
-    total7DaysAgo: Optional[float] = Field(None, description="Total 7 days ago")
-    total30DaysAgo: Optional[float] = Field(None, description="Total 30 days ago")
-    defillamaId: Optional[str] = Field(None, description="DeFi Llama ID")
+    total7DaysAgo: float | None = Field(None, description="Total 7 days ago")
+    total30DaysAgo: float | None = Field(None, description="Total 30 days ago")
+    defillamaId: str | None = Field(None, description="DeFi Llama ID")
     name: str = Field(..., description="Protocol name")
     displayName: str = Field(..., description="Display name")
     module: str = Field(..., description="Module name")
     category: str = Field(..., description="Protocol category")
-    logo: Optional[str] = Field(None, description="Logo URL")
-    chains: List[str] = Field(..., description="Supported chains")
+    logo: str | None = Field(None, description="Logo URL")
+    chains: list[str] = Field(..., description="Supported chains")
     protocolType: str = Field(..., description="Protocol type")
-    methodologyURL: Optional[str] = Field(None, description="Methodology URL")
-    methodology: Optional[MethodologyInfo] = Field(
-        None, description="Methodology details"
-    )
+    methodologyURL: str | None = Field(None, description="Methodology URL")
+    methodology: MethodologyInfo | None = Field(None, description="Methodology details")
     latestFetchIsOk: bool = Field(..., description="Latest fetch status")
-    disabled: Optional[bool] = Field(None, description="Whether protocol is disabled")
-    parentProtocol: Optional[str] = Field(None, description="Parent protocol")
+    disabled: bool | None = Field(None, description="Whether protocol is disabled")
+    parentProtocol: str | None = Field(None, description="Parent protocol")
     slug: str = Field(..., description="Protocol slug")
-    linkedProtocols: Optional[List[str]] = Field(None, description="Linked protocols")
+    linkedProtocols: list[str] | None = Field(None, description="Linked protocols")
     id: str = Field(..., description="Protocol ID")
 
 
 class FetchDexOverviewResponse(BaseModel):
     """Response schema for DEX overview data."""
 
-    totalDataChart: List = Field(
+    totalDataChart: list = Field(
         default_factory=list, description="Total data chart points"
     )
-    totalDataChartBreakdown: List = Field(
+    totalDataChartBreakdown: list = Field(
         default_factory=list, description="Total data chart breakdown"
     )
-    breakdown24h: Optional[Dict[str, Dict[str, float]]] = Field(
+    breakdown24h: dict[str, dict[str, float]] | None = Field(
         None, description="24h breakdown by chain"
     )
-    breakdown30d: Optional[Dict[str, Dict[str, float]]] = Field(
+    breakdown30d: dict[str, dict[str, float]] | None = Field(
         None, description="30d breakdown by chain"
     )
-    chain: Optional[str] = Field(None, description="Specific chain")
-    allChains: List[str] = Field(..., description="List of all chains")
+    chain: str | None = Field(None, description="Specific chain")
+    allChains: list[str] = Field(..., description="List of all chains")
     total24h: float = Field(..., description="24h total")
     total48hto24h: float = Field(..., description="48h to 24h total")
     total7d: float = Field(..., description="7d total")
@@ -106,8 +100,8 @@ class FetchDexOverviewResponse(BaseModel):
     change_30dover30d: float = Field(..., description="30d over 30d change")
     total7DaysAgo: float = Field(..., description="Total 7 days ago")
     total30DaysAgo: float = Field(..., description="Total 30 days ago")
-    protocols: List[ProtocolInfo] = Field(..., description="List of protocol data")
-    error: Optional[str] = Field(None, description="Error message if any")
+    protocols: list[ProtocolInfo] = Field(..., description="List of protocol data")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchDexOverview(DefiLlamaBaseTool):
@@ -118,9 +112,9 @@ class DefiLlamaFetchDexOverview(DefiLlamaBaseTool):
 
     Example:
         overview_tool = DefiLlamaFetchDexOverview(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await overview_tool._arun()
     """

intentkit/skills/defillama/volumes/fetch_dex_summary.py

@@ -1,7 +1,5 @@
 """Tool for fetching DEX protocol summary data via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_dex_summary
@@ -30,44 +28,44 @@ class FetchDexSummaryResponse(BaseModel):
 
     id: str = Field(..., description="Protocol ID")
     name: str = Field(..., description="Protocol name")
-    url: Optional[str] = Field(None, description="Protocol website URL")
-    description: Optional[str] = Field(None, description="Protocol description")
-    logo: Optional[str] = Field(None, description="Logo URL")
-    gecko_id: Optional[str] = Field(None, description="CoinGecko ID")
-    cmcId: Optional[str] = Field(None, description="CoinMarketCap ID")
-    chains: List[str] = Field(default_factory=list, description="Supported chains")
-    twitter: Optional[str] = Field(None, description="Twitter handle")
-    treasury: Optional[str] = Field(None, description="Treasury identifier")
-    governanceID: Optional[List[str]] = Field(None, description="Governance IDs")
-    github: Optional[List[str]] = Field(None, description="GitHub organizations")
-    childProtocols: Optional[List[str]] = Field(None, description="Child protocols")
-    linkedProtocols: Optional[List[str]] = Field(None, description="Linked protocols")
-    disabled: Optional[bool] = Field(None, description="Whether protocol is disabled")
+    url: str | None = Field(None, description="Protocol website URL")
+    description: str | None = Field(None, description="Protocol description")
+    logo: str | None = Field(None, description="Logo URL")
+    gecko_id: str | None = Field(None, description="CoinGecko ID")
+    cmcId: str | None = Field(None, description="CoinMarketCap ID")
+    chains: list[str] = Field(default_factory=list, description="Supported chains")
+    twitter: str | None = Field(None, description="Twitter handle")
+    treasury: str | None = Field(None, description="Treasury identifier")
+    governanceID: list[str] | None = Field(None, description="Governance IDs")
+    github: list[str] | None = Field(None, description="GitHub organizations")
+    childProtocols: list[str] | None = Field(None, description="Child protocols")
+    linkedProtocols: list[str] | None = Field(None, description="Linked protocols")
+    disabled: bool | None = Field(None, description="Whether protocol is disabled")
     displayName: str = Field(..., description="Display name")
-    module: Optional[str] = Field(None, description="Module name")
-    category: Optional[str] = Field(None, description="Protocol category")
-    methodologyURL: Optional[str] = Field(None, description="Methodology URL")
-    methodology: Optional[Dict] = Field(None, description="Methodology details")
-    forkedFrom: Optional[List[str]] = Field(None, description="Forked from protocols")
-    audits: Optional[str] = Field(None, description="Audit information")
-    address: Optional[str] = Field(None, description="Contract address")
-    audit_links: Optional[List[str]] = Field(None, description="Audit links")
-    versionKey: Optional[str] = Field(None, description="Version key")
-    parentProtocol: Optional[str] = Field(None, description="Parent protocol")
-    previousNames: Optional[List[str]] = Field(None, description="Previous names")
+    module: str | None = Field(None, description="Module name")
+    category: str | None = Field(None, description="Protocol category")
+    methodologyURL: str | None = Field(None, description="Methodology URL")
+    methodology: dict | None = Field(None, description="Methodology details")
+    forkedFrom: list[str] | None = Field(None, description="Forked from protocols")
+    audits: str | None = Field(None, description="Audit information")
+    address: str | None = Field(None, description="Contract address")
+    audit_links: list[str] | None = Field(None, description="Audit links")
+    versionKey: str | None = Field(None, description="Version key")
+    parentProtocol: str | None = Field(None, description="Parent protocol")
+    previousNames: list[str] | None = Field(None, description="Previous names")
     latestFetchIsOk: bool = Field(..., description="Latest fetch status")
     slug: str = Field(..., description="Protocol slug")
     protocolType: str = Field(..., description="Protocol type")
-    total24h: Optional[float] = Field(None, description="24h total volume")
-    total48hto24h: Optional[float] = Field(None, description="48h to 24h total volume")
-    total7d: Optional[float] = Field(None, description="7d total volume")
-    totalAllTime: Optional[float] = Field(None, description="All time total volume")
-    totalDataChart: List = Field(default_factory=list, description="Total data chart")
-    totalDataChartBreakdown: List = Field(
+    total24h: float | None = Field(None, description="24h total volume")
+    total48hto24h: float | None = Field(None, description="48h to 24h total volume")
+    total7d: float | None = Field(None, description="7d total volume")
+    totalAllTime: float | None = Field(None, description="All time total volume")
+    totalDataChart: list = Field(default_factory=list, description="Total data chart")
+    totalDataChartBreakdown: list = Field(
         default_factory=list, description="Chart breakdown"
     )
-    change_1d: Optional[float] = Field(None, description="1d change percentage")
-    error: Optional[str] = Field(None, description="Error message if any")
+    change_1d: float | None = Field(None, description="1d change percentage")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchDexSummary(DefiLlamaBaseTool):
@@ -78,16 +76,16 @@ class DefiLlamaFetchDexSummary(DefiLlamaBaseTool):
 
     Example:
         summary_tool = DefiLlamaFetchDexSummary(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await summary_tool._arun(protocol="uniswap")
     """
 
     name: str = "defillama_fetch_dex_summary"
     description: str = FETCH_DEX_SUMMARY_PROMPT
-    args_schema: Type[BaseModel] = FetchDexSummaryInput
+    args_schema: type[BaseModel] = FetchDexSummaryInput
 
     async def _arun(self, protocol: str) -> FetchDexSummaryResponse:
         """Fetch summary data for the given DEX protocol.

intentkit/skills/defillama/volumes/fetch_options_overview.py

@@ -1,7 +1,5 @@
 """Tool for fetching options 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_options_overview
@@ -20,16 +18,14 @@ Returns detailed metrics including:
 class ProtocolMethodology(BaseModel):
     """Model representing protocol methodology data."""
 
-    UserFees: Optional[str] = Field(None, description="User fees description")
-    Fees: Optional[str] = Field(None, description="Fees description")
-    Revenue: Optional[str] = Field(None, description="Revenue description")
-    ProtocolRevenue: Optional[str] = Field(
+    UserFees: str | None = Field(None, description="User fees description")
+    Fees: str | None = Field(None, description="Fees description")
+    Revenue: str | None = Field(None, description="Revenue description")
+    ProtocolRevenue: str | None = Field(
         None, description="Protocol revenue description"
     )
-    HoldersRevenue: Optional[str] = Field(
-        None, description="Holders revenue description"
-    )
-    SupplySideRevenue: Optional[str] = Field(
+    HoldersRevenue: str | None = Field(None, description="Holders revenue description")
+    SupplySideRevenue: str | None = Field(
         None, description="Supply side revenue description"
     )
 
@@ -42,23 +38,23 @@ class Protocol(BaseModel):
     defillamaId: str = Field(..., description="DeFi Llama ID")
     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")
-    total7d: Optional[float] = Field(None, description="7-day total")
-    total30d: Optional[float] = Field(None, description="30-day total")
-    total1y: Optional[float] = Field(None, description="1-year total")
-    totalAllTime: Optional[float] = Field(None, description="All-time total")
-    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")
+    total7d: float | None = Field(None, description="7-day total")
+    total30d: float | None = Field(None, description="30-day total")
+    total1y: float | None = Field(None, description="1-year total")
+    totalAllTime: float | None = Field(None, description="All-time total")
+    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"
     )
 
@@ -73,9 +69,9 @@ class FetchOptionsOverviewResponse(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 DefiLlamaFetchOptionsOverview(DefiLlamaBaseTool):
@@ -86,9 +82,9 @@ class DefiLlamaFetchOptionsOverview(DefiLlamaBaseTool):
 
     Example:
         overview_tool = DefiLlamaFetchOptionsOverview(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await overview_tool._arun()
     """
@@ -101,7 +97,7 @@ class DefiLlamaFetchOptionsOverview(DefiLlamaBaseTool):
 
         pass
 
-    args_schema: Type[BaseModel] = EmptyArgsSchema
+    args_schema: type[BaseModel] = EmptyArgsSchema
 
     async def _arun(self, **kwargs) -> FetchOptionsOverviewResponse:
         """Fetch overview data for all options protocols.

intentkit/skills/defillama/yields/fetch_pool_chart.py

@@ -1,7 +1,5 @@
 """Tool for fetching pool chart data via DeFi Llama API."""
 
-from typing import List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_pool_chart
@@ -23,11 +21,11 @@ class PoolDataPoint(BaseModel):
 
     timestamp: str = Field(..., description="ISO formatted timestamp of the data point")
     tvlUsd: float = Field(..., description="Total Value Locked in USD")
-    apy: Optional[float] = Field(None, description="Total APY including rewards")
-    apyBase: Optional[float] = Field(None, description="Base APY without rewards")
-    apyReward: Optional[float] = Field(None, description="Additional APY from rewards")
-    il7d: Optional[float] = Field(None, description="7-day impermanent loss")
-    apyBase7d: Optional[float] = Field(None, description="7-day base APY")
+    apy: float | None = Field(None, description="Total APY including rewards")
+    apyBase: float | None = Field(None, description="Base APY without rewards")
+    apyReward: float | None = Field(None, description="Additional APY from rewards")
+    il7d: float | None = Field(None, description="7-day impermanent loss")
+    apyBase7d: float | None = Field(None, description="7-day base APY")
 
 
 class FetchPoolChartInput(BaseModel):
@@ -40,10 +38,10 @@ class FetchPoolChartResponse(BaseModel):
     """Response schema for pool chart data."""
 
     status: str = Field("success", description="Response status")
-    data: List[PoolDataPoint] = Field(
+    data: list[PoolDataPoint] = Field(
         default_factory=list, description="List of historical data points"
     )
-    error: Optional[str] = Field(None, description="Error message if any")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchPoolChart(DefiLlamaBaseTool):
@@ -54,9 +52,9 @@ class DefiLlamaFetchPoolChart(DefiLlamaBaseTool):
 
     Example:
         chart_tool = DefiLlamaFetchPoolChart(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await chart_tool._arun(
             pool_id="747c1d2a-c668-4682-b9f9-296708a3dd90"
@@ -65,7 +63,7 @@ class DefiLlamaFetchPoolChart(DefiLlamaBaseTool):
 
     name: str = "defillama_fetch_pool_chart"
     description: str = FETCH_POOL_CHART_PROMPT
-    args_schema: Type[BaseModel] = FetchPoolChartInput
+    args_schema: type[BaseModel] = FetchPoolChartInput
 
     async def _arun(self, pool_id: str) -> FetchPoolChartResponse:
         """Fetch historical chart data for the given pool.