diff --git a/literalai/api/score_helpers.py b/literalai/api/score_helpers.py index 52fa4f8..6d0d456 100644 --- a/literalai/api/score_helpers.py +++ b/literalai/api/score_helpers.py @@ -71,6 +71,7 @@ def create_scores_fields_builder(scores: List[ScoreDict]): generated += f"""$name_{id}: String! $type_{id}: ScoreType! $value_{id}: Float! + $label_{id}: String $stepId_{id}: String $datasetExperimentItemId_{id}: String $scorer_{id}: String @@ -88,6 +89,7 @@ def create_scores_args_builder(scores: List[ScoreDict]): name: $name_{id} type: $type_{id} value: $value_{id} + label: $label_{id} stepId: $stepId_{id} datasetExperimentItemId: $datasetExperimentItemId_{id} scorer: $scorer_{id} @@ -98,6 +100,7 @@ def create_scores_args_builder(scores: List[ScoreDict]): name type value + label comment scorer }} diff --git a/literalai/observability/step.py b/literalai/observability/step.py index 1d31ba9..f62eeae 100644 --- a/literalai/observability/step.py +++ b/literalai/observability/step.py @@ -48,6 +48,7 @@ class ScoreDict(TypedDict, total=False): name: str type: ScoreType value: float + label: Optional[str] stepId: Optional[str] datasetExperimentItemId: Optional[str] comment: Optional[str]