scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

refactor(managed_agents/vms/warm_pool): move warm_pool/manager.py out of proxy

Browse source 
Phase 2: warm pool is part of VM management — moves to
litellm/managed_agents/vms/warm_pool/ alongside the providers.

Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:30:02 -07:00
parent b265c750c7
commit 32cfd1b00c
No known key found for this signature in database
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/agent_session_endpoints/warm_pool/manager.py → litellm/managed_agents/vms/warm_pool/manager.py
View file

@ -23,16 +23,16 @@ from datetime import datetime, timedelta, timezone
from typing import Any, Dict, List, Optional
from litellm._logging import verbose_proxy_logger
from litellm.proxy.agent_session_endpoints.vm_providers.base import (
from litellm.managed_agents.vms.base import (
AwsCreds,
Ec2Config,
ProvisionContext,
Repo,
)
from litellm.proxy.agent_session_endpoints.vm_providers.factory import (
from litellm.managed_agents.vms.factory import (
build_vm_provider,
)
from litellm.proxy.agent_session_endpoints.vm_providers.team_config import (
from litellm.managed_agents.vms.team_config import (
get_team_vm_config,
)
@ -330,7 +330,7 @@ class WarmPoolManager:
provider = build_vm_provider(
{"vm_provider": "ec2", "ec2": self._provider_settings}
)
from litellm.proxy.agent_session_endpoints.vm_providers.base import (
from litellm.managed_agents.vms.base import (
VMHandle,
)