cd5cf32a20
- Added steps to pull necessary Docker images for sandbox testing in both CI workflows. - Updated the AI test execution to utilize sandbox configurations, ensuring proper environment setup. - Introduced sandbox initialization in the Assistant's Stream method, allowing for execution of coding agents like Claude and Cursor. - Enhanced context management to support sandbox execution, improving flexibility in handling agent operations.
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package sandbox
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/yaoapp/yao/sandbox/ipc"
|
|
)
|
|
|
|
// Container represents a sandbox container
|
|
type Container struct {
|
|
ID string // Docker container ID
|
|
Name string // Container name: yao-sandbox-{userID}-{chatID}
|
|
UserID string // User identifier
|
|
ChatID string // Chat/session identifier
|
|
Status string // created, running, stopped
|
|
CreatedAt time.Time // Container creation time
|
|
LastUsedAt time.Time // Last activity time
|
|
IPCSession *ipc.Session // Associated IPC session
|
|
}
|
|
|
|
// ExecOptions configures command execution
|
|
type ExecOptions struct {
|
|
WorkDir string // Working directory inside container
|
|
Env map[string]string // Environment variables
|
|
Stdin io.Reader // Standard input
|
|
Timeout time.Duration // Execution timeout (0 = no timeout)
|
|
}
|
|
|
|
// ExecResult contains the result of command execution
|
|
type ExecResult struct {
|
|
ExitCode int // Exit code
|
|
Stdout string // Standard output
|
|
Stderr string // Standard error
|
|
}
|
|
|
|
// FileInfo represents file metadata
|
|
type FileInfo struct {
|
|
Name string // File name
|
|
Path string // Full path
|
|
Size int64 // Size in bytes
|
|
Mode os.FileMode // File mode
|
|
ModTime time.Time // Modification time
|
|
IsDir bool // Is directory
|
|
}
|
|
|
|
// GetName returns the file name (implements context.SandboxFileInfo)
|
|
func (f FileInfo) GetName() string {
|
|
return f.Name
|
|
}
|
|
|
|
// GetSize returns the file size (implements context.SandboxFileInfo)
|
|
func (f FileInfo) GetSize() int64 {
|
|
return f.Size
|
|
}
|
|
|
|
// GetIsDir returns whether this is a directory (implements context.SandboxFileInfo)
|
|
func (f FileInfo) GetIsDir() bool {
|
|
return f.IsDir
|
|
}
|
|
|
|
// ContainerStatus constants
|
|
const (
|
|
StatusCreated = "created"
|
|
StatusRunning = "running"
|
|
StatusStopped = "stopped"
|
|
)
|