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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (46) hide show
  1. {opik_optimizer-0.7.6/src/opik_optimizer.egg-info → opik_optimizer-0.7.7}/PKG-INFO +1 -1
  2. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/setup.py +1 -1
  3. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/mipro_optimizer/mipro_optimizer.py +4 -2
  4. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7/src/opik_optimizer.egg-info}/PKG-INFO +1 -1
  5. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/LICENSE +0 -0
  6. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/README.md +0 -0
  7. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/setup.cfg +0 -0
  8. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/__init__.py +0 -0
  9. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/_throttle.py +0 -0
  10. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/base_optimizer.py +0 -0
  11. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/cache_config.py +0 -0
  12. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/data/hotpot-500.json +0 -0
  13. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/demo/__init__.py +0 -0
  14. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/demo/cache.py +0 -0
  15. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/demo/datasets.py +0 -0
  16. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/evolutionary_optimizer/__init__.py +0 -0
  17. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/evolutionary_optimizer/evolutionary_optimizer.py +0 -0
  18. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/few_shot_bayesian_optimizer/__init__.py +0 -0
  19. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/few_shot_bayesian_optimizer/few_shot_bayesian_optimizer.py +0 -0
  20. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/few_shot_bayesian_optimizer/prompt_parameter.py +0 -0
  21. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/few_shot_bayesian_optimizer/prompt_templates.py +0 -0
  22. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/integrations/__init__.py +0 -0
  23. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/logging_config.py +0 -0
  24. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/meta_prompt_optimizer.py +0 -0
  25. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/mipro_optimizer/__init__.py +0 -0
  26. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/mipro_optimizer/_lm.py +0 -0
  27. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/mipro_optimizer/_mipro_optimizer_v2.py +0 -0
  28. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/mipro_optimizer/utils.py +0 -0
  29. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/optimization_config/__init__.py +0 -0
  30. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/optimization_config/configs.py +0 -0
  31. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/optimization_config/mappers.py +0 -0
  32. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/optimization_result.py +0 -0
  33. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/task_evaluator.py +0 -0
  34. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer/utils.py +0 -0
  35. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer.egg-info/SOURCES.txt +0 -0
  36. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer.egg-info/dependency_links.txt +0 -0
  37. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer.egg-info/requires.txt +0 -0
  38. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/src/opik_optimizer.egg-info/top_level.txt +0 -0
  39. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_base_optimizer.py +0 -0
  40. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_example.py +0 -0
  41. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_few_shot_bayesian_optimizer.py +0 -0
  42. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_mappers.py +0 -0
  43. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_optimization_dsl.py +0 -0
  44. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_optimization_result.py +0 -0
  45. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_task_evaluator.py +0 -0
  46. {opik_optimizer-0.7.6 → opik_optimizer-0.7.7}/tests/test_utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: opik_optimizer
3
- Version: 0.7.6
3
+ Version: 0.7.7
4
4
  Summary: Agent optimization with Opik
5
5
  Home-page: https://github.com/comet-ml/opik
6
6
  Author: Comet ML
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
2
2
 
3
3
  setup(
4
4
  name="opik_optimizer",
5
- version="0.7.6",
5
+ version="0.7.7",
6
6
  description="Agent optimization with Opik",
7
7
  author="Comet ML",
8
8
  author_email="support@comet.com",
@@ -1,4 +1,4 @@
1
- from typing import Any, Dict, List, Tuple, Union, Optional
1
+ from typing import Any, Dict, List, Tuple, Union, Optional, Literal
2
2
  import os
3
3
  import random
4
4
 
@@ -271,6 +271,7 @@ class MiproOptimizer(BaseOptimizer):
271
271
  experiment_config: Optional[Dict] = None,
272
272
  optimization_id: Optional[str] = None,
273
273
  num_trials: Optional[int] = 3,
274
+ auto: Optional[Literal["light", "medium", "heavy"]] = "medium",
274
275
  **kwargs,
275
276
  ) -> None:
276
277
  # FIXME: Intermediate values:
@@ -285,6 +286,7 @@ class MiproOptimizer(BaseOptimizer):
285
286
  self.output_key = output_key
286
287
  self.prompt = prompt
287
288
  self.num_trials = num_trials
289
+ self.auto = auto
288
290
 
289
291
  # Convert to values for MIPRO:
290
292
  if isinstance(dataset, str):
@@ -333,7 +335,7 @@ class MiproOptimizer(BaseOptimizer):
333
335
  # Initialize the optimizer:
334
336
  self.optimizer = MIPROv2(
335
337
  metric=self.metric_function,
336
- auto="light",
338
+ auto=self.auto,
337
339
  num_threads=self.num_threads,
338
340
  verbose=False,
339
341
  num_candidates=self.num_candidates,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: opik_optimizer
3
- Version: 0.7.6
3
+ Version: 0.7.7
4
4
  Summary: Agent optimization with Opik
5
5
  Home-page: https://github.com/comet-ml/opik
6
6
  Author: Comet ML
File without changes
File without changes
File without changes