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

intentkit/skills/defillama/stablecoins/fetch_stablecoin_chains.py

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

intentkit/skills/defillama/stablecoins/fetch_stablecoin_charts.py

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

intentkit/skills/defillama/stablecoins/fetch_stablecoin_prices.py

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

intentkit/skills/defillama/stablecoins/fetch_stablecoins.py

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

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

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

intentkit/skills/defillama/tvl/fetch_chain_historical_tvl.py

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

intentkit/skills/defillama/tvl/fetch_chains.py

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

intentkit/skills/defillama/tvl/fetch_historical_tvl.py

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

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.