intentkit 0.6.9.dev2__py3-none-any.whl → 0.6.10.dev2__py3-none-any.whl

intentkit/skills/dapplooker/base.py

@@ -1,9 +1,10 @@
-from typing import Optional, Type
+from typing import Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill, SkillContext
+from intentkit.skills.base import IntentKitSkill
 
 
 class DappLookerBaseTool(IntentKitSkill):
@@ -16,10 +17,19 @@ class DappLookerBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
-    def get_api_key(self, context: SkillContext) -> Optional[str]:
-        if "api_key" in context.config and context.config["api_key"]:
-            return context.config["api_key"]
-        return self.skill_store.get_system_config("dapplooker_api_key")
+    def get_api_key(self) -> str:
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        api_key_provider = skill_config.get("api_key_provider")
+        if api_key_provider == "platform":
+            return self.skill_store.get_system_config("dapplooker_api_key")
+        # for backward compatibility, may only have api_key in skill_config
+        elif skill_config.get("api_key"):
+            return skill_config.get("api_key")
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}, or no api_key in config"
+            )
 
     @property
     def category(self) -> str:

intentkit/skills/dapplooker/dapplooker_token_data.py

@@ -3,7 +3,6 @@ import logging
 from typing import Any, Dict, List, Optional, Type
 
 import httpx
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.dapplooker.base import DappLookerBaseTool
@@ -58,7 +57,6 @@ class DappLookerTokenData(DappLookerBaseTool):
         token_tickers: Optional[str] = None,
         token_addresses: Optional[str] = None,
         chain: str = "base",
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the DappLooker token data tool.
@@ -72,13 +70,13 @@ class DappLookerTokenData(DappLookerBaseTool):
         Returns:
             str: Formatted token data with market metrics and analytics.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(
             f"dapplooker_token_data.py: Fetching token data with context {context}"
         )
 
         # Get the API key from the agent's configuration or environment variable
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return "Error: No DappLooker API key provided in the configuration or environment."
 

intentkit/skills/defillama/coins/fetch_batch_historical_prices.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_batch_historical_prices
@@ -82,7 +81,7 @@ class DefiLlamaFetchBatchHistoricalPrices(DefiLlamaBaseTool):
     args_schema: Type[BaseModel] = FetchBatchHistoricalPricesInput
 
     async def _arun(
-        self, config: RunnableConfig, coins_timestamps: Dict[str, List[int]]
+        self, coins_timestamps: Dict[str, List[int]]
     ) -> FetchBatchHistoricalPricesResponse:
         """Fetch historical prices for the given tokens at specified timestamps.
 
@@ -95,7 +94,7 @@ class DefiLlamaFetchBatchHistoricalPrices(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchBatchHistoricalPricesResponse(error=error_msg)

intentkit/skills/defillama/coins/fetch_block.py

@@ -2,7 +2,6 @@
 
 from typing import Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_block
@@ -58,7 +57,7 @@ class DefiLlamaFetchBlock(DefiLlamaBaseTool):
     description: str = FETCH_BLOCK_PROMPT
     args_schema: Type[BaseModel] = FetchBlockInput
 
-    async def _arun(self, config: RunnableConfig, chain: str) -> FetchBlockResponse:
+    async def _arun(self, chain: str) -> FetchBlockResponse:
         """Fetch current block data for the given chain.
 
         Args:
@@ -75,7 +74,7 @@ class DefiLlamaFetchBlock(DefiLlamaBaseTool):
                 return FetchBlockResponse(chain=chain, error=f"Invalid chain: {chain}")
 
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchBlockResponse(chain=normalized_chain, error=error_msg)

intentkit/skills/defillama/coins/fetch_current_prices.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_current_prices
@@ -72,9 +71,7 @@ class DefiLlamaFetchCurrentPrices(DefiLlamaBaseTool):
     description: str = FETCH_PRICES_PROMPT
     args_schema: Type[BaseModel] = FetchCurrentPricesInput
 
-    async def _arun(
-        self, config: RunnableConfig, coins: List[str]
-    ) -> FetchCurrentPricesResponse:
+    async def _arun(self, coins: List[str]) -> FetchCurrentPricesResponse:
         """Fetch current prices for the given tokens.
 
         Args:
@@ -86,7 +83,7 @@ class DefiLlamaFetchCurrentPrices(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchCurrentPricesResponse(error=error_msg)

intentkit/skills/defillama/coins/fetch_first_price.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_first_price
@@ -67,9 +66,7 @@ class DefiLlamaFetchFirstPrice(DefiLlamaBaseTool):
     description: str = FETCH_FIRST_PRICE_PROMPT
     args_schema: Type[BaseModel] = FetchFirstPriceInput
 
-    async def _arun(
-        self, config: RunnableConfig, coins: List[str]
-    ) -> FetchFirstPriceResponse:
+    async def _arun(self, coins: List[str]) -> FetchFirstPriceResponse:
         """Fetch first recorded prices for the given tokens.
 
         Args:
@@ -81,7 +78,7 @@ class DefiLlamaFetchFirstPrice(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchFirstPriceResponse(error=error_msg)

intentkit/skills/defillama/coins/fetch_historical_prices.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_historical_prices
@@ -77,7 +76,7 @@ class DefiLlamaFetchHistoricalPrices(DefiLlamaBaseTool):
     args_schema: Type[BaseModel] = FetchHistoricalPricesInput
 
     async def _arun(
-        self, config: RunnableConfig, timestamp: int, coins: List[str]
+        self, timestamp: int, coins: List[str]
     ) -> FetchHistoricalPricesResponse:
         """Fetch historical prices for the given tokens at the specified time.
 
@@ -91,7 +90,7 @@ class DefiLlamaFetchHistoricalPrices(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchHistoricalPricesResponse(error=error_msg)

intentkit/skills/defillama/coins/fetch_price_chart.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_price_chart
@@ -76,9 +75,7 @@ class DefiLlamaFetchPriceChart(DefiLlamaBaseTool):
     description: str = FETCH_PRICE_CHART_PROMPT
     args_schema: Type[BaseModel] = FetchPriceChartInput
 
-    async def _arun(
-        self, config: RunnableConfig, coins: List[str]
-    ) -> FetchPriceChartResponse:
+    async def _arun(self, coins: List[str]) -> FetchPriceChartResponse:
         """Fetch price charts for the given tokens.
 
         Args:
@@ -90,7 +87,7 @@ class DefiLlamaFetchPriceChart(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchPriceChartResponse(error=error_msg)

intentkit/skills/defillama/coins/fetch_price_percentage.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_price_percentage
@@ -60,9 +59,7 @@ class DefiLlamaFetchPricePercentage(DefiLlamaBaseTool):
     description: str = FETCH_PRICE_PERCENTAGE_PROMPT
     args_schema: Type[BaseModel] = FetchPricePercentageInput
 
-    async def _arun(
-        self, config: RunnableConfig, coins: List[str]
-    ) -> FetchPricePercentageResponse:
+    async def _arun(self, coins: List[str]) -> FetchPricePercentageResponse:
         """Fetch price percentage changes for the given tokens.
 
         Args:
@@ -74,7 +71,7 @@ class DefiLlamaFetchPricePercentage(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchPricePercentageResponse(error=error_msg)

intentkit/skills/defillama/fees/fetch_fees_overview.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_fees_overview
@@ -103,7 +102,7 @@ class DefiLlamaFetchFeesOverview(DefiLlamaBaseTool):
 
     args_schema: Type[BaseModel] = EmptyArgsSchema
 
-    async def _arun(self, config: RunnableConfig) -> FetchFeesOverviewResponse:
+    async def _arun(self, **kwargs) -> FetchFeesOverviewResponse:
         """Fetch overview data for protocol fees.
 
         Returns:
@@ -111,7 +110,7 @@ class DefiLlamaFetchFeesOverview(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchFeesOverviewResponse(error=error_msg)

intentkit/skills/defillama/stablecoins/fetch_stablecoin_chains.py

@@ -2,7 +2,6 @@
 
 from typing import List, Optional
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_chains
@@ -73,7 +72,7 @@ class DefiLlamaFetchStablecoinChains(DefiLlamaBaseTool):
     description: str = FETCH_STABLECOIN_CHAINS_PROMPT
     args_schema: None = None  # No input parameters needed
 
-    async def _arun(self, config: RunnableConfig) -> FetchStablecoinChainsResponse:
+    async def _arun(self, **kwargs) -> FetchStablecoinChainsResponse:
         """Fetch stablecoin distribution data across chains.
 
         Returns:
@@ -81,7 +80,7 @@ class DefiLlamaFetchStablecoinChains(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchStablecoinChainsResponse(error=error_msg)

intentkit/skills/defillama/stablecoins/fetch_stablecoin_charts.py

@@ -2,7 +2,6 @@
 
 from typing import List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_charts
@@ -85,7 +84,7 @@ class DefiLlamaFetchStablecoinCharts(DefiLlamaBaseTool):
     args_schema: Type[BaseModel] = FetchStablecoinChartsInput
 
     async def _arun(
-        self, config: RunnableConfig, stablecoin_id: str, chain: Optional[str] = None
+        self, stablecoin_id: str, chain: Optional[str] = None
     ) -> FetchStablecoinChartsResponse:
         """Fetch historical chart data for the given stablecoin.
 
@@ -108,7 +107,7 @@ class DefiLlamaFetchStablecoinCharts(DefiLlamaBaseTool):
                 chain = normalized_chain
 
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchStablecoinChartsResponse(error=error_msg)

intentkit/skills/defillama/stablecoins/fetch_stablecoin_prices.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoin_prices
@@ -54,7 +53,7 @@ class DefiLlamaFetchStablecoinPrices(DefiLlamaBaseTool):
     description: str = FETCH_STABLECOIN_PRICES_PROMPT
     args_schema: None = None  # No input parameters needed
 
-    async def _arun(self, config: RunnableConfig) -> FetchStablecoinPricesResponse:
+    async def _arun(self, **kwargs) -> FetchStablecoinPricesResponse:
         """Fetch stablecoin price data.
 
         Returns:
@@ -62,7 +61,7 @@ class DefiLlamaFetchStablecoinPrices(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchStablecoinPricesResponse(error=error_msg)

intentkit/skills/defillama/stablecoins/fetch_stablecoins.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_stablecoins
@@ -99,7 +98,7 @@ class DefiLlamaFetchStablecoins(DefiLlamaBaseTool):
     description: str = FETCH_STABLECOINS_PROMPT
     args_schema: None = None  # No input parameters needed
 
-    async def _arun(self, config: RunnableConfig) -> FetchStablecoinsResponse:
+    async def _arun(self, **kwargs) -> FetchStablecoinsResponse:
         """Fetch stablecoin data.
 
         Returns:
@@ -107,7 +106,7 @@ class DefiLlamaFetchStablecoins(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchStablecoinsResponse(error=error_msg)

intentkit/skills/defillama/tvl/fetch_chain_historical_tvl.py

@@ -2,7 +2,6 @@
 
 from typing import List, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_chain_historical_tvl
@@ -60,9 +59,7 @@ class DefiLlamaFetchChainHistoricalTvl(DefiLlamaBaseTool):
     description: str = FETCH_HISTORICAL_TVL_PROMPT
     args_schema: Type[BaseModel] = FetchChainHistoricalTVLInput
 
-    async def _arun(
-        self, config: RunnableConfig, chain: str
-    ) -> FetchChainHistoricalTVLResponse:
+    async def _arun(self, chain: str) -> FetchChainHistoricalTVLResponse:
         """Fetch historical TVL data for the given chain.
 
         Args:
@@ -74,7 +71,7 @@ class DefiLlamaFetchChainHistoricalTvl(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchChainHistoricalTVLResponse(chain=chain, error=error_msg)

intentkit/skills/defillama/tvl/fetch_chains.py

@@ -2,7 +2,6 @@
 
 from typing import List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_chains
@@ -75,7 +74,7 @@ class DefiLlamaFetchChains(DefiLlamaBaseTool):
     description: str = FETCH_CHAINS_PROMPT
     args_schema: Type[BaseModel] = FetchChainsInput
 
-    async def _arun(self, config: RunnableConfig) -> FetchChainsResponse:
+    async def _arun(self, **kwargs) -> FetchChainsResponse:
         """Fetch TVL data for all chains.
 
         Returns:
@@ -83,7 +82,7 @@ class DefiLlamaFetchChains(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchChainsResponse(chains=[], total_tvl=0, error=error_msg)

intentkit/skills/defillama/tvl/fetch_historical_tvl.py

@@ -2,7 +2,6 @@
 
 from typing import List, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_historical_tvl
@@ -62,7 +61,7 @@ class DefiLlamaFetchHistoricalTvl(DefiLlamaBaseTool):
     description: str = FETCH_TOTAL_HISTORICAL_TVL_PROMPT
     args_schema: Type[BaseModel] = FetchHistoricalTVLInput
 
-    async def _arun(self, config: RunnableConfig) -> FetchHistoricalTVLResponse:
+    async def _arun(self, **kwargs) -> FetchHistoricalTVLResponse:
         """Fetch historical TVL data across all chains.
 
         Returns:
@@ -70,7 +69,7 @@ class DefiLlamaFetchHistoricalTvl(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchHistoricalTVLResponse(error=error_msg)

intentkit/skills/defillama/tvl/fetch_protocol.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocol
@@ -156,9 +155,7 @@ class DefiLlamaFetchProtocol(DefiLlamaBaseTool):
     description: str = FETCH_PROTOCOL_PROMPT
     args_schema: Type[BaseModel] = DefiLlamaProtocolInput
 
-    async def _arun(
-        self, config: RunnableConfig, protocol: str
-    ) -> DefiLlamaProtocolOutput:
+    async def _arun(self, protocol: str) -> DefiLlamaProtocolOutput:
         """Fetch detailed information about a specific protocol.
 
         Args:
@@ -170,7 +167,7 @@ class DefiLlamaFetchProtocol(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return DefiLlamaProtocolOutput(error=error_msg)

intentkit/skills/defillama/tvl/fetch_protocol_current_tvl.py

@@ -2,7 +2,6 @@
 
 from typing import Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocol_current_tvl
@@ -50,9 +49,7 @@ class DefiLlamaFetchProtocolCurrentTvl(DefiLlamaBaseTool):
     description: str = FETCH_TVL_PROMPT
     args_schema: Type[BaseModel] = FetchProtocolCurrentTVLInput
 
-    async def _arun(
-        self, config: RunnableConfig, protocol: str
-    ) -> FetchProtocolCurrentTVLResponse:
+    async def _arun(self, protocol: str) -> FetchProtocolCurrentTVLResponse:
         """Fetch current TVL for the given protocol.
 
         Args:
@@ -64,7 +61,7 @@ class DefiLlamaFetchProtocolCurrentTvl(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchProtocolCurrentTVLResponse(

intentkit/skills/defillama/tvl/fetch_protocols.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type, Union
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_protocols
@@ -146,7 +145,7 @@ class DefiLlamaFetchProtocols(DefiLlamaBaseTool):
 
     args_schema: Type[BaseModel] = EmptyArgsSchema
 
-    async def _arun(self, config: RunnableConfig) -> DefiLlamaProtocolsOutput:
+    async def _arun(self, **kwargs) -> DefiLlamaProtocolsOutput:
         """Fetch information about all protocols.
 
         Returns:
@@ -154,7 +153,7 @@ class DefiLlamaFetchProtocols(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return DefiLlamaProtocolsOutput(error=error_msg)

intentkit/skills/defillama/volumes/fetch_dex_overview.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_dex_overview
@@ -130,7 +129,7 @@ class DefiLlamaFetchDexOverview(DefiLlamaBaseTool):
     description: str = FETCH_DEX_OVERVIEW_PROMPT
     args_schema: None = None  # No input parameters needed
 
-    async def _arun(self, config: RunnableConfig) -> FetchDexOverviewResponse:
+    async def _arun(self, **kwargs) -> FetchDexOverviewResponse:
         """Fetch DEX overview data.
 
         Returns:
@@ -138,7 +137,7 @@ class DefiLlamaFetchDexOverview(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchDexOverviewResponse(error=error_msg)

intentkit/skills/defillama/volumes/fetch_dex_summary.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_dex_summary
@@ -90,9 +89,7 @@ class DefiLlamaFetchDexSummary(DefiLlamaBaseTool):
     description: str = FETCH_DEX_SUMMARY_PROMPT
     args_schema: Type[BaseModel] = FetchDexSummaryInput
 
-    async def _arun(
-        self, config: RunnableConfig, protocol: str
-    ) -> FetchDexSummaryResponse:
+    async def _arun(self, protocol: str) -> FetchDexSummaryResponse:
         """Fetch summary data for the given DEX protocol.
 
         Args:
@@ -104,7 +101,7 @@ class DefiLlamaFetchDexSummary(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchDexSummaryResponse(error=error_msg)

intentkit/skills/defillama/volumes/fetch_options_overview.py

@@ -2,7 +2,6 @@
 
 from typing import Dict, List, Optional, Type
 
-from langchain.schema.runnable import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_options_overview
@@ -104,7 +103,7 @@ class DefiLlamaFetchOptionsOverview(DefiLlamaBaseTool):
 
     args_schema: Type[BaseModel] = EmptyArgsSchema
 
-    async def _arun(self, config: RunnableConfig) -> FetchOptionsOverviewResponse:
+    async def _arun(self, **kwargs) -> FetchOptionsOverviewResponse:
         """Fetch overview data for all options protocols.
 
         Returns:
@@ -112,7 +111,7 @@ class DefiLlamaFetchOptionsOverview(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchOptionsOverviewResponse(error=error_msg)

intentkit/skills/defillama/yields/fetch_pool_chart.py

@@ -2,7 +2,6 @@
 
 from typing import List, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.defillama.api import fetch_pool_chart
@@ -68,9 +67,7 @@ class DefiLlamaFetchPoolChart(DefiLlamaBaseTool):
     description: str = FETCH_POOL_CHART_PROMPT
     args_schema: Type[BaseModel] = FetchPoolChartInput
 
-    async def _arun(
-        self, config: RunnableConfig, pool_id: str
-    ) -> FetchPoolChartResponse:
+    async def _arun(self, pool_id: str) -> FetchPoolChartResponse:
         """Fetch historical chart data for the given pool.
 
         Args:
@@ -81,7 +78,7 @@ class DefiLlamaFetchPoolChart(DefiLlamaBaseTool):
         """
         try:
             # Check rate limiting
-            context = self.context_from_config(config)
+            context = self.get_context()
             is_rate_limited, error_msg = await self.check_rate_limit(context)
             if is_rate_limited:
                 return FetchPoolChartResponse(error=error_msg)