Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion answer_rocket/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
10 changes: 1 addition & 9 deletions answer_rocket/skill.py
Original file line number Diff line number Diff line change
Expand Up @@ -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).

Expand All @@ -33,31 +33,24 @@ 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 = {
"copilotId": UUID(copilot_id),
"skillName": skill_name,
'parameters': parameters or {},
'validateParameters': validate_parameters,
'toolDefinition': tool_definition or {},
}

preview_query_vars = {
'copilot_id': Arg(non_null(GQL_UUID)),
'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)
Expand All @@ -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:
Expand Down