wehub-resource-sync
e768098d0e
tools_continuous_delivery / Private PyPI non-main branch release (push) Has been skipped
tools_continuous_delivery / Private PyPI main branch release (push) Failing after 2m42s
Publish Promptflow Doc / Build (push) Has been cancelled
Publish Promptflow Doc / Deploy (push) Has been cancelled
Flake8 Lint / flake8 (push) Has been cancelled
Spell check CI / Spell_Check (push) Has been cancelled
18 lines
769 B
Python
18 lines
769 B
Python
from promptflow.core import tool
|
|
|
|
|
|
# The inputs section will change based on the arguments of the tool function, after you save the code
|
|
# Adding type to arguments and return value will help the system show the types properly
|
|
# Please update the function name/signature per need
|
|
@tool
|
|
def select_metrics(metrics: str) -> dict:
|
|
supported_metrics = ('answer_relevance', 'conversation_quality', 'creativity', 'grounding')
|
|
user_selected_metrics = [metric.strip() for metric in metrics.split(',') if metric]
|
|
metric_selection_dict = {}
|
|
for metric in supported_metrics:
|
|
if metric in user_selected_metrics:
|
|
metric_selection_dict[metric] = True
|
|
else:
|
|
metric_selection_dict[metric] = False
|
|
return metric_selection_dict
|