From 095fe3f59feccd1aead0eb6063121d35e9473881 Mon Sep 17 00:00:00 2001
From: katiue <vinhkhhangnguyen38@gmail.com>
Date: Mon, 13 Jan 2025 14:21:45 +0700
Subject: [PATCH] suppress socket error

---
 src/utils/utils.py |  1 -
 webui.py           | 19 +++++++++++++++++++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/utils/utils.py b/src/utils/utils.py
index 81698b9..7e05dda 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -15,7 +15,6 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_ollama import ChatOllama
 from langchain_openai import AzureChatOpenAI, ChatOpenAI
 import gradio as gr
-from src.browser.custom_context import CustomBrowserContext
 
 def get_llm_model(provider: str, **kwargs):
     """
diff --git a/webui.py b/webui.py
index 85d44a7..7aac425 100644
--- a/webui.py
+++ b/webui.py
@@ -16,6 +16,8 @@ import glob
 import asyncio
 import argparse
 import os
+import warnings
+import socket
 
 logger = logging.getLogger(__name__)
 
@@ -51,6 +53,19 @@ _global_browser_context = None
 # Create the global agent state instance
 _global_agent_state = AgentState()
 
+def suppress_connection_reset(func):
+    async def wrapper(*args, **kwargs):
+        try:
+            return await func(*args, **kwargs)
+        except ConnectionResetError:
+            # Suppress the ConnectionResetError
+            warnings.warn("Connection was reset. This is usually harmless.", RuntimeWarning)
+            pass
+        except socket.error as e:
+            if e.winerror != 10054:  # If it's not the specific connection reset error
+                raise
+    return wrapper
+
 async def stop_agent():
     """Request the agent to stop and update UI with enhanced feedback"""
     global _global_agent_state, _global_browser_context, _global_browser
@@ -538,6 +553,7 @@ theme_map = {
     "Base": Base()
 }
 
+@suppress_connection_reset
 async def close_global_browser():
     global _global_browser, _global_browser_context
 
@@ -853,6 +869,9 @@ def create_ui(theme_name="Ocean"):
     return demo
 
 def main():
+    # Suppress asyncio connection reset warnings
+    warnings.filterwarnings("ignore", message=".*forcibly closed.*")
+    
     parser = argparse.ArgumentParser(description="Gradio UI for Browser Agent")
     parser.add_argument("--ip", type=str, default="127.0.0.1", help="IP address to bind to")
     parser.add_argument("--port", type=int, default=7788, help="Port to listen on")