Update requirements and insfra versions
This commit is contained in:
@@ -311,6 +311,7 @@ async def websocket_training_progress(websocket: WebSocket, tenant_id: str, job_
|
||||
try:
|
||||
# Connect to training service WebSocket
|
||||
import websockets
|
||||
from websockets.protocol import State
|
||||
|
||||
training_ws = await websockets.connect(
|
||||
training_ws_url,
|
||||
@@ -325,7 +326,7 @@ async def websocket_training_progress(websocket: WebSocket, tenant_id: str, job_
|
||||
async def forward_frontend_to_training():
|
||||
"""Forward messages from frontend to training service"""
|
||||
try:
|
||||
while training_ws and training_ws.open:
|
||||
while training_ws and training_ws.state == State.OPEN:
|
||||
data = await websocket.receive()
|
||||
|
||||
if data.get("type") == "websocket.receive":
|
||||
@@ -342,7 +343,7 @@ async def websocket_training_progress(websocket: WebSocket, tenant_id: str, job_
|
||||
"""Forward messages from training service to frontend"""
|
||||
message_count = 0
|
||||
try:
|
||||
while training_ws and training_ws.open:
|
||||
while training_ws and training_ws.state == State.OPEN:
|
||||
message = await training_ws.recv()
|
||||
await websocket.send_text(message)
|
||||
message_count += 1
|
||||
@@ -369,7 +370,7 @@ async def websocket_training_progress(websocket: WebSocket, tenant_id: str, job_
|
||||
logger.error("WebSocket proxy error", job_id=job_id, error=str(e))
|
||||
finally:
|
||||
# Cleanup
|
||||
if training_ws and not training_ws.closed:
|
||||
if training_ws and training_ws.state == State.OPEN:
|
||||
try:
|
||||
await training_ws.close()
|
||||
except:
|
||||
|
||||
@@ -1,19 +1,20 @@
|
||||
fastapi==0.104.1
|
||||
uvicorn[standard]==0.24.0
|
||||
httpx==0.25.2
|
||||
redis==5.0.1
|
||||
pydantic==2.5.0
|
||||
pydantic-settings==2.1.0
|
||||
fastapi==0.119.0
|
||||
uvicorn[standard]==0.32.1
|
||||
httpx==0.28.1
|
||||
redis==6.4.0
|
||||
pydantic==2.12.3
|
||||
pydantic-settings==2.7.1
|
||||
python-jose[cryptography]==3.3.0
|
||||
PyJWT==2.8.0
|
||||
PyJWT==2.10.1
|
||||
python-multipart==0.0.6
|
||||
prometheus-client==0.17.1
|
||||
python-json-logger==2.0.4
|
||||
email-validator==2.0.0
|
||||
aio-pika==9.3.0
|
||||
pytz==2023.3
|
||||
prometheus-client==0.23.1
|
||||
python-json-logger==3.3.0
|
||||
email-validator==2.2.0
|
||||
aio-pika==9.4.3
|
||||
pytz==2024.2
|
||||
python-logstash==0.4.8
|
||||
structlog==23.2.0
|
||||
websockets==12.0
|
||||
sqlalchemy==2.0.23
|
||||
asyncpg==0.29.0
|
||||
structlog==25.4.0
|
||||
websockets==14.1
|
||||
sqlalchemy==2.0.44
|
||||
asyncpg==0.30.0
|
||||
cryptography==44.0.0
|
||||
|
||||
Reference in New Issue
Block a user