diff --git a/src/uipath/eval/helpers.py b/src/uipath/eval/helpers.py index 8134f65c5..bc6676b7e 100644 --- a/src/uipath/eval/helpers.py +++ b/src/uipath/eval/helpers.py @@ -8,11 +8,10 @@ import click from pydantic import ValidationError -from uipath._cli._evals._conversational_utils import UiPathLegacyEvalChatMessagesMapper - from .evaluators.base_evaluator import GenericBaseEvaluator from .evaluators.evaluator_factory import EvaluatorFactory from .mocks._types import InputMockingStrategy, LLMMockingStrategy +from .models._conversational_utils import UiPathLegacyEvalChatMessagesMapper from .models.evaluation_set import ( EvaluationItem, EvaluationSet, diff --git a/src/uipath/_cli/_evals/_conversational_utils.py b/src/uipath/eval/models/_conversational_utils.py similarity index 100% rename from src/uipath/_cli/_evals/_conversational_utils.py rename to src/uipath/eval/models/_conversational_utils.py diff --git a/src/uipath/eval/models/evaluation_set.py b/src/uipath/eval/models/evaluation_set.py index 89356f474..22e6ce244 100644 --- a/src/uipath/eval/models/evaluation_set.py +++ b/src/uipath/eval/models/evaluation_set.py @@ -5,16 +5,15 @@ from pydantic import BaseModel, ConfigDict, Field from pydantic.alias_generators import to_camel -from uipath._cli._evals._conversational_utils import ( - LegacyConversationalEvalInput, - LegacyConversationalEvalOutput, -) - from ..mocks._types import ( InputMockingStrategy, MockingStrategy, ToolSimulation, ) +from ._conversational_utils import ( + LegacyConversationalEvalInput, + LegacyConversationalEvalOutput, +) class EvaluatorReference(BaseModel): diff --git a/tests/cli/eval/test_conversational_utils.py b/tests/cli/eval/test_conversational_utils.py index 2030eca7f..39d3261d5 100644 --- a/tests/cli/eval/test_conversational_utils.py +++ b/tests/cli/eval/test_conversational_utils.py @@ -1,6 +1,7 @@ """Tests for conversational eval utilities.""" -from uipath._cli._evals._conversational_utils import ( +from uipath.core.chat import UiPathInlineValue +from uipath.eval.models._conversational_utils import ( LegacyConversationalEvalInput, LegacyConversationalEvalInputAgentMessage, LegacyConversationalEvalInputToolCall, @@ -11,7 +12,6 @@ LegacyConversationalEvalUserMessage, UiPathLegacyEvalChatMessagesMapper, ) -from uipath.core.chat import UiPathInlineValue class TestLegacyConversationalEvalInputToUiPathMessages: