Spaces:
Runtime error
Runtime error
"""Zapier Toolkit.""" | |
from typing import List | |
from langchain.agents.agent_toolkits.base import BaseToolkit | |
from langchain.tools import BaseTool | |
from langchain.tools.zapier.tool import ZapierNLARunAction | |
from langchain.utilities.zapier import ZapierNLAWrapper | |
class ZapierToolkit(BaseToolkit): | |
"""Zapier Toolkit.""" | |
tools: List[BaseTool] = [] | |
def from_zapier_nla_wrapper( | |
cls, zapier_nla_wrapper: ZapierNLAWrapper | |
) -> "ZapierToolkit": | |
"""Create a toolkit from a ZapierNLAWrapper.""" | |
actions = zapier_nla_wrapper.list() | |
tools = [ | |
ZapierNLARunAction( | |
action_id=action["id"], | |
zapier_description=action["description"], | |
params_schema=action["params"], | |
api_wrapper=zapier_nla_wrapper, | |
) | |
for action in actions | |
] | |
return cls(tools=tools) | |
def get_tools(self) -> List[BaseTool]: | |
"""Get the tools in the toolkit.""" | |
return self.tools | |