diff --git a/BUILD b/BUILD index c9bda93068..8d74b81436 100644 --- a/BUILD +++ b/BUILD @@ -23,7 +23,7 @@ python_requirements( "types-PyYAML": ["yaml"], "types-python-dateutil": ["dateutil", "dateutil.parser", "dateutil.tz"], "types-redis": ["redis"], - "types-six": ["six"], + "types-six": ["six", "graphql", "promise"], "types-tabulate": ["tabulate"], }, ) diff --git a/changes/1312.fix.md b/changes/1312.fix.md new file mode 100644 index 0000000000..ad5ee7ec5e --- /dev/null +++ b/changes/1312.fix.md @@ -0,0 +1 @@ +Handle buggy ORM field loading when destroy session. diff --git a/src/ai/backend/manager/registry.py b/src/ai/backend/manager/registry.py index 1304625127..2ffa47130f 100644 --- a/src/ai/backend/manager/registry.py +++ b/src/ai/backend/manager/registry.py @@ -1629,7 +1629,7 @@ async def destroy_session( await self.event_producer.produce_event( SessionCancelledEvent( session_id, - kernel.session_creation_id, + target_session.creation_id, reason, ), )