opik-optimizer 0.7.5__tar.gz → 0.7.6__tar.gz

{opik_optimizer-0.7.5/src/opik_optimizer.egg-info → opik_optimizer-0.7.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opik_optimizer
-Version: 0.7.5
+Version: 0.7.6
 Summary: Agent optimization with Opik
 Home-page: https://github.com/comet-ml/opik
 Author: Comet ML

{opik_optimizer-0.7.5 → opik_optimizer-0.7.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="opik_optimizer",
-    version="0.7.5",
+    version="0.7.6",
     description="Agent optimization with Opik",
     author="Comet ML",
     author_email="support@comet.com",

{opik_optimizer-0.7.5 → opik_optimizer-0.7.6}/src/opik_optimizer/few_shot_bayesian_optimizer/few_shot_bayesian_optimizer.py

@@ -25,6 +25,7 @@ _limiter = _throttle.get_rate_limiter_for_current_opik_installation()
 
 logger = logging.getLogger(__name__)
 
+
 @_throttle.rate_limited(_limiter)
 def _call_model(model, messages, seed, model_kwargs):
     model_kwargs = opik_litellm_monitor.try_add_opik_monitoring_to_params(model_kwargs)
@@ -84,7 +85,7 @@ class FewShotBayesianOptimizer(base_optimizer.BaseOptimizer):
         split_idx = int(len(dataset) * train_ratio)
         return dataset[:split_idx], dataset[split_idx:]
 
-    def _optimize_prompt(   
+    def _optimize_prompt(
         self,
         dataset: Union[str, Dataset],
         metric_config: MetricConfig,
@@ -171,7 +172,9 @@ class FewShotBayesianOptimizer(base_optimizer.BaseOptimizer):
                 "n_examples", self.min_examples, self.max_examples
             )
             example_indices = [
-                trial.suggest_categorical(f"example_{i}", list(range(len(dataset_items))))
+                trial.suggest_categorical(
+                    f"example_{i}", list(range(len(dataset_items)))
+                )
                 for i in range(n_examples)
             ]
             trial.set_user_attr("example_indices", example_indices)
@@ -256,6 +259,7 @@ class FewShotBayesianOptimizer(base_optimizer.BaseOptimizer):
         main_prompt_string = best_param.instruction
 
         return optimization_result.OptimizationResult(
+            optimizer=self.__class__.__name__,
             prompt=main_prompt_string,
             score=best_score,
             metric_name=metric_config.metric.name,

{opik_optimizer-0.7.5 → opik_optimizer-0.7.6}/src/opik_optimizer/meta_prompt_optimizer.py

@@ -29,6 +29,7 @@ logger = logging.getLogger(__name__)  # Gets logger configured by setup_logging
 
 _rate_limiter = _throttle.get_rate_limiter_for_current_opik_installation()
 
+
 class MetaPromptOptimizer(BaseOptimizer):
     """Optimizer that uses meta-prompting to improve prompts based on examples and performance."""
 
@@ -873,6 +874,7 @@ class MetaPromptOptimizer(BaseOptimizer):
         }
 
         return OptimizationResult(
+            optimizer=self.__class__.__name__,
             prompt=best_prompt,
             score=best_score,
             metric_name=metric_config.metric.name,

{opik_optimizer-0.7.5 → opik_optimizer-0.7.6}/src/opik_optimizer/mipro_optimizer/mipro_optimizer.py

@@ -325,7 +325,7 @@ class MiproOptimizer(BaseOptimizer):
                 "metric": metric.name,
                 "num_threads": self.num_threads,
                 "num_candidates": self.num_candidates,
-                "num_trails": self.num_trails,
+                "num_trails": self.num_trials,
                 "dataset": dataset.name,
             },
         }

{opik_optimizer-0.7.5 → opik_optimizer-0.7.6/src/opik_optimizer.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opik_optimizer
-Version: 0.7.5
+Version: 0.7.6
 Summary: Agent optimization with Opik
 Home-page: https://github.com/comet-ml/opik
 Author: Comet ML