diff --git a/answer_rocket/__init__.py b/answer_rocket/__init__.py index 71ee6d2..21947cb 100644 --- a/answer_rocket/__init__.py +++ b/answer_rocket/__init__.py @@ -5,7 +5,7 @@ 'MetaDataFrame' ] -__version__ = "0.2.77" +__version__ = "0.2.78" from answer_rocket.client import AnswerRocketClient from answer_rocket.error import AnswerRocketClientError diff --git a/answer_rocket/skill.py b/answer_rocket/skill.py index 92a0b66..72a10bf 100644 --- a/answer_rocket/skill.py +++ b/answer_rocket/skill.py @@ -24,7 +24,7 @@ def __init__(self, config: ClientConfig, gql_client: GraphQlClient): self._config = config self._gql_client = gql_client - def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, validate_parameters: bool = False, tool_definition: dict | None = None) -> RunSkillResult: + def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, validate_parameters: bool = False) -> RunSkillResult: """ Runs a skill and returns its full output (does not stream intermediate skill output). @@ -33,15 +33,10 @@ def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, :param parameters: a dict of parameters to pass to the skill where keys are the param keys and values are the values to populate them with :param validate_parameters: boolean switch which applies guardrails to the parameters before the skill is run - :param tool_definition: a dictionary of the hydrated report corresponding to the skill.b Can be fetched from config.get_copilot_hydrated_reports. Required when validate_parameters is True - :return the full output object of the skill """ - if validate_parameters and tool_definition is None: - raise ValueError("tool_definition is required when validate_parameters is True") - final_result = RunSkillResult(None) preview_query_args = { @@ -49,7 +44,6 @@ def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, "skillName": skill_name, 'parameters': parameters or {}, 'validateParameters': validate_parameters, - 'toolDefinition': tool_definition or {}, } preview_query_vars = { @@ -57,7 +51,6 @@ def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, 'skill_name': Arg(non_null(String)), 'parameters': Arg(JSON), 'validate_parameters': Arg(Boolean), - 'tool_definition': Arg(JSON), } operation = self._gql_client.query(variables=preview_query_vars) @@ -67,7 +60,6 @@ def run(self, copilot_id: str, skill_name: str, parameters: dict | None = None, skill_name=Variable('skill_name'), parameters=Variable('parameters'), validate_parameters=Variable('validate_parameters'), - tool_definition=Variable('tool_definition'), ) try: