opik-optimizer 2.1.0__tar.gz → 2.1.2__tar.gz

{opik_optimizer-2.1.0 → opik_optimizer-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opik_optimizer
-Version: 2.1.0
+Version: 2.1.2
 Summary: Agent optimization with Opik
 Home-page: https://github.com/comet-ml/opik
 Author: Comet ML
@@ -18,7 +18,7 @@ Requires-Dist: dspy<3
 Requires-Dist: gepa>=0.0.7
 Requires-Dist: ujson
 Requires-Dist: hf_xet
-Requires-Dist: litellm
+Requires-Dist: litellm<=1.75.6
 Requires-Dist: mcp>=1.0.0
 Requires-Dist: opik>=1.7.17
 Requires-Dist: optuna

{opik_optimizer-2.1.0 → opik_optimizer-2.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "opik_optimizer"
-version = "2.1.0"
+version = "2.1.2"
 description = "Agent optimization with Opik"
 authors = [
     {name = "Comet ML", email = "support@comet.com"}
@@ -16,7 +16,7 @@ dependencies = [
     "gepa>=0.0.7",
     "ujson",
     "hf_xet",
-    "litellm",
+    "litellm<=1.75.6",
     "mcp>=1.0.0",
     "opik>=1.7.17",
     "optuna",

{opik_optimizer-2.1.0 → opik_optimizer-2.1.2}/src/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-2.1.0 → opik_optimizer-2.1.2}/src/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-2.1.0 → opik_optimizer-2.1.2}/src/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-2.1.2/src/opik_optimizer/hierarchical_reflective_optimizer/__init__.py

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