prepared last tool

This commit is contained in:
didier committed 2026-06-16 13:39:41 -04:00
1 parent 8fa9d43530
commit a65d39cea0
2 files changed
+116

No files matched your search

+60
View File
@@ -58,6 +58,7 @@ func main() {
registerStoreFlashcard(mcpServer, store, searchEngine, logger)
registerSearchFlashcard(mcpServer, searchEngine)
registerListRecentFlashcard(mcpServer, store, logger)
registerUpdateFlashcard(mcpServer, store, searchEngine, logger)
registerDeleteFlashcard(mcpServer, store, searchEngine, logger)
registerVoteFlashcard(mcpServer, store, logger)
@@ -245,6 +246,65 @@ func registerSearchFlashcard(mcpServer *server.MCPServer, engine SearchEngine) {
)
}
func registerListRecentFlashcard(mcpServer *server.MCPServer, sqlite *sqliteStore, logger *zap.Logger) {
mcpServer.AddTool(
mcp.NewTool("list_recent_flashcards",
mcp.WithDescription("List recently created flashcards, optionally filtered by tag, repo, or host. Use this at the start of a session to recall what was learned recently."),
mcp.WithNumber("limit",
mcp.Description("Maximum number of results to return (default: 20)"),
),
mcp.WithString("tag",
mcp.Description("Filter by tag (e.g. 'preference', 'convention')"),
),
mcp.WithString("repo",
mcp.Description("Filter by repository name (e.g. 'st-pierre', 'loopagent')"),
),
mcp.WithString("host",
mcp.Description("Filter by hostname (e.g. 'shulgi')"),
),
),
handleListRecentFlashcard(sqlite, logger),
)
}
func handleListRecentFlashcard(sqlite *sqliteStore, logger *zap.Logger) server.ToolHandlerFunc {
return func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
args, ok := request.Params.Arguments.(map[string]any)
if !ok {
return mcp.NewToolResultError("invalid arguments"), nil
}
tag := parseOptionalString(args["tag"])
repo := parseOptionalString(args["repo"])
host := parseOptionalString(args["host"])
limit := uint64(20)
if rawLimit := args["limit"]; rawLimit != nil {
switch v := rawLimit.(type) {
case float64:
limit = uint64(v)
case int:
limit = uint64(v)
case int64:
limit = uint64(v)
}
}
entries, err := sqlite.listRecent(ctx, tag, repo, host, limit)
if err != nil {
logger.Error("list recent failed", zap.Error(err))
return mcp.NewToolResultError(fmt.Sprintf("list recent failed: %v", err)), nil
}
output, err := json.MarshalIndent(entries, "", " ")
if err != nil {
return mcp.NewToolResultError(fmt.Sprintf("failed to marshal results: %v", err)), nil
}
return mcp.NewToolResultText(string(output)), nil
}
}
func registerUpdateFlashcard(mcpServer *server.MCPServer, sqlite *sqliteStore, engine SearchEngine, logger *zap.Logger) {
mcpServer.AddTool(
mcp.NewTool("update_flashcard",