push

Author: hassiebp

langfuse/_client/client.py

@@ -56,17 +56,6 @@
     LANGFUSE_TRACING_ENABLED,
     LANGFUSE_TRACING_ENVIRONMENT,
 )
-from langfuse._client.experiments import (
-    EvaluatorFunction,
-    ExperimentData,
-    ExperimentItem,
-    ExperimentItemResult,
-    ExperimentResult,
-    RunEvaluatorFunction,
-    TaskFunction,
-    _run_evaluator,
-    _run_task,
-)
 from langfuse._client.resource_manager import LangfuseResourceManager
 from langfuse._client.span import (
     LangfuseAgent,
@@ -92,6 +81,17 @@
     Prompt_Chat,
     Prompt_Text,
 )
+from langfuse.experiment import (
+    EvaluatorFunction,
+    ExperimentData,
+    ExperimentItem,
+    ExperimentItemResult,
+    ExperimentResult,
+    RunEvaluatorFunction,
+    TaskFunction,
+    _run_evaluator,
+    _run_task,
+)
 from langfuse.logger import langfuse_logger
 from langfuse.media import LangfuseMedia
 from langfuse.model import (

langfuse/_client/datasets.py

@@ -4,7 +4,7 @@
 
 from opentelemetry.util._decorator import _agnosticcontextmanager
 
-from langfuse._client.experiments import (
+from langfuse.experiment import (
     EvaluatorFunction,
     RunEvaluatorFunction,
     TaskFunction,