From d955639aa988fad2cab3c05161f5ca74bd69e18e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 16:05:04 -0700 Subject: [PATCH] fix(vm_providers): registry adapter threads session_id into terminate metadata for tests --- .../agent_session_endpoints/vm_providers/registry.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/agent_session_endpoints/vm_providers/registry.py b/litellm/proxy/agent_session_endpoints/vm_providers/registry.py index 1130a10527b..8ab1c5b5257 100644 --- a/litellm/proxy/agent_session_endpoints/vm_providers/registry.py +++ b/litellm/proxy/agent_session_endpoints/vm_providers/registry.py @@ -79,9 +79,11 @@ class _ProviderAdapter: return from litellm.proxy.agent_session_endpoints.vm_providers.base import VMHandle - handle = VMHandle( - vm_id=vm_id, provider=self.name, metadata=dict(metadata or {}) - ) + # Carry session_id in metadata so test/recording noops can correlate + # the terminate call to the originating session without changing the + # ABC's terminate(handle) signature. + handle_metadata = {**(metadata or {}), "session_id": session_id} + handle = VMHandle(vm_id=vm_id, provider=self.name, metadata=handle_metadata) try: await self._inner.terminate(handle) except TypeError: