opik-optimizer 2.1.0__py3-none-any.whl → 2.1.2__py3-none-any.whl

opik_optimizer/__init__.py

@@ -16,6 +16,7 @@ from .gepa_optimizer import GepaOptimizer
 from .logging_config import setup_logging
 from .meta_prompt_optimizer import MetaPromptOptimizer
 from .mipro_optimizer import MiproOptimizer
+from .hierarchical_reflective_optimizer import HierarchicalReflectiveOptimizer
 from .optimization_config.configs import TaskConfig
 from .optimization_result import OptimizationResult
 from .parameter_optimizer import (
@@ -40,6 +41,7 @@ __all__ = [
     "MetaPromptOptimizer",
     "MiproOptimizer",
     "EvolutionaryOptimizer",
+    "HierarchicalReflectiveOptimizer",
     "ParameterOptimizer",
     "OptimizationResult",
     "OptimizableAgent",

opik_optimizer/base_optimizer.py

@@ -280,6 +280,7 @@ class BaseOptimizer(ABC):
         agent_config["project_name"] = getattr(prompt, "project_name", None)
         agent_config["model"] = getattr(prompt, "model", None) or self.model
         agent_config["tools"] = self._serialize_tools(prompt)
+        agent_config["optimizer"] = self.__class__.__name__
         return self._drop_none(agent_config)
 
     def get_optimizer_metadata(self) -> dict[str, Any]:
@@ -341,6 +342,7 @@ class BaseOptimizer(ABC):
             "metric": getattr(metric, "__name__", str(metric)),
             "dataset": getattr(dataset, "name", None),
             "dataset_id": dataset_id,
+            "optimizer": self.__class__.__name__,
             "optimizer_metadata": self._build_optimizer_metadata(),
             "tool_signatures": self._summarize_tool_signatures(prompt),
             "configuration": {

opik_optimizer/gepa_optimizer/gepa_optimizer.py

@@ -253,7 +253,7 @@ class GepaOptimizer(BaseOptimizer):
                     opt_id = None
 
             gepa_reporting.display_header(
-                algorithm="GEPA",
+                algorithm=self.__class__.__name__,
                 optimization_id=opt_id,
                 dataset_id=getattr(dataset, "id", None),
                 verbose=self.verbose,
@@ -264,7 +264,7 @@ class GepaOptimizer(BaseOptimizer):
             _display_config(
                 messages=prompt.get_messages(),
                 optimizer_config={
-                    "optimizer": "GEPA",
+                    "optimizer": self.__class__.__name__,
                     "model": self.model,
                     "reflection_model": self.reflection_model,
                     "max_metric_calls": max_metric_calls,
@@ -422,7 +422,7 @@ class GepaOptimizer(BaseOptimizer):
                     "system_prompt": candidate_prompt,
                     "gepa_score": val_scores[idx] if idx < len(val_scores) else None,
                     "opik_score": score,
-                    "source": "GEPA",
+                    "source": self.__class__.__name__,
                 }
             )
             history.append(

opik_optimizer/hierarchical_reflective_optimizer/__init__.py

@@ -0,0 +1,5 @@
+from .hierarchical_reflective_optimizer import HierarchicalReflectiveOptimizer
+
+__all__ = [
+    "HierarchicalReflectiveOptimizer",
+]