Update app.py

app.py

@@ -77,13 +77,7 @@ def format_model_name(model):
 async def handle_root(request):
     logger.info("Received request to root path /")
     return web.Response(text="GCP Vertex Claude API Proxy", status=200)
-
-async def safe_write(response, data):
-    try:
-        await response.write(data)
-    except ConnectionResetError:
-        logger.warning("Failed to write response, client likely disconnected")
-
+    
 async def handle_request(request):
     logger.info(f"Received {request.method} request to {request.path}")
     
@@ -129,47 +123,37 @@ async def handle_request(request):
 
     logger.info(f"Sending request to Anthropic API: {api_url}")
     
+    # 创建一个 StreamResponse 对象
     response = StreamResponse(status=200)
     response.headers['Content-Type'] = 'application/json'
     response.headers['Access-Control-Allow-Origin'] = '*'
     response.headers['Access-Control-Allow-Methods'] = 'POST, GET, OPTIONS'
     response.headers['Access-Control-Allow-Headers'] = 'Content-Type, Authorization, x-api-key, anthropic-version, model'
     
-    try:
-        await response.prepare(request)
+    await response.prepare(request)
 
+    try:
         async with aiohttp.ClientSession() as session:
             async with session.post(api_url, json=request_body, headers=headers, timeout=ClientTimeout(total=30)) as api_response:
                 api_response.raise_for_status()
                 logger.info(f"Received response from Anthropic API. Status: {api_response.status}")
                 logger.info(f"Response headers: {dict(api_response.headers)}")
                 
-                total_bytes_sent = 0
                 async for chunk in api_response.content.iter_any():
-                    try:
-                        await response.write(chunk)
-                        total_bytes_sent += len(chunk)
-                        logger.debug(f"Sent {total_bytes_sent} bytes to client")
-                    except ConnectionResetError:
-                        logger.warning(f"Client closed connection prematurely after receiving {total_bytes_sent} bytes")
-                        break
+                    await response.write(chunk)
                     
-                logger.info(f"Finished streaming response from Anthropic API. Total bytes sent: {total_bytes_sent}")
+                logger.info("Finished streaming response from Anthropic API")
     except aiohttp.ClientError as e:
         logger.error(f"Error communicating with Anthropic API: {str(e)}")
-        await safe_write(response, json.dumps({'error': 'Error communicating with AI service'}).encode())
+        await response.write(json.dumps({'error': 'Error communicating with AI service'}).encode())
     except asyncio.TimeoutError:
         logger.error("Request to Anthropic API timed out")
-        await safe_write(response, json.dumps({'error': 'Request timed out'}).encode())
+        await response.write(json.dumps({'error': 'Request timed out'}).encode())
     except Exception as e:
         logger.error(f"Unexpected error: {str(e)}")
-        await safe_write(response, json.dumps({'error': 'An unexpected error occurred'}).encode())
+        await response.write(json.dumps({'error': 'An unexpected error occurred'}).encode())
     finally:
-        try:
-            await response.write_eof()
-            logger.info("Successfully wrote EOF to response")
-        except ConnectionResetError:
-            logger.warning("Failed to write EOF, client likely disconnected")
+        await response.write_eof()
         logger.info("Finished sending response back to client")
 
     return response