diff --git a/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer.py b/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer.py index 793a720..94d0aed 100644 --- a/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer.py +++ b/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer.py @@ -58,7 +58,7 @@ def generate(self, prompt: str, temperature: float, max_tokens: int) -> str: model=self.model, messages=[{"role": "user", "content": prompt}], temperature=temperature, - max_tokens=max_tokens, + max_completion_tokens=max_tokens, ) return response.choices[0].message.content diff --git a/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer_cfg.py b/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer_cfg.py index 8f81a09..df60f01 100644 --- a/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer_cfg.py +++ b/source/autosim/autosim/decomposers/llm_decomposer/llm_decomposer_cfg.py @@ -21,7 +21,7 @@ class LLMDecomposerCfg(DecomposerCfg): base_url: str = "https://api.openai.com/v1" """The base URL for the LLM API.""" - model: str = "gpt-3.5-turbo" + model: str = "gpt-5.4" """The model name for the LLM.""" temperature: float = 0.3 diff --git a/source/autosim/autosim/decomposers/llm_decomposer/prompts/task_decompose.jinja b/source/autosim/autosim/decomposers/llm_decomposer/prompts/task_decompose.jinja index a76b923..e1ec1ee 100644 --- a/source/autosim/autosim/decomposers/llm_decomposer/prompts/task_decompose.jinja +++ b/source/autosim/autosim/decomposers/llm_decomposer/prompts/task_decompose.jinja @@ -80,7 +80,7 @@ Output the task decomposition result in JSON format with the following fields: 5. **Push/Pull Pattern** (for pushing/pulling objects): ``` - moveto(object) → reach(object) → grasp(object) → push(forward) # or pull(back) + moveto(object) → reach(object) → grasp(object) → push(object) # or pull(object) ``` 6. **Combined Patterns**: For complex tasks, combine multiple patterns in sequence.