Talking to AI from your terminal: WezTerm ✕ llm

brew install --HEAD wez/wezterm/wezterm        # or pkg, deb, scoop…
brew install jq                                # helper needs jq
uv tool install -U llm --with llm-anthropic --with llm-gemini
llm auth openai                                # or: llm auth anthropic

#!/usr/bin/env bash
set -euo pipefail

[[ -f $HOME/.bashrc ]] && source "$HOME/.bashrc"   # optional: pull in API keys

new=0
if [[ ${1:-} == --new ]]; then new=1; shift; fi

pane="$1"; tmpfile="$2"; shift 2
question="$*"
context=$(<"$tmpfile")

cache_dir="${XDG_CACHE_HOME:-$HOME/.cache}/wezterm-llm"
cid_file="$cache_dir/$pane"
mkdir -p "$cache_dir"
[[ $new -eq 1 ]] && rm -f "$cid_file"

if [[ -f $cid_file ]]; then
  cid=$(<"$cid_file")
  reply=$(printf '%s
' "$context"           | llm --cid "$cid" --no-stream "$question")
else
  reply=$(printf '%s
' "$context"           | llm --no-stream "$question")
  cid=$(llm logs --limit 1 --json | jq -r '.[0].conversation_id')
  echo "$cid" > "$cid_file"
fi

printf '%s' "$reply"

chmod +x ~/.local/bin/llm-from-wez

local wezterm = require 'wezterm'

local function call_helper(pane_id, text, question, new)
  local tmp = os.tmpname()
  local f   = io.open(tmp, 'w'); f:write(text); f:close()

  local args = { os.getenv('HOME') .. '/.local/bin/llm-from-wez' }
  if new then table.insert(args, '--new') end
  table.insert(args, tostring(pane_id))
  table.insert(args, tmp)
  table.insert(args, question)

  return wezterm.run_child_process(args)  -- { ok, stdout, stderr }
end

local M = {}

function M.ask(win, pane, n_lines, new)
  local buf = n_lines and pane:get_lines_as_text(n_lines)
                    or pane:get_lines_as_text()

  win:perform_action(
    wezterm.action.PromptInputLine{
      description = 'Ask LLM:',
      action = wezterm.action_callback(function(w, p, q)
        if not q or #q == 0 then return end
        local ok, out, err = call_helper(p:pane_id(), buf, q, new)
        if not ok then
          w:toast_notification('LLM error', err, nil, 6000); return
        end

        local tmp_out = os.tmpname()
        local fo = io.open(tmp_out, 'w'); fo:write(out); fo:close()

        w:perform_action(
          wezterm.action.SplitPane{
            direction = 'Right',
            size      = { Percent = 45 },
            command   = { args = { 'bash', '-lc', 'less -R ' .. tmp_out } },
          },
          p
        )
      end)
    },
    pane
  )
end

function M.reset(pane_id)
  wezterm.run_child_process{
    os.getenv('HOME') .. '/.local/bin/llm-from-wez',
    '--new', tostring(pane_id), os.tmpname(), ''
  }
end

return M

local wezterm = require 'wezterm'
local act     = wezterm.action
local llm     = require 'wez_llm'

local config = {}

config.keys = {
  { key = 'l', mods = 'CTRL|ALT', action = act.EmitEvent 'LLM_VIEWPORT' },
  { key = 'L', mods = 'CTRL|ALT', action = act.EmitEvent 'LLM_SCROLLBACK' },
  { key = 'N', mods = 'CTRL|ALT', action = act.EmitEvent 'LLM_RESET' },
}

wezterm.on('LLM_VIEWPORT', function(win, pane)
  llm.ask(win, pane, nil, false)
end)

wezterm.on('LLM_SCROLLBACK', function(win, pane)
  win:perform_action(
    act.PromptInputLine{
      description = 'Lines of scroll-back to send:',
      action = wezterm.action_callback(function(w, p, line)
        local n = tonumber(line or '') or 0
        if n > 0 then llm.ask(w, p, n, false) end
      end)
    }, pane)
end)

wezterm.on('LLM_RESET', function(win, pane)
  llm.reset(pane:pane_id())
  win:toast_notification('LLM', 'New conversation started', nil, 3000)
end)

return config