Method RegisterColab

internal/controller/registry.go:147–183

RegisterColab registers a Colab agent that executes a local Python file on a remote Colab session via the colab CLI.

(cfg config.ColabAgentConfig)

// RegisterColab registers a Colab agent that executes a local Python file
// on a remote Colab session via the colab CLI.
func (r *Registry) RegisterColab(cfg config.ColabAgentConfig) error {
	r.mu.Lock()
	defer r.mu.Unlock()

	if err := validateID(cfg.ID); err != nil {
		return err
	}

	if _, ok := r.agents[cfg.ID]; ok {
		return fmt.Errorf("agent %s already registered", cfg.ID)
	}

	colabAgent, err := expagent.NewColabAgent(expagent.ColabAgentConfig{
		ID:              cfg.ID,
		LocalFile:       cfg.LocalFile,
		DriveFile:       cfg.DriveFile,
		Accelerator:     cfg.Accelerator,
		DriveMountPath:  cfg.DriveMountPath,
		Requirements:    cfg.Requirements,
		InputFlag:       cfg.InputFlag,
		OutputImage:     cfg.OutputImage,
		OutputDrivePath: cfg.OutputDrivePath,
	})
	if err != nil {
		return fmt.Errorf("failed to create colab agent: %w", err)
	}

	r.agents[cfg.ID] = colabAgent
	r.agentInfo[cfg.ID] = &agent.AgentInfo{
		ID:          cfg.ID,
		Name:        cfg.Name,
		Description: cfg.Description,
		Metadata:    cfg.Metadata,
	}

	return nil
}

// Get retrieves an agent by ID.
func (r *Registry) Get(id string) (agent.Agent, error) {