mirror of
https://github.com/mark3labs/kit.git
synced 2026-06-14 03:30:26 +00:00
Ed Zynda
23c16bb197
Phase 2+3 extension API additions:
- Tool management: GetAllTools, SetActiveTools (plan-mode support)
- Model management: SetModel, GetAvailableModels, ModelChangedEvent
- Extension options: RegisterOption, GetOption, SetOption (env/config/default)
- Inter-extension event bus: OnCustomEvent, EmitCustomEvent
- Direct LLM completion: ctx.Complete with streaming/blocking modes
- Steer delivery mode: CancelAndSend for interrupt-and-redirect
New example extensions (10):
- plan-mode.go: read-only exploration with /plan toggle
- summarize.go: conversation summarization via ctx.Complete
- bookmark.go: persistent bookmarks via AppendEntry/GetEntries
- auto-commit.go: auto-commit on exit using last assistant message
- permission-gate.go: confirm dangerous bash commands
- protected-paths.go: block writes to .env, .git/, secrets/
- notify.go: desktop notifications on agent completion
- inline-bash.go: !{cmd} expansion in prompts
- pirate.go: system prompt persona injection
- project-rules.go: load .kit/rules/*.md into system prompt
Always-wrap tools through runner for SetActiveTools disabled-tool checking.
Removed phase1/phase2 test extensions from examples.
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
//go:build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"kit/ext"
|
|
)
|
|
|
|
// Init adds a /summarize command that generates a concise summary of the
|
|
// current conversation using a direct LLM completion. Demonstrates the
|
|
// ctx.Complete API (Gap 17). Inspired by Pi's summarize.ts.
|
|
//
|
|
// The summary is displayed in a styled block and can optionally be saved
|
|
// to the session via AppendEntry for later retrieval.
|
|
//
|
|
// Usage: kit -e examples/extensions/summarize.go
|
|
func Init(api ext.API) {
|
|
api.RegisterCommand(ext.CommandDef{
|
|
Name: "summarize",
|
|
Description: "Summarize the current conversation",
|
|
Execute: func(args string, ctx ext.Context) (string, error) {
|
|
msgs := ctx.GetMessages()
|
|
if len(msgs) == 0 {
|
|
ctx.PrintInfo("Nothing to summarize — no messages yet.")
|
|
return "", nil
|
|
}
|
|
|
|
// Build a text representation of the conversation.
|
|
var parts []string
|
|
for _, m := range msgs {
|
|
content := m.Content
|
|
if len(content) > 2000 {
|
|
content = content[:1997] + "..."
|
|
}
|
|
parts = append(parts, fmt.Sprintf("[%s]: %s", m.Role, content))
|
|
}
|
|
conversation := strings.Join(parts, "\n\n")
|
|
|
|
ctx.PrintInfo("Generating summary...")
|
|
|
|
resp, err := ctx.Complete(ext.CompleteRequest{
|
|
System: `You are a concise summarization assistant. Summarize the conversation below in 3-5 bullet points. Focus on:
|
|
- What was discussed or requested
|
|
- Key decisions or outcomes
|
|
- Any pending action items
|
|
|
|
Be concise. Use plain text, no markdown headers.`,
|
|
Prompt: conversation,
|
|
})
|
|
if err != nil {
|
|
ctx.PrintError("Summary failed: " + err.Error())
|
|
return "", nil
|
|
}
|
|
|
|
summary := strings.TrimSpace(resp.Text)
|
|
|
|
ctx.PrintBlock(ext.PrintBlockOpts{
|
|
Text: summary,
|
|
BorderColor: "#89b4fa",
|
|
Subtitle: fmt.Sprintf("Summary (%d messages, %d tokens used)", len(msgs), resp.InputTokens+resp.OutputTokens),
|
|
})
|
|
|
|
// Persist the summary in the session for later retrieval.
|
|
ctx.AppendEntry("summary", summary)
|
|
|
|
return "", nil
|
|
},
|
|
})
|
|
|
|
// /summaries — list all saved summaries.
|
|
api.RegisterCommand(ext.CommandDef{
|
|
Name: "summaries",
|
|
Description: "List saved conversation summaries",
|
|
Execute: func(args string, ctx ext.Context) (string, error) {
|
|
entries := ctx.GetEntries("summary")
|
|
if len(entries) == 0 {
|
|
ctx.PrintInfo("No summaries saved yet. Use /summarize to create one.")
|
|
return "", nil
|
|
}
|
|
for i, e := range entries {
|
|
ctx.PrintBlock(ext.PrintBlockOpts{
|
|
Text: e.Data,
|
|
BorderColor: "#89b4fa",
|
|
Subtitle: fmt.Sprintf("Summary #%d (%s)", i+1, e.Timestamp[:19]),
|
|
})
|
|
}
|
|
return "", nil
|
|
},
|
|
})
|
|
}
|