diff --git a/oai.py b/oai.py
index be8c1b7..f64a7fe 100644
--- a/oai.py
+++ b/oai.py
@@ -5034,9 +5034,18 @@ All queries are read-only. INSERT/UPDATE/DELETE are not allowed."""
                     try:
                         if hasattr(response, 'close'):
                             response.close()
+                        elif hasattr(response, '__exit__'):
+                            response.__exit__(None, None, None)
                     except:
                         pass
                     
+                    # Recreate client to be safe
+                    try:
+                        client = OpenRouter(api_key=API_KEY)
+                        app_logger.info("Client recreated after stream interruption")
+                    except:
+                        pass
+    
                     continue  # Now it's safe to continue
             else:
                 full_response = response.choices[0].message.content if response.choices else ""