diff --git a/evaluators/contrib/galileo/src/agent_control_evaluator_galileo/records/factory.py b/evaluators/contrib/galileo/src/agent_control_evaluator_galileo/records/factory.py index 75a3e5b6..59fb62b5 100644 --- a/evaluators/contrib/galileo/src/agent_control_evaluator_galileo/records/factory.py +++ b/evaluators/contrib/galileo/src/agent_control_evaluator_galileo/records/factory.py @@ -8,12 +8,10 @@ from agent_control_models import Step from galileo_core.schemas.logging.session import Session from galileo_core.schemas.logging.span import ( - AgentSpan, - ControlSpan, LlmSpan, RetrieverSpan, + Span, ToolSpan, - WorkflowSpan, ) from galileo_core.schemas.logging.trace import Trace @@ -21,7 +19,6 @@ type GalileoRecord = LlmSpan | ToolSpan | RetrieverSpan | Trace | Session type GalileoSpan = LlmSpan | ToolSpan | RetrieverSpan -type GalileoCoreSpan = AgentSpan | WorkflowSpan | LlmSpan | RetrieverSpan | ToolSpan | ControlSpan _MISSING = object() _SPAN_TYPES = {"llm", "tool", "retriever"} _RECORD_TYPES = _SPAN_TYPES | {"trace", "session"} @@ -54,8 +51,8 @@ def _selected_values( return selected_data, None -def _span_children(children: list[GalileoRecord], *, parent_type: str) -> list[GalileoCoreSpan]: - spans: list[GalileoCoreSpan] = [] +def _span_children(children: list[GalileoRecord], *, parent_type: str) -> list[Span]: + spans: list[Span] = [] for child in children: if not isinstance(child, (LlmSpan, ToolSpan, RetrieverSpan)): raise RecordFactoryError(f"Galileo {parent_type} children must be span steps.")