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

feat: vm_providers package public re-exports (LIT-2878)

Browse source
This commit is contained in:
Ishaan Jaff 2026-05-06 15:26:10 -07:00
parent a6c10b75c4
commit b3a6dd0f4c
No known key found for this signature in database
1 changed files with 46 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
litellm/proxy/agent_session_endpoints/vm_providers/__init__.py Normal file
View file

@ -0,0 +1,46 @@
"""
Pluggable VM providers for `agent_session_endpoints`.
Public re-exports:
- `AgentVMProvider` — the ABC every provider implements
- `ProvisionContext` / `VMHandle` / `VMStatus` / `VMState` — typed I/O
- `AwsCreds` / `Ec2Config` — BYOC inputs
- `get_vm_provider` — factory keyed off `agent_settings.vm_provider`
- `ProvisionError` / `InvalidCredentialsError` — error types
"""
from litellm.proxy.agent_session_endpoints.vm_providers.base import (
AgentVMProvider,
AwsCreds,
Ec2Config,
InvalidCredentialsError,
ProvisionContext,
ProvisionError,
Repo,
VMHandle,
VMState,
VMStatus,
)
from litellm.proxy.agent_session_endpoints.vm_providers.ec2 import EC2Provider
from litellm.proxy.agent_session_endpoints.vm_providers.factory import (
SUPPORTED_PROVIDERS,
get_vm_provider,
)
from litellm.proxy.agent_session_endpoints.vm_providers.noop import NoopProvider
__all__ = [
"AgentVMProvider",
"AwsCreds",
"Ec2Config",
"EC2Provider",
"InvalidCredentialsError",
"NoopProvider",
"ProvisionContext",
"ProvisionError",
"Repo",
"SUPPORTED_PROVIDERS",
"VMHandle",
"VMState",
"VMStatus",
"get_vm_provider",
]