diff --git a/litellm/proxy/agent_session_endpoints/vm_providers/__init__.py b/litellm/proxy/agent_session_endpoints/vm_providers/__init__.py new file mode 100644 index 00000000000..44aeb8cfd7d --- /dev/null +++ b/litellm/proxy/agent_session_endpoints/vm_providers/__init__.py @@ -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", +]