Update app.py

app.py

@@ -513,42 +513,23 @@ async def reconnect_for_token_refresh():
         logger.error(f"令牌刷新失败: {e}")
 
 async def async_main():
-    if not CONFIG['LOGIN']['USER_ID'] or not CONFIG['LOGIN']['AUTH']:
-        logger.error("用户ID或授权令牌未设置")
-        return
-    headers = {
-        "authorization": f"{CONFIG['LOGIN']['USER_ID']} {CONFIG['LOGIN']['AUTH']}",
-        "x-zed-protocol-version": "68",
-        "x-zed-app-version": "0.178.0",
-        "x-zed-release-channel": "stable"
-    }
-    ssl_context = ssl.create_default_context()
-    ssl_context.check_hostname = False
-    ssl_context.verify_mode = ssl.CERT_NONE
-    logger.info("连接到Websocket服务器")
-    async for websocket in connect(CONFIG['API']['WS_URL'], additional_headers=headers, ssl=ssl_context):
-        try:
-            ping_task = asyncio.create_task(ping_periodically(websocket))
-            token_request_task = asyncio.create_task(delayed_token_request(websocket, delay=2))
-            await handle_messages(websocket)
-            break
-        except ConnectionClosed:
-            continue
-        except Exception as e:
-            logger.error(f"意外错误: {e}")
-            await asyncio.sleep(1)
-            continue
-        finally:
-            ping_task.cancel()
-            try:
-                await ping_task
-            except asyncio.CancelledError:
-                pass
-            token_request_task.cancel()
-            try:
-                await token_request_task
-            except asyncio.CancelledError:
-                pass
+    app = web.Application()
+    app.router.add_post('/v1/messages', handle_message_request)
+    app.router.add_post('/v1/chat/completions', handle_message_request)
+    app.router.add_get('/v1/models', handle_models_request)
+    
+    async def health_check():
+        return web.json_response({
+            "status": "ok",
+            "message": "Zed LLM proxy is running"
+        })
+
+    app.router.add_get('/', health_check)
+
+    runner = web.AppRunner(app)
+    await runner.setup()
+    site = web.TCPSite(runner, 'localhost', CONFIG['SERVER']['PORT'])
+    await site.start()
 
 async def delayed_token_request(websocket, delay=2):
     await asyncio.sleep(delay)