diff --git a/cfg/global.yaml b/cfg/global.yaml index 8f76dd8..0dda41b 100644 --- a/cfg/global.yaml +++ b/cfg/global.yaml @@ -68,6 +68,7 @@ env: - DISPLAY - WAYLAND_DISPLAY - AGENT_ID + - ANVILLM_BACKEND advanced: ldd: false add_exec: true diff --git a/internal/backend/tmux/tmux.go b/internal/backend/tmux/tmux.go index f7f63c3..ba5b116 100644 --- a/internal/backend/tmux/tmux.go +++ b/internal/backend/tmux/tmux.go @@ -258,10 +258,10 @@ func (b *Backend) CreateSession(ctx context.Context, opts backend.SessionOptions return nil, fmt.Errorf("failed to change directory: %w", err) } - // Export AGENT_ID to shell environment - if err := sendKeys(target, fmt.Sprintf("export AGENT_ID=%s", id), "C-m"); err != nil { + // Export AGENT_ID and ANVILLM_BACKEND to shell environment + if err := sendKeys(target, fmt.Sprintf("export AGENT_ID=%s ANVILLM_BACKEND=%s", id, b.cfg.Name), "C-m"); err != nil { killWindow(b.tmuxSession, windowName) - return nil, fmt.Errorf("failed to export AGENT_ID: %w", err) + return nil, fmt.Errorf("failed to export AGENT_ID/ANVILLM_BACKEND: %w", err) } // Send the command diff --git a/internal/backends/ollama.go b/internal/backends/ollie.go similarity index 100% rename from internal/backends/ollama.go rename to internal/backends/ollie.go