ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter9/gaia-experience/AWorld/aworld/sandbox/models.py

@@ -0,0 +1,66 @@
+import uuid
+from dataclasses import dataclass
+from enum import Enum
+
+from pydantic import BaseModel
+from typing_extensions import Optional, Dict,Any
+
+
+class SandboxStatus(Enum):
+    """Sandbox status enumeration."""
+    INIT = 'Pending'       # Initialization state
+    RUNNING = 'Running'    # Running
+    STOPPED = 'Stopped'    # Stopped
+    ERROR = 'Failed'      # Error state
+    REMOVED = 'Removed'    # Removed
+    UNKNOWN = 'Unknown'  # Removed
+
+class SandboxEnvType(Enum):
+    """Sandbox env type enumeration."""
+    LOCAL = 1
+    K8S = 2
+    SUPERCOMPUTER = 3
+
+
+
+@dataclass
+class SandboxCreateResponse:
+    sandbox_id: str = str(uuid.uuid4())
+    env_type: int = SandboxEnvType.LOCAL
+    status: Optional[str] = None
+    mcp_config: Optional[Any] = None
+
+@dataclass
+class SandboxK8sResponse(SandboxCreateResponse):
+    pod_name: Optional[str] = None
+    service_name: Optional[str] = None
+    cluster_ip: Optional[str] = None
+    host: Optional[str] = None
+
+
+@dataclass
+class SandboxLocalResponse(SandboxCreateResponse):
+    host: Optional[str] = None
+
+@dataclass
+class SandboxSuperResponse(SandboxCreateResponse):
+    host: Optional[str] = None
+
+@dataclass
+class SandboxInfo:
+    """Information about a sandbox."""
+
+    sandbox_id: str
+    """Sandbox ID."""
+    status: str
+    """sandbox status"""
+    metadata: Dict[str, str]
+    """Saved sandbox metadata."""
+
+
+
+class EnvConfig(BaseModel):
+    """Data structure contained in the environment"""
+    name: str = "default"
+    version: str = "1.0.0"
+    dockerfile: Optional[str] = None    #Dockerfile of the image required when creating the environment