scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

fix(vm_providers): registry adapter threads session_id into terminate metadata for tests

Browse source
This commit is contained in:
Ishaan Jaffer 2026-05-06 16:05:04 -07:00
parent 29146f012b
commit d955639aa9
No known key found for this signature in database
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/agent_session_endpoints/vm_providers/registry.py
View file

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