scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test: collect owned cleanup processes immutably

Browse source
This commit is contained in:
Yuneng Jiang 2026-09-14 04:18:59 -07:00
parent 16d7e695c0
commit 7ff8919827
No known key found for this signature in database
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
.circleci/scripts/stop_integration_processes.py
View file

@ -4,17 +4,17 @@ from typing import Final
import psutil
def is_owned(process: psutil.Process, identity: str, owner_uid: int) -> bool:
try:
return process.uids().real == owner_uid and process.environ().get("INTEGRATION_RUN_ID") == identity
except psutil.NoSuchProcess:
return False
def owned_processes(identity: str, owner_uid: int) -> tuple[psutil.Process, ...]:
owned: Final[list[psutil.Process]] = []
for process in psutil.process_iter():
try:
if process.uids().real != owner_uid:
continue
if process.environ().get("INTEGRATION_RUN_ID") == identity:
owned.append(process)
except psutil.NoSuchProcess:
continue
return tuple(owned)
# The runner invokes this helper with sudo. Unreadable ownership remains a hard
# failure: skipping it would make a successful cleanup unverifiable.
return tuple(process for process in psutil.process_iter() if is_owned(process, identity, owner_uid))
def main(identity: str, owner_uid: int, root_pids: tuple[int, ...]) -> int: