chore: import upstream snapshot with attribution
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
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
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
from dataclasses import dataclass
|
||||
from typing_extensions import Never
|
||||
from agent_framework import Executor, WorkflowBuilder, WorkflowContext, handler
|
||||
|
||||
|
||||
def string_to_number(raw_string: str) -> float:
|
||||
float_number = 0.0
|
||||
try:
|
||||
float_number = float(raw_string)
|
||||
except Exception:
|
||||
if '/' in raw_string:
|
||||
split_list = raw_string.split('/')
|
||||
if len(split_list) == 2:
|
||||
numerator, denominator = split_list
|
||||
try:
|
||||
float_number = float(numerator) / float(denominator)
|
||||
except Exception:
|
||||
return None
|
||||
else:
|
||||
return None
|
||||
else:
|
||||
return None
|
||||
return float_number
|
||||
|
||||
|
||||
@dataclass
|
||||
class EvalInput:
|
||||
groundtruth: str
|
||||
prediction: str
|
||||
|
||||
|
||||
class LineProcessExecutor(Executor):
|
||||
@handler
|
||||
async def process(self, input: EvalInput, ctx: WorkflowContext[Never, int]) -> None:
|
||||
pred_float = string_to_number(input.prediction)
|
||||
if pred_float is None:
|
||||
await ctx.yield_output(-1)
|
||||
return
|
||||
gt_float = string_to_number(input.groundtruth)
|
||||
if gt_float is None:
|
||||
await ctx.yield_output(-1)
|
||||
return
|
||||
if round(pred_float, 10) == round(gt_float, 10):
|
||||
await ctx.yield_output(1)
|
||||
else:
|
||||
await ctx.yield_output(-1)
|
||||
|
||||
|
||||
def create_workflow():
|
||||
_line_process = LineProcessExecutor(id="line_process")
|
||||
return WorkflowBuilder(name="EvalChatMathRow", start_executor=_line_process).build()
|
||||
Reference in New Issue
Block a user