Update requirements and insfra versions

This commit is contained in:
Urtzi Alfaro
2025-10-17 23:09:40 +02:00
parent 7e089b80cf
commit 312e36c893
81 changed files with 524 additions and 476 deletions

View File

@@ -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:

View File

@@ -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