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

intentkit/skills/enso/base.py

@@ -2,11 +2,13 @@ from typing import Optional, Type
 
 from cdp import EvmServerAccount
 from coinbase_agentkit import CdpEvmServerWalletProvider
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
+from intentkit.abstracts.graph import AgentContext
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.clients import CdpClient, get_cdp_client
-from intentkit.skills.base import IntentKitSkill, SkillContext
+from intentkit.skills.base import IntentKitSkill
 from intentkit.utils.chain import ChainProvider, NetworkId
 
 base_url = "https://api.enso.finance"
@@ -23,7 +25,7 @@ class EnsoBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
-    async def get_account(self, context: SkillContext) -> Optional[EvmServerAccount]:
+    async def get_account(self, context: AgentContext) -> Optional[EvmServerAccount]:
         """Get the account object from the CDP client.
 
         Args:
@@ -32,11 +34,11 @@ class EnsoBaseTool(IntentKitSkill):
         Returns:
             Optional[EvmServerAccount]: The account object if available.
         """
-        client: CdpClient = await get_cdp_client(context.agent_id, self.skill_store)
+        client: CdpClient = await get_cdp_client(context.agent.id, self.skill_store)
         return await client.get_account()
 
     async def get_wallet_provider(
-        self, context: SkillContext
+        self, context: AgentContext
     ) -> Optional[CdpEvmServerWalletProvider]:
         """Get the wallet provider from the CDP client.
 
@@ -46,21 +48,31 @@ class EnsoBaseTool(IntentKitSkill):
         Returns:
             Optional[CdpEvmServerWalletProvider]: The wallet provider if available.
         """
-        client: CdpClient = await get_cdp_client(context.agent_id, self.skill_store)
+        client: CdpClient = await get_cdp_client(context.agent.id, self.skill_store)
         return await client.get_wallet_provider()
 
-    def get_chain_provider(self, context: SkillContext) -> Optional[ChainProvider]:
+    def get_chain_provider(self, context: AgentContext) -> Optional[ChainProvider]:
         return self.skill_store.get_system_config("chain_provider")
 
-    def get_main_tokens(self, context: SkillContext) -> list[str]:
-        if "main_tokens" in context.config and context.config["main_tokens"]:
-            return context.config["main_tokens"]
+    def get_main_tokens(self, context: AgentContext) -> list[str]:
+        skill_config = context.agent.skill_config(self.category)
+        if "main_tokens" in skill_config and skill_config["main_tokens"]:
+            return skill_config["main_tokens"]
         return []
 
-    def get_api_token(self, context: SkillContext) -> Optional[str]:
-        if "api_token" in context.config and context.config["api_token"]:
-            return context.config["api_token"]
-        return self.skill_store.get_system_config("enso_api_token")
+    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("enso_api_token")
+        # for backward compatibility, may only have api_token in skill_config
+        elif skill_config.get("api_token"):
+            return skill_config.get("api_token")
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}, or no api_token in config"
+            )
 
     @property
     def category(self) -> str:

intentkit/skills/enso/best_yield.py

@@ -2,10 +2,8 @@ from typing import List, Optional, Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
 from intentkit.skills.enso.base import (
     EnsoBaseTool,
     base_url,
@@ -81,7 +79,6 @@ class EnsoGetBestYield(EnsoBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         token_symbol: str = "USDC",
         chain_id: int = BASE_CHAIN_ID,
         top_n: int = 5,
@@ -101,7 +98,7 @@ class EnsoGetBestYield(EnsoBaseTool):
         Raises:
             ToolException: If there's an error accessing the Enso API.
         """
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
 
         if not api_token:

intentkit/skills/enso/networks.py

@@ -3,11 +3,8 @@ from typing import Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
-
 from .base import EnsoBaseTool, base_url
 
 logger = logging.getLogger(__name__)
@@ -50,7 +47,7 @@ class EnsoGetNetworks(EnsoBaseTool):
     description: str = "Retrieve networks supported by the Enso API"
     args_schema: Type[BaseModel] = EnsoGetNetworksInput
 
-    async def _arun(self, config: RunnableConfig, **kwargs) -> EnsoGetNetworksOutput:
+    async def _arun(self, **kwargs) -> EnsoGetNetworksOutput:
         """
         Function to request the list of supported networks and their chain id and name.
 
@@ -59,7 +56,7 @@ class EnsoGetNetworks(EnsoBaseTool):
         """
         url = f"{base_url}/api/v1/networks"
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
         logger.debug(f"api_token: {api_token}")
         headers = {

intentkit/skills/enso/prices.py

@@ -2,11 +2,8 @@ from typing import Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
-
 from .base import EnsoBaseTool, base_url, default_chain_id
 
 
@@ -46,7 +43,6 @@ class EnsoGetPrices(EnsoBaseTool):
     async def _arun(
         self,
         address: str,
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         **kwargs,
     ) -> EnsoGetPricesOutput:
@@ -62,7 +58,7 @@ class EnsoGetPrices(EnsoBaseTool):
         """
         url = f"{base_url}/api/v1/prices/{str(chainId)}/{address}"
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
 
         headers = {

intentkit/skills/enso/route.py

@@ -2,10 +2,8 @@ from typing import Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
 from intentkit.skills.enso.networks import EnsoGetNetworks
 
 from .base import EnsoBaseTool, base_url, default_chain_id
@@ -164,7 +162,6 @@ class EnsoRouteShortcut(EnsoBaseTool):
         amountIn: list[int],
         tokenIn: list[str],
         tokenOut: list[str],
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         broadcast_requested: bool = False,
         **kwargs,
@@ -183,7 +180,7 @@ class EnsoRouteShortcut(EnsoBaseTool):
             EnsoRouteShortcutOutput: The response containing route shortcut information.
         """
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
         api_token = self.get_api_token(context)
         account = await self.get_account(context)
@@ -204,7 +201,7 @@ class EnsoRouteShortcut(EnsoBaseTool):
                 if network_name is None:
                     networks = await EnsoGetNetworks(
                         skill_store=self.skill_store,
-                    ).arun(config)
+                    ).arun()
 
                     for network in networks.res:
                         if network.id == chainId:

intentkit/skills/enso/tokens.py

@@ -2,10 +2,8 @@ from typing import Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
 from intentkit.skills.enso.base import (
     EnsoBaseTool,
     base_url,
@@ -140,7 +138,6 @@ class EnsoGetTokens(EnsoBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         protocolSlug: str | None = None,
         **kwargs,
@@ -157,7 +154,7 @@ class EnsoGetTokens(EnsoBaseTool):
         """
         url = f"{base_url}/api/v1/tokens"
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
         api_token = self.get_api_token(context)
         main_tokens = self.get_main_tokens(context)

intentkit/skills/enso/wallet.py

@@ -2,11 +2,8 @@ from typing import Literal, Tuple, Type
 
 import httpx
 from langchain.tools.base import ToolException
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
-from intentkit.skills.base import SkillContext
-
 from .base import EnsoBaseTool, base_url, default_chain_id
 
 
@@ -62,7 +59,6 @@ class EnsoGetWalletBalances(EnsoBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         **kwargs,
     ) -> EnsoGetBalancesOutput:
@@ -77,7 +73,7 @@ class EnsoGetWalletBalances(EnsoBaseTool):
         """
         url = f"{base_url}/api/v1/wallet/balances"
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
         account = await self.get_account(context)
         headers = {
@@ -158,7 +154,6 @@ class EnsoGetWalletApprovals(EnsoBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         **kwargs,
     ) -> EnsoGetApprovalsOutput:
@@ -174,7 +169,7 @@ class EnsoGetWalletApprovals(EnsoBaseTool):
         """
         url = f"{base_url}/api/v1/wallet/approvals"
 
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
         account = await self.get_account(context)
 
@@ -297,7 +292,6 @@ class EnsoWalletApprove(EnsoBaseTool):
         self,
         tokenAddress: str,
         amount: int,
-        config: RunnableConfig,
         chainId: int = default_chain_id,
         **kwargs,
     ) -> Tuple[EnsoWalletApproveOutput, EnsoWalletApproveArtifact]:
@@ -314,7 +308,7 @@ class EnsoWalletApprove(EnsoBaseTool):
             Tuple[EnsoBroadcastWalletApproveOutput, EnsoBroadcastWalletApproveArtifact]: The list of approve transaction output or an error message.
         """
         url = f"{base_url}/api/v1/wallet/approve"
-        context: SkillContext = self.context_from_config(config)
+        context = self.get_context()
         api_token = self.get_api_token(context)
         account = await self.get_account(context)
 

intentkit/skills/firecrawl/base.py

@@ -1,9 +1,10 @@
 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 FirecrawlBaseTool(IntentKitSkill):
@@ -16,12 +17,21 @@ class FirecrawlBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
-    def get_api_key(self, context: SkillContext) -> str:
+    def get_api_key(self) -> str:
         """Get the Firecrawl API key from configuration."""
-        skill_config = context.config
-        if skill_config.get("api_key_provider") == "agent_owner":
-            return skill_config.get("api_key")
-        return self.skill_store.get_system_config("firecrawl_api_key")
+        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 == "agent_owner":
+            api_key = skill_config.get("api_key")
+            if api_key:
+                return api_key
+            else:
+                raise ToolException("No api_key found in agent_owner configuration")
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}. Only 'agent_owner' is supported for Firecrawl."
+            )
 
     @property
     def category(self) -> str:

intentkit/skills/firecrawl/clear.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.firecrawl.base import FirecrawlBaseTool
@@ -36,7 +35,6 @@ class FirecrawlClearIndexedContent(FirecrawlBaseTool):
     async def _arun(
         self,
         confirm: bool = False,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Clear all indexed Firecrawl content for the agent.
@@ -48,7 +46,7 @@ class FirecrawlClearIndexedContent(FirecrawlBaseTool):
         Returns:
             str: Confirmation message
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         if not agent_id:

intentkit/skills/firecrawl/crawl.py

@@ -4,7 +4,6 @@ from typing import List, Optional, Type
 
 import httpx
 from langchain_core.documents import Document
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.firecrawl.base import FirecrawlBaseTool
@@ -109,7 +108,6 @@ class FirecrawlCrawl(FirecrawlBaseTool):
         index_content: bool = True,
         chunk_size: int = 1000,
         chunk_overlap: int = 200,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the Firecrawl crawl tool.
@@ -130,21 +128,22 @@ class FirecrawlCrawl(FirecrawlBaseTool):
         Returns:
             str: Formatted crawled content from all pages.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
         logger.debug(f"firecrawl_crawl: Running crawl with context {context}")
 
-        if context.config.get("api_key_provider") == "agent_owner":
-            if context.config.get("rate_limit_number") and context.config.get(
+        if skill_config.get("api_key_provider") == "agent_owner":
+            if skill_config.get("rate_limit_number") and skill_config.get(
                 "rate_limit_minutes"
             ):
                 await self.user_rate_limit_by_category(
                     context.user_id,
-                    context.config["rate_limit_number"],
-                    context.config["rate_limit_minutes"],
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
                 )
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return "Error: No Firecrawl API key provided in the configuration."
 

intentkit/skills/firecrawl/query.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.firecrawl.base import FirecrawlBaseTool
@@ -44,16 +43,15 @@ class FirecrawlQueryIndexedContent(FirecrawlBaseTool):
         self,
         query: str,
         max_results: int = 4,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Query the indexed Firecrawl content."""
         try:
             # Get agent context - throw error if not available
-            if not config:
-                raise ValueError("Configuration is required but not provided")
+            # Configuration is always available in new runtime
+            pass
 
-            context = self.context_from_config(config)
+            context = self.get_context()
             if not context or not context.agent_id:
                 raise ValueError("Agent ID is required but not found in configuration")
 
@@ -109,10 +107,10 @@ class FirecrawlQueryIndexedContent(FirecrawlBaseTool):
             # Extract agent_id for error logging if possible
             agent_id = "UNKNOWN"
             try:
-                if config:
-                    context = self.context_from_config(config)
-                    if context and context.agent_id:
-                        agent_id = context.agent_id
+                # TODO: Fix config reference
+                context = self.get_context()
+                if context and context.agent_id:
+                    agent_id = context.agent_id
             except Exception:
                 pass
 

intentkit/skills/firecrawl/scrape.py

@@ -3,7 +3,6 @@ from typing import List, Optional, Type
 
 import httpx
 from langchain_core.documents import Document
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.firecrawl.base import FirecrawlBaseTool
@@ -95,7 +94,6 @@ class FirecrawlScrape(FirecrawlBaseTool):
         index_content: bool = True,
         chunk_size: int = 1000,
         chunk_overlap: int = 200,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the Firecrawl scrape tool.
@@ -116,21 +114,22 @@ class FirecrawlScrape(FirecrawlBaseTool):
         Returns:
             str: Formatted scraped content based on the requested formats.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
         logger.debug(f"firecrawl_scrape: Running scrape with context {context}")
 
-        if context.config.get("api_key_provider") == "agent_owner":
-            if context.config.get("rate_limit_number") and context.config.get(
+        if skill_config.get("api_key_provider") == "agent_owner":
+            if skill_config.get("rate_limit_number") and skill_config.get(
                 "rate_limit_minutes"
             ):
                 await self.user_rate_limit_by_category(
                     context.user_id,
-                    context.config["rate_limit_number"],
-                    context.config["rate_limit_minutes"],
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
                 )
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return "Error: No Firecrawl API key provided in the configuration."
 

intentkit/skills/github/github_search.py

@@ -3,7 +3,6 @@ from enum import Enum
 from typing import Type
 
 import httpx
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.github.base import GitHubBaseTool
@@ -62,7 +61,6 @@ class GitHubSearch(GitHubBaseTool):
         query: str,
         search_type: SearchType = SearchType.REPOSITORIES,
         max_results: int = 5,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the GitHub search tool.
@@ -76,7 +74,7 @@ class GitHubSearch(GitHubBaseTool):
         Returns:
             str: Formatted search results based on the search type.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(f"github_search.py: Running GitHub search with context {context}")
 
         # Limit max_results to a reasonable range

intentkit/skills/heurist/base.py

@@ -2,6 +2,7 @@
 
 from typing import Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.skill import SkillStoreABC
@@ -21,6 +22,20 @@ class HeuristBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
+    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("heurist_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:
         return "heurist"

intentkit/skills/heurist/image_generation_animagine_xl.py

@@ -3,7 +3,6 @@ from typing import Optional, Type
 
 import httpx
 from epyxid import XID
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.heurist.base import HeuristBaseTool
@@ -61,7 +60,6 @@ class ImageGenerationAnimagineXL(HeuristBaseTool):
         neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
         width: Optional[int] = 1024,
         height: Optional[int] = 680,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate Japanese anime-style images using Heurist AI's AnimagineXL model.
@@ -77,8 +75,9 @@ class ImageGenerationAnimagineXL(HeuristBaseTool):
         Returns:
             str: URL of the generated image.
         """
-        context = self.context_from_config(config)
-        skill_config = context.config
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from the skill store
         if "api_key" in skill_config and skill_config["api_key"]:

intentkit/skills/heurist/image_generation_arthemy_comics.py

@@ -3,7 +3,6 @@ from typing import Optional, Type
 
 import httpx
 from epyxid import XID
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.heurist.base import HeuristBaseTool
@@ -61,7 +60,6 @@ class ImageGenerationArthemyComics(HeuristBaseTool):
         neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
         width: Optional[int] = 1024,
         height: Optional[int] = 680,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate comic-style images using Heurist AI's ArthemyComics model.
@@ -76,8 +74,9 @@ class ImageGenerationArthemyComics(HeuristBaseTool):
         Returns:
             str: URL of the generated image.
         """
-        context = self.context_from_config(config)
-        skill_config = context.config
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from the skill store
         if "api_key" in skill_config and skill_config["api_key"]:

intentkit/skills/heurist/image_generation_arthemy_real.py

@@ -3,7 +3,6 @@ from typing import Optional, Type
 
 import httpx
 from epyxid import XID
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.heurist.base import HeuristBaseTool
@@ -61,7 +60,6 @@ class ImageGenerationArthemyReal(HeuristBaseTool):
         neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
         width: Optional[int] = 1024,
         height: Optional[int] = 680,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate realistic images using Heurist AI's ArthemyReal model.
@@ -76,8 +74,9 @@ class ImageGenerationArthemyReal(HeuristBaseTool):
         Returns:
             str: URL of the generated image.
         """
-        context = self.context_from_config(config)
-        skill_config = context.config
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from the skill store
         if "api_key" in skill_config and skill_config["api_key"]:

intentkit/skills/heurist/image_generation_braindance.py

@@ -3,7 +3,6 @@ from typing import Optional, Type
 
 import httpx
 from epyxid import XID
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.heurist.base import HeuristBaseTool
@@ -61,7 +60,6 @@ class ImageGenerationBrainDance(HeuristBaseTool):
         neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
         width: Optional[int] = 1024,
         height: Optional[int] = 680,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate artistic images using Heurist AI's BrainDance model.
@@ -76,8 +74,9 @@ class ImageGenerationBrainDance(HeuristBaseTool):
         Returns:
             str: URL of the generated image.
         """
-        context = self.context_from_config(config)
-        skill_config = context.config
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from the skill store
         if "api_key" in skill_config and skill_config["api_key"]: