opik-optimizer 0.7.3__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.3.dist-info → opik_optimizer-0.7.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opik_optimizer
-Version: 0.7.3
+Version: 0.7.5
 Summary: Agent optimization with Opik
 Home-page: https://github.com/comet-ml/opik
 Author: Comet ML
@@ -22,6 +22,7 @@ Requires-Dist: pydantic
 Requires-Dist: pandas
 Requires-Dist: hf_xet
 Requires-Dist: pyrate-limiter
+Requires-Dist: deap>=1.4.3
 Provides-Extra: dev
 Requires-Dist: adalflow; extra == "dev"
 Requires-Dist: pytest; extra == "dev"

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

@@ -1,4 +1,4 @@
-opik_optimizer/__init__.py,sha256=x5QSFom-TtmmUNzqyYIJY2AujMJXbyhXs2oz44-4Af0,1121
+opik_optimizer/__init__.py,sha256=Tbny_W_oUWKdf_33fM-nNTBqvQvarLfB8hMNd_goXT0,1245
 opik_optimizer/_throttle.py,sha256=ztub8qlwz4u0GVA2TIoLig0D1Cs0hJ7_o_SnT_C7Nmk,1360
 opik_optimizer/base_optimizer.py,sha256=f4gNX9j3Z3TGst8F0gm1nMHHpHKAlChmeCVAcTdTIR4,4883
 opik_optimizer/cache_config.py,sha256=EzF4RAzxhSG8vtMJANdiUpNHQ9HzL2CrCXp0iik0f4A,580
@@ -7,9 +7,12 @@ opik_optimizer/meta_prompt_optimizer.py,sha256=AoFskD01Mu9VbH-Ys4CUoqZk3Gn3iFrOz
 opik_optimizer/optimization_result.py,sha256=9zdDV2MXeLYk7U8OqgMmSU-DdPV6qgYQWS2rtkO6Dzw,8693
 opik_optimizer/task_evaluator.py,sha256=MafDMaLeW0_yGPrumLvYF0HzQUKrnpAlM_0N_TPG8tw,3695
 opik_optimizer/utils.py,sha256=HivUsNzbt7BcuZeEvikdER1DaTPUFLJrpaVQ8raZYD8,3637
+opik_optimizer/data/hotpot-500.json,sha256=ToYC9l3lebrOoeWFhELgp63XVFYWcS0HODlQO9g5V3c,101861
 opik_optimizer/demo/__init__.py,sha256=KSpFYhzN7fTmLEsIaciRHwxcJDeAiX5NDmYLdPsfpT8,150
 opik_optimizer/demo/cache.py,sha256=5WqK8rSiijzU6s4VHIjLuL1LR5i1yHtY-x5FZTduSus,3669
-opik_optimizer/demo/datasets.py,sha256=hD6JZAQotEDQb4nK7dbnurquILqQsrFRF7nUwon_iXE,22930
+opik_optimizer/demo/datasets.py,sha256=-qmdeFk5X3CIcg9QX7nbGUAVo2khh0XGbmkHh8jgvSY,22633
+opik_optimizer/evolutionary_optimizer/__init__.py,sha256=OQ2ART5g-7EVGOISvTGY-AbmEyyDFEJJCsmJBzGJIpw,57
+opik_optimizer/evolutionary_optimizer/evolutionary_optimizer.py,sha256=652TAHa4U4C7_0_wBulDmizfVijqV9E3ngR38nLcpt4,76568
 opik_optimizer/few_shot_bayesian_optimizer/__init__.py,sha256=VuH7FOROyGcjMPryejtZC-5Y0QHlVTFLTGUDgNqRAFw,113
 opik_optimizer/few_shot_bayesian_optimizer/few_shot_bayesian_optimizer.py,sha256=HIoIVDW9v0p2Y9B736DncXjJPPYoDfphgeSqiB4MIVM,15235
 opik_optimizer/few_shot_bayesian_optimizer/prompt_parameter.py,sha256=EDsSIFAUOfiZKWLrOAaBDB7Exk7cmIs4ccI95kVa7JY,3118
@@ -17,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.3.dist-info/licenses/LICENSE,sha256=dTRSwwCHdWeSjzodvnivYqcwi8x3Qfr21yv65QUWWBE,1062
-opik_optimizer-0.7.3.dist-info/METADATA,sha256=ng9ZwnuLM631y5mrK-DSi70r763Db51thVyE1QNuQlI,3826
-opik_optimizer-0.7.3.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-opik_optimizer-0.7.3.dist-info/top_level.txt,sha256=ondOlpq6_yFckqpxoAHSfzZS2N-JfgmA-QQhOJfz7m0,15
-opik_optimizer-0.7.3.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,,

{opik_optimizer-0.7.3.dist-info → opik_optimizer-0.7.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any