diff --git a/libs/agent/agent/providers/omni/clients/oaicompat.py b/libs/agent/agent/providers/omni/clients/oaicompat.py index 6a95896a..b15515fd 100644 --- a/libs/agent/agent/providers/omni/clients/oaicompat.py +++ b/libs/agent/agent/providers/omni/clients/oaicompat.py @@ -93,7 +93,14 @@ class OAICompatClient(BaseOmniClient): """ headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.api_key}"} - final_messages = [{"role": "system", "content": system}] + final_messages = [ + { + "role": "system", + "content": [ + { "type": "text", "text": system } + ] + } + ] # Process messages for item in messages: @@ -117,7 +124,10 @@ class OAICompatClient(BaseOmniClient): else: message = { "role": item["role"], - "content": [{"type": "text", "text": item["content"]}], + "content": [{ + "type": "text", + "text": item["content"] + }], } final_messages.append(message) else: diff --git a/libs/agent/agent/providers/uitars/loop.py b/libs/agent/agent/providers/uitars/loop.py index 84393bd2..71d2c739 100644 --- a/libs/agent/agent/providers/uitars/loop.py +++ b/libs/agent/agent/providers/uitars/loop.py @@ -232,8 +232,11 @@ class UITARSLoop(BaseLoop): if self.client is None: raise RuntimeError("Failed to initialize client") - # Convert messages to UI-TARS format + # Get messages in standard format from the message manager + self.message_manager.messages = messages.copy() prepared_messages = self.message_manager.get_messages() + + # Convert messages to UI-TARS format uitars_messages = self.to_uitars_format(prepared_messages) # Log request