Module naptha_sdk.modules.tool

Classes

class Tool (tool_deployment, *args, **kwargs)

Expand source code

class Tool:
    def __init__(self, 
        tool_deployment,
        *args,
        **kwargs
    ):
        self.tool_deployment = tool_deployment
        self.tool_node = NodeClient(self.tool_deployment.node)

    async def call_tool_func(self, module_run: Union[AgentRun, ToolRunInput]):
        logger.info(f"Running tool on worker node {self.tool_node}")
        tool_run = await self.tool_node.run_module(module_type="tool", run_input=module_run.model_dict())
        return tool_run

Methods

async def call_tool_func(self,
module_run: AgentRun | ToolRunInput)

Expand source code

async def call_tool_func(self, module_run: Union[AgentRun, ToolRunInput]):
    logger.info(f"Running tool on worker node {self.tool_node}")
    tool_run = await self.tool_node.run_module(module_type="tool", run_input=module_run.model_dict())
    return tool_run