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

intentkit/skills/defillama/tvl/fetch_protocol.py

@@ -1,7 +1,5 @@
 """Tool for fetching specific protocol details via DeFi Llama API."""
 
-from typing import Dict, List, Optional, Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocol
@@ -24,17 +22,15 @@ class TokenAmount(BaseModel):
     """Model representing token amounts at a specific date."""
 
     date: int = Field(..., description="Unix timestamp")
-    tokens: Dict[str, float] = Field(..., description="Token amounts keyed by symbol")
+    tokens: dict[str, float] = Field(..., description="Token amounts keyed by symbol")
 
 
 class ChainTVLData(BaseModel):
     """Model representing TVL data for a specific chain."""
 
-    tvl: List[Dict[str, float]] = Field(..., description="Historical TVL data points")
-    tokens: Optional[Dict[str, float]] = Field(
-        None, description="Current token amounts"
-    )
-    tokensInUsd: Optional[Dict[str, float]] = Field(
+    tvl: list[dict[str, float]] = Field(..., description="Historical TVL data points")
+    tokens: dict[str, float] | None = Field(None, description="Current token amounts")
+    tokensInUsd: dict[str, float] | None = Field(
         None, description="Current token amounts in USD"
     )
 
@@ -53,17 +49,17 @@ class Raise(BaseModel):
     name: str = Field(..., description="Protocol name")
     round: str = Field(..., description="Funding round type")
     amount: float = Field(..., description="Amount raised in millions")
-    chains: List[str] = Field(..., description="Chains involved")
+    chains: list[str] = Field(..., description="Chains involved")
     sector: str = Field(..., description="Business sector")
     category: str = Field(..., description="Protocol category")
     categoryGroup: str = Field(..., description="Category group")
     source: str = Field(..., description="Information source")
-    leadInvestors: List[str] = Field(default_factory=list, description="Lead investors")
-    otherInvestors: List[str] = Field(
+    leadInvestors: list[str] = Field(default_factory=list, description="Lead investors")
+    otherInvestors: list[str] = Field(
         default_factory=list, description="Other investors"
     )
-    valuation: Optional[float] = Field(None, description="Valuation at time of raise")
-    defillamaId: Optional[str] = Field(None, description="DefiLlama ID")
+    valuation: float | None = Field(None, description="Valuation at time of raise")
+    defillamaId: str | None = Field(None, description="DefiLlama ID")
 
 
 class Hallmark(BaseModel):
@@ -79,48 +75,46 @@ class ProtocolDetail(BaseModel):
     # Basic Info
     id: str = Field(..., description="Protocol unique identifier")
     name: str = Field(..., description="Protocol name")
-    address: Optional[str] = Field(None, description="Protocol address")
+    address: str | None = Field(None, description="Protocol address")
     symbol: str = Field(..., description="Protocol token symbol")
     url: str = Field(..., description="Protocol website")
     description: str = Field(..., description="Protocol description")
     logo: str = Field(..., description="Logo URL")
 
     # Chain Info
-    chains: List[str] = Field(default_factory=list, description="Supported chains")
-    currentChainTvls: Dict[str, float] = Field(..., description="Current TVL by chain")
-    chainTvls: Dict[str, ChainTVLData] = Field(
+    chains: list[str] = Field(default_factory=list, description="Supported chains")
+    currentChainTvls: dict[str, float] = Field(..., description="Current TVL by chain")
+    chainTvls: dict[str, ChainTVLData] = Field(
         ..., description="Historical TVL data by chain"
     )
 
     # Identifiers
-    gecko_id: Optional[str] = Field(None, description="CoinGecko ID")
-    cmcId: Optional[str] = Field(None, description="CoinMarketCap ID")
+    gecko_id: str | None = Field(None, description="CoinGecko ID")
+    cmcId: str | None = Field(None, description="CoinMarketCap ID")
 
     # Social & Development
-    twitter: Optional[str] = Field(None, description="Twitter handle")
-    treasury: Optional[str] = Field(None, description="Treasury information")
-    governanceID: Optional[List[str]] = Field(
-        None, description="Governance identifiers"
-    )
-    github: Optional[List[str]] = Field(None, description="GitHub repositories")
+    twitter: str | None = Field(None, description="Twitter handle")
+    treasury: str | None = Field(None, description="Treasury information")
+    governanceID: list[str] | None = Field(None, description="Governance identifiers")
+    github: list[str] | None = Field(None, description="GitHub repositories")
 
     # Protocol Relationships
-    isParentProtocol: Optional[bool] = Field(
+    isParentProtocol: bool | None = Field(
         None, description="Whether this is a parent protocol"
     )
-    otherProtocols: Optional[List[str]] = Field(None, description="Related protocols")
+    otherProtocols: list[str] | None = Field(None, description="Related protocols")
 
     # Historical Data
-    tokens: List[TokenAmount] = Field(
+    tokens: list[TokenAmount] = Field(
         default_factory=list, description="Historical token amounts"
     )
-    tvl: List[HistoricalTVL] = Field(..., description="Historical TVL data points")
-    raises: Optional[List[Raise]] = Field(None, description="Funding rounds")
-    hallmarks: Optional[List[Hallmark]] = Field(None, description="Significant events")
+    tvl: list[HistoricalTVL] = Field(..., description="Historical TVL data points")
+    raises: list[Raise] | None = Field(None, description="Funding rounds")
+    hallmarks: list[Hallmark] | None = Field(None, description="Significant events")
 
     # Market Data
-    mcap: Optional[float] = Field(None, description="Market capitalization")
-    metrics: Dict = Field(default_factory=dict, description="Additional metrics")
+    mcap: float | None = Field(None, description="Market capitalization")
+    metrics: dict = Field(default_factory=dict, description="Additional metrics")
 
 
 class DefiLlamaProtocolInput(BaseModel):
@@ -132,8 +126,8 @@ class DefiLlamaProtocolInput(BaseModel):
 class DefiLlamaProtocolOutput(BaseModel):
     """Output model for the protocol fetching tool."""
 
-    protocol: Optional[ProtocolDetail] = Field(None, description="Protocol details")
-    error: Optional[str] = Field(None, description="Error message if any")
+    protocol: ProtocolDetail | None = Field(None, description="Protocol details")
+    error: str | None = Field(None, description="Error message if any")
 
 
 class DefiLlamaFetchProtocol(DefiLlamaBaseTool):
@@ -144,16 +138,16 @@ class DefiLlamaFetchProtocol(DefiLlamaBaseTool):
 
     Example:
         protocol_tool = DefiLlamaFetchProtocol(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await protocol_tool._arun(protocol="aave")
     """
 
     name: str = "defillama_fetch_protocol"
     description: str = FETCH_PROTOCOL_PROMPT
-    args_schema: Type[BaseModel] = DefiLlamaProtocolInput
+    args_schema: type[BaseModel] = DefiLlamaProtocolInput
 
     async def _arun(self, protocol: str) -> DefiLlamaProtocolOutput:
         """Fetch detailed information about a specific protocol.

intentkit/skills/defillama/tvl/fetch_protocol_current_tvl.py

@@ -1,7 +1,5 @@
 """Tool for fetching protocol TVL via DeFiLlama API."""
 
-from typing import Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocol_current_tvl
@@ -38,16 +36,16 @@ class DefiLlamaFetchProtocolCurrentTvl(DefiLlamaBaseTool):
 
     Example:
         tvl_tool = DefiLlamaFetchProtocolCurrentTvl(
-            skill_store=store,
+            ,
             agent_id="agent_123",
-            agent_store=agent_store
+            agent=agent
         )
         result = await tvl_tool._arun(protocol="aave")
     """
 
     name: str = "defillama_fetch_protocol_tvl"
     description: str = FETCH_TVL_PROMPT
-    args_schema: Type[BaseModel] = FetchProtocolCurrentTVLInput
+    args_schema: type[BaseModel] = FetchProtocolCurrentTVLInput
 
     async def _arun(self, protocol: str) -> FetchProtocolCurrentTVLResponse:
         """Fetch current TVL for the given protocol.

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.