fix timeout

This commit is contained in:
yuanmengqi
2025-06-04 10:23:45 +00:00
parent b87cbe69e5
commit b211df3385
6 changed files with 56 additions and 15 deletions

View File

@@ -301,8 +301,7 @@ class OpenAICUAAgent:
Raises:
requests.exceptions.RequestException: If the API request fails
"""
retry_count = 0
while retry_count < 3:
while True:
try:
from openai import OpenAI
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY_CUA"))
@@ -319,13 +318,8 @@ class OpenAICUAAgent:
logger.info(f"Response: {response}")
return response
except Exception as e:
logger.error(f"OpenAI API error: {str(e)}")
new_screenshot = self.env._get_obs()
new_screenshot_base64 = base64.b64encode(new_screenshot["screenshot"]).decode('utf-8')
self.cua_messages[-1]["output"]["image_url"] = f"data:image/png;base64,{new_screenshot_base64}"
retry_count += 1
logger.error(f"OpenAI API error: {str(e)}will retry in 1s...")
time.sleep(1)
raise Exception("Failed to make OpenAI API call after 3 retries")
def _handle_item(self, item: Dict[str, Any]) -> Optional[Union[str, Dict[str, Any]]]:
"""Parse a response item from the OpenAI API.