From 2e10e0922ab1b883d3bb4cae0f14db9fb9784ea7 Mon Sep 17 00:00:00 2001 From: Dillon DuPont Date: Tue, 29 Apr 2025 12:09:30 -0700 Subject: [PATCH] add top_p to uitars --- libs/agent/agent/providers/uitars/clients/oaicompat.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/libs/agent/agent/providers/uitars/clients/oaicompat.py b/libs/agent/agent/providers/uitars/clients/oaicompat.py index 963fb05b..1b512997 100644 --- a/libs/agent/agent/providers/uitars/clients/oaicompat.py +++ b/libs/agent/agent/providers/uitars/clients/oaicompat.py @@ -145,8 +145,13 @@ class OAICompatClient(BaseUITarsClient): message = {"role": "user", "content": [{"type": "text", "text": item}]} final_messages.append(message) - payload = {"model": self.model, "messages": final_messages, "temperature": self.temperature} - payload["max_tokens"] = max_tokens or self.max_tokens + payload = { + "model": self.model, + "messages": final_messages, + "max_tokens": max_tokens or self.max_tokens, + "temperature": self.temperature, + "top_p": 0.7, + } try: async with aiohttp.ClientSession() as session: