opik-optimizer 0.7.4__py3-none-any.whl → 0.7.5__py3-none-any.whl

opik_optimizer/mipro_optimizer/_mipro_optimizer_v2.py

@@ -54,6 +54,16 @@ from opik_optimizer import task_evaluator
 from opik_optimizer.optimization_config.configs import MetricConfig, TaskConfig
 from opik_optimizer.optimization_config import mappers
 
+def get_prompt(program):
+    """
+    Get the system prompt from the program
+    """
+    if hasattr(program, "react"):
+        instructions = program.react.signature.instructions
+    else:
+        instructions = program.signature.instructions
+
+    return instructions
 
 class MIPROv2(Teleprompter):
     def __init__(
@@ -547,7 +557,7 @@ class MIPROv2(Teleprompter):
             for l in demo:
                 for example in l:
                     examples.append(example.toDict())
-        prompt = program.signature.instructions
+        prompt = get_prompt(program)
         experiment_config = {
             **self.experiment_config,
             **{"configuration": {
@@ -928,7 +938,7 @@ class MIPROv2(Teleprompter):
             for l in demo:
                 for example in l:
                     examples.append(example.toDict())
-        prompt = highest_mean_program.signature.instructions
+        prompt = get_prompt(highest_mean_program)
         experiment_config = {
             **self.experiment_config,
             **{"configuration": {

opik_optimizer/mipro_optimizer/mipro_optimizer.py

@@ -196,6 +196,7 @@ class MiproOptimizer(BaseOptimizer):
         task_config: TaskConfig,
         num_candidates: int = 10,
         experiment_config: Optional[Dict] = None,
+        num_trials: Optional[int] = 3,
         **kwargs,
     ) -> OptimizationResult:
         self._opik_client = opik.Opik()
@@ -222,6 +223,7 @@ class MiproOptimizer(BaseOptimizer):
                 num_candidates=num_candidates,
                 experiment_config=experiment_config,
                 optimization_id=optimization.id if optimization is not None else None,
+                num_trials=num_trials,
                 **kwargs,
             )
             if optimization:
@@ -241,6 +243,7 @@ class MiproOptimizer(BaseOptimizer):
         num_candidates: int = 10,
         experiment_config: Optional[Dict] = None,
         optimization_id: Optional[str] = None,
+        num_trials: Optional[int] = 3,
         **kwargs,
     ) -> OptimizationResult:
         logger.info("Preparing MIPRO optimization...")
@@ -251,6 +254,7 @@ class MiproOptimizer(BaseOptimizer):
             num_candidates=num_candidates,
             experiment_config=experiment_config,
             optimization_id=optimization_id,
+            num_trials=num_trials,
             **kwargs,
         )
         logger.info("Starting MIPRO compilation...")
@@ -266,6 +270,7 @@ class MiproOptimizer(BaseOptimizer):
         num_candidates: int = 10,
         experiment_config: Optional[Dict] = None,
         optimization_id: Optional[str] = None,
+        num_trials: Optional[int] = 3,
         **kwargs,
     ) -> None:
         # FIXME: Intermediate values:
@@ -279,6 +284,7 @@ class MiproOptimizer(BaseOptimizer):
         self.input_key = input_key
         self.output_key = output_key
         self.prompt = prompt
+        self.num_trials = num_trials
 
         # Convert to values for MIPRO:
         if isinstance(dataset, str):
@@ -319,6 +325,7 @@ class MiproOptimizer(BaseOptimizer):
                 "metric": metric.name,
                 "num_threads": self.num_threads,
                 "num_candidates": self.num_candidates,
+                "num_trails": self.num_trails,
                 "dataset": dataset.name,
             },
         }
@@ -359,7 +366,7 @@ class MiproOptimizer(BaseOptimizer):
             trainset=self.trainset,
             provide_traceback=True,
             requires_permission_to_run=False,
-            num_trials=3,
+            num_trials=self.num_trials,
         )
         self.best_programs = sorted(
             self.results.candidate_programs,

{opik_optimizer-0.7.4.dist-info → opik_optimizer-0.7.5.dist-info}/METADATA

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

{opik_optimizer-0.7.4.dist-info → opik_optimizer-0.7.5.dist-info}/RECORD

@@ -20,14 +20,14 @@ opik_optimizer/few_shot_bayesian_optimizer/prompt_templates.py,sha256=HmvD-UeT3a
 opik_optimizer/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 opik_optimizer/mipro_optimizer/__init__.py,sha256=CF9TVXjOxTobDO1kAS8CD4eyLVzEozxjfgoKwIO6ZpU,44
 opik_optimizer/mipro_optimizer/_lm.py,sha256=bcTy2Y5HjSaFQOATIpUaA86eIp3vKHaMuDI2_RvN2ww,16376
-opik_optimizer/mipro_optimizer/_mipro_optimizer_v2.py,sha256=r8FKaqvtZq_R7FwGnXqp1foCLk7M7r6M-CMvWbJtP5c,39512
-opik_optimizer/mipro_optimizer/mipro_optimizer.py,sha256=5QS7OKqOMKe4CD_8W2FMD_qJNmulkvxmOT_YtJ3BllM,14755
+opik_optimizer/mipro_optimizer/_mipro_optimizer_v2.py,sha256=CiQWe39LCp-81SZmLyMybIw2lc_0RBKcxclLEuSXQgI,39757
+opik_optimizer/mipro_optimizer/mipro_optimizer.py,sha256=VCQ0V1U_3aEsCqBumu127QLUEjkPA-kRWZou1qtZ9gU,15044
 opik_optimizer/mipro_optimizer/utils.py,sha256=4et1JA1QInX3h6Is-_RqzliFwJqkm6tlA0X5CryG60I,3142
 opik_optimizer/optimization_config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 opik_optimizer/optimization_config/configs.py,sha256=MYL9H2UAqeyGBlBGWbOZ-6Snto4ZMuXnypgvVuUSW1Y,1132
 opik_optimizer/optimization_config/mappers.py,sha256=RXgTMxPzTQ1AHGke6Zca6rTcfCI7IkCKhQYciaEGSAo,1698
-opik_optimizer-0.7.4.dist-info/licenses/LICENSE,sha256=dTRSwwCHdWeSjzodvnivYqcwi8x3Qfr21yv65QUWWBE,1062
-opik_optimizer-0.7.4.dist-info/METADATA,sha256=f6Sdl2YLvbLhLxn51e13YaejNITxjMd2HgrCvMYtcVY,3853
-opik_optimizer-0.7.4.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-opik_optimizer-0.7.4.dist-info/top_level.txt,sha256=ondOlpq6_yFckqpxoAHSfzZS2N-JfgmA-QQhOJfz7m0,15
-opik_optimizer-0.7.4.dist-info/RECORD,,
+opik_optimizer-0.7.5.dist-info/licenses/LICENSE,sha256=dTRSwwCHdWeSjzodvnivYqcwi8x3Qfr21yv65QUWWBE,1062
+opik_optimizer-0.7.5.dist-info/METADATA,sha256=_1cqmztaDxXDGs0It8INn18IuQAVpWomu0bKITYkkBI,3853
+opik_optimizer-0.7.5.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+opik_optimizer-0.7.5.dist-info/top_level.txt,sha256=ondOlpq6_yFckqpxoAHSfzZS2N-JfgmA-QQhOJfz7m0,15
+opik_optimizer-0.7.5.dist-info/RECORD,,