diff --git a/opto/optimizers/optoprime.py b/opto/optimizers/optoprime.py index 13548494..351e590b 100644 --- a/opto/optimizers/optoprime.py +++ b/opto/optimizers/optoprime.py @@ -349,7 +349,7 @@ def repr_node_constraint(node_dict): temp_list.append(f"(code) {k}: {v[1]}") return "\n".join(temp_list) - def probelm_instance(self, summary, mask=None): + def problem_instance(self, summary, mask=None): mask = mask or [] return ProblemInstance( instruction=self.objective if '#Instruction' not in mask else "", @@ -369,7 +369,7 @@ def construct_prompt(self, summary, mask=None, *args, **kwargs): """Construct the system and user prompt.""" system_prompt = self.representation_prompt + self.output_format_prompt # generic representation + output rule user_prompt = self.user_prompt_template.format( - problem_instance=str(self.probelm_instance(summary, mask=mask)) + problem_instance=str(self.problem_instance(summary, mask=mask)) ) # problem instance if self.include_example: user_prompt = ( @@ -429,7 +429,7 @@ def _step(self, verbose=False, mask=None, *args, **kwargs) -> Dict[ParameterNode if self.log is not None: self.log.append({"system_prompt": system_prompt, "user_prompt": user_prompt, "response": response}) - self.summary_log.append({'problem_instance': self.probelm_instance(summary), 'summary': summary}) + self.summary_log.append({'problem_instance': self.problem_instance(summary), 'summary': summary}) return update_dict diff --git a/opto/version.py b/opto/version.py index d1f2e39a..48fef323 100644 --- a/opto/version.py +++ b/opto/version.py @@ -1 +1 @@ -__version__ = "0.1.1" \ No newline at end of file +__version__ = "0.1.2" \ No newline at end of file diff --git a/setup.py b/setup.py index 6f4adafd..5f15139a 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ install_requires = [ - "pyautogen>=0.2.31", + "autogen-agentchat~=0.2", "graphviz>=0.20.1", "scikit-learn", "xgboost",