proximos-passos

1import { atom, read, update } from 'claude-code'
2import type { Register } from 'claude-code'
3
4import type { Opcao } from '../types'
5
6// Depois de cada resposta, o Haiku propõe 3 próximos pedidos (botões acima da caixa).
7// Medido em 02/10: o Haiku lê ~1 mil tokens; o modelo principal releria a conversa inteira (~850 mil).
8const opcoes = atom({ plugin: 'proximos-passos', key: 'opcoes' } as const, [])
9const gerando = atom({ plugin: 'proximos-passos', key: 'gerando' } as const, false)
10const nativa = atom({ plugin: 'proximos-passos', key: 'nativa' } as const, '')
11
12const SISTEMA = `Você sugere os próximos passos de uma conversa de trabalho entre um usuário e o assistente Claude.
13O usuário escreve curto e direto e decide o que fazer; o assistente executa. Escreva os pedidos no idioma em que o usuário escreve.
14Leia o último pedido do usuário e a resposta do assistente e proponha exatamente 3 próximos pedidos que o USUÁRIO mandaria agora.
15- Se a resposta terminou com uma pergunta ou com opções, as sugestões são as respostas plausíveis a ela (a recomendada primeiro).
16- Senão, os passos mais úteis a partir dali: continuar, conferir/testar, ou mudar de rumo.
17- Cada pedido escrito como o usuário escreveria, na primeira pessoa dele, imperativo curto ("sim, pode aplicar").
18- Nada genérico ("continue", "obrigado"). Nada que repita o que já foi feito.
19Responda APENAS um JSON válido, sem cercas de código: [{"rotulo": "<até 6 palavras>", "pedido": "<o texto que será enviado>"}, ...]`
20
21let ultimoPedido = ''
22
23// Tokens exatamente como a resposta da Anthropic informa (campo usage).
24async function medir($: any, modo: 'turno' | 'haiku', ms: number, u: any) {
25  const m = {
26    modo, ms,
27    novos: Number(u?.input_tokens ?? 0),
28    cache: Number(u?.cache_read_input_tokens ?? 0),
29    escrita: Number(u?.cache_creation_input_tokens ?? 0),
30    saida: Number(u?.output_tokens ?? 0),
31  }
32  const lista = ((await $.store.get('medicoes2')) as any[] | undefined) ?? []
33  await $.store.set('medicoes2', [...lista, m].slice(-600))
34}
35
36function ler(texto: string): Opcao[] {
37  const j = JSON.parse(String(texto).replace(/^[\s\S]*?(\[)/, '$1').replace(/\][^\]]*$/, ']'))
38  return Array.isArray(j)
39    ? j.filter((o: any) => o?.rotulo && o?.pedido).slice(0, 3)
40      .map((o: any) => ({ rotulo: String(o.rotulo).slice(0, 60), pedido: String(o.pedido) }))
41    : []
42}
43
44async function sugerir($: any, resposta: string) {
45  await update($, gerando, () => true)
46  let lista: Opcao[] = []
47  try {
48    const t0 = await $.clock.now()
49    const r: any = await $.model.complete({
50      model: 'haiku', system: SISTEMA, maxTokens: 300, timeoutMs: 15_000,
51      prompt: `Último pedido do usuário:\n${ultimoPedido.slice(0, 1000)}\n\nFim da resposta do assistente:\n${resposta.slice(-3000)}`,
52    })
53    await medir($, 'haiku', (await $.clock.now()) - t0, r.usage).catch(() => undefined)
54    if (r.isAnswered) lista = ler(r.text)
55  } catch {
56    /* erro, limite de tempo ou resposta fora do formato: sem sugestão desta vez */
57  } finally {
58    // Sempre desliga o "pensando", senão a faixa fica presa.
59    await update($, opcoes, () => lista)
60    await update($, gerando, () => false)
61  }
62}
63
64const fmt = (n: number) => Math.round(n).toLocaleString('pt-BR')
65
66export const register: Register = on => {
67  on('session.start', async ($, e, next) => {
68    await $.command.register({ name: 'custos', description: 'Tokens das sugestões comparados com uma resposta normal' })
69    // Um recarregamento no meio de uma sugestão deixaria o "pensando" ligado para sempre.
70    await update($, gerando, () => false)
71    return next(e)
72  })
73
74  on('command.run', { command: 'custos' }, async $ => {
75    const todas = (((await $.store.get('medicoes2')) as any[] | undefined) ?? []).filter(m => m.modo !== 'claude')
76    if (!todas.length) return { text: 'Ainda não há medições.' }
77    const nomes: Record<string, string> = { turno: 'Resposta normal do Claude (referência)', haiku: 'Sugestões pelo Haiku' }
78    const linhas: string[] = ['Médias por chamada, em tokens, como a Anthropic informa:']
79    for (const modo of ['turno', 'haiku']) {
80      const l = todas.filter(m => m.modo === modo)
81      if (!l.length) continue
82      const media = (k: string) => l.reduce((s, m) => s + Number(m[k] ?? 0), 0) / l.length
83      const ms = l.map(m => m.ms).filter(Boolean).sort((a, b) => a - b)
84      linhas.push(
85        `• ${nomes[modo]} (${l.length}): ${fmt(media('cache'))} lidos do cache · ${fmt(media('novos'))} novos · ${fmt(media('escrita'))} gravados no cache · ${fmt(media('saida'))} de saída` +
86        (ms.length ? ` · ${(ms[Math.floor(ms.length / 2)] / 1000).toFixed(1)} s` : ''),
87      )
88    }
89    return { text: linhas.join('\n') }
90  })
91
92  // Só para não repetir nos botões o que já está no Tab.
93  on('prompt.suggest', async ($, e: any, next) => {
94    if (e.origin?.kind === 'suggestion' && e.text) await update($, nativa, () => String(e.text))
95    return next(e)
96  })
97
98  on('prompt.submit', async ($, e, next) => {
99    ultimoPedido = e.text
100    await update($, opcoes, () => [])
101    await update($, nativa, () => '')
102    return next(e)
103  })
104
105  on('turn.complete', async ($, e, next) => {
106    const r = await next(e)
107    if (e.agentId || e.reason !== 'answer') return r
108    if (e.usage) await medir($, 'turno', e.durationMs, e.usage).catch(() => undefined)
109    // Mensagens automáticas de outros mods e respostas curtas não pedem sugestão.
110    // Mensagem automática de outro mod começa com "[nome-do-mod]".
111    if (/^\s*\[[\w-]+\]/.test(ultimoPedido) || e.answer.split(/\s+/).length < 15) return r
112    void sugerir($, e.answer)
113    return r
114  })
115
116  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
117    const n = await read($, nativa)
118    const mesmo = (a: string, b: string) => a.trim().toLowerCase() === b.trim().toLowerCase()
119    const botoes = (await read($, opcoes)).filter(o => !n || !mesmo(o.pedido, n))
120    const ocupado = await read($, gerando)
121    if (e.props.hasSurvey || e.props.isWorking || (!botoes.length && !ocupado)) return next(e)
122    const { Box, Button, Text } = $.ui.resolve(e)
123    const enviar = (o: Opcao) => async () => {
124      await update($, opcoes, () => [])
125      await $.prompt.submit({ text: o.pedido, asUser: true })
126    }
127    return (
128      <Box flexDirection="row" flexWrap="wrap" columnGap={2}>
129        {ocupado && !botoes.length && <Text dimColor>pensando nos próximos passos…</Text>}
130        {botoes.map((o, i) => (
131          <Button key={`op${i}`} label={o.rotulo} hotkey={String(i + 1)} {...(i === 0 ? { variant: 'primary' as const } : {})} onPress={enviar(o)} />
132        ))}
133        {botoes.length > 0 && <Button key="fechar" plain label="✕" role="dismiss" onPress={() => update($, opcoes, () => [])} />}
134      </Box>
135    )
136  })
137}
138

1export type Opcao = { rotulo: string; pedido: string }
2
3declare module 'claude-code' {
4  interface PluginState {
5    'proximos-passos': { opcoes: Opcao[]; gerando: boolean; nativa: string }
6  }
7}
8