Hein (Warky)
72b4f7ce3d
- Added file upload handler to process both multipart and raw file uploads. - Implemented parsing logic for upload requests, including handling file metadata. - Introduced SaveFileDecodedInput structure for handling decoded file uploads. - Created unit tests for file upload parsing and validation. feat: add metadata retry configuration and functionality - Introduced MetadataRetryConfig to the application configuration. - Implemented MetadataRetryer to handle retrying metadata extraction for thoughts. - Added new tool for retrying failed metadata extractions. - Updated thought metadata structure to include status and timestamps for metadata processing. fix: enhance metadata normalization and error handling - Updated metadata normalization functions to track status and errors. - Improved handling of metadata extraction failures during thought updates and captures. - Ensured that metadata status is correctly set during various operations. refactor: streamline file saving logic in FilesTool - Refactored Save method in FilesTool to utilize new SaveDecoded method. - Simplified project and thought ID resolution logic during file saving.
316 lines
10 KiB
Go
316 lines
10 KiB
Go
package mcpserver
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
|
|
"git.warky.dev/wdevs/amcs/internal/config"
|
|
"git.warky.dev/wdevs/amcs/internal/tools"
|
|
)
|
|
|
|
type ToolSet struct {
|
|
Capture *tools.CaptureTool
|
|
Search *tools.SearchTool
|
|
List *tools.ListTool
|
|
Stats *tools.StatsTool
|
|
Get *tools.GetTool
|
|
Update *tools.UpdateTool
|
|
Delete *tools.DeleteTool
|
|
Archive *tools.ArchiveTool
|
|
Projects *tools.ProjectsTool
|
|
Context *tools.ContextTool
|
|
Recall *tools.RecallTool
|
|
Summarize *tools.SummarizeTool
|
|
Links *tools.LinksTool
|
|
Files *tools.FilesTool
|
|
Backfill *tools.BackfillTool
|
|
Reparse *tools.ReparseMetadataTool
|
|
RetryMetadata *tools.RetryMetadataTool
|
|
Household *tools.HouseholdTool
|
|
Maintenance *tools.MaintenanceTool
|
|
Calendar *tools.CalendarTool
|
|
Meals *tools.MealsTool
|
|
CRM *tools.CRMTool
|
|
}
|
|
|
|
func New(cfg config.MCPConfig, toolSet ToolSet) http.Handler {
|
|
server := mcp.NewServer(&mcp.Implementation{
|
|
Name: cfg.ServerName,
|
|
Version: cfg.Version,
|
|
}, nil)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "capture_thought",
|
|
Description: "Store a thought with generated embeddings and extracted metadata.",
|
|
}, toolSet.Capture.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_thoughts",
|
|
Description: "Search stored thoughts by semantic similarity.",
|
|
}, toolSet.Search.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "list_thoughts",
|
|
Description: "List recent thoughts with optional metadata filters.",
|
|
}, toolSet.List.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "thought_stats",
|
|
Description: "Get counts and top metadata buckets across stored thoughts.",
|
|
}, toolSet.Stats.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_thought",
|
|
Description: "Retrieve a full thought by id.",
|
|
}, toolSet.Get.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "update_thought",
|
|
Description: "Update thought content or merge metadata.",
|
|
}, toolSet.Update.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "delete_thought",
|
|
Description: "Hard-delete a thought by id.",
|
|
}, toolSet.Delete.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "archive_thought",
|
|
Description: "Archive a thought so it is hidden from default search and listing.",
|
|
}, toolSet.Archive.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "create_project",
|
|
Description: "Create a named project container for thoughts.",
|
|
}, toolSet.Projects.Create)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "list_projects",
|
|
Description: "List projects and their current thought counts.",
|
|
}, toolSet.Projects.List)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "set_active_project",
|
|
Description: "Set the active project for the current MCP session.",
|
|
}, toolSet.Projects.SetActive)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_active_project",
|
|
Description: "Return the active project for the current MCP session.",
|
|
}, toolSet.Projects.GetActive)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_project_context",
|
|
Description: "Get recent and semantic context for a project.",
|
|
}, toolSet.Context.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "recall_context",
|
|
Description: "Recall semantically relevant and recent context.",
|
|
}, toolSet.Recall.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "summarize_thoughts",
|
|
Description: "Summarize a filtered or searched set of thoughts.",
|
|
}, toolSet.Summarize.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "link_thoughts",
|
|
Description: "Create a typed relationship between two thoughts.",
|
|
}, toolSet.Links.Link)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "related_thoughts",
|
|
Description: "Retrieve explicit links and semantic neighbors for a thought.",
|
|
}, toolSet.Links.Related)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "save_file",
|
|
Description: "Store a base64-encoded file such as an image, document, or audio clip, optionally linking it to a thought.",
|
|
}, toolSet.Files.Save)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "load_file",
|
|
Description: "Load a previously stored file by id and return its metadata and base64 content.",
|
|
}, toolSet.Files.Load)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "list_files",
|
|
Description: "List stored files, optionally filtered by thought, project, or kind.",
|
|
}, toolSet.Files.List)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "backfill_embeddings",
|
|
Description: "Generate missing embeddings for stored thoughts using the active embedding model.",
|
|
}, toolSet.Backfill.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "reparse_thought_metadata",
|
|
Description: "Re-extract and normalize metadata for stored thoughts from their content.",
|
|
}, toolSet.Reparse.Handle)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "retry_failed_metadata",
|
|
Description: "Retry metadata extraction for thoughts still marked pending or failed.",
|
|
}, toolSet.RetryMetadata.Handle)
|
|
|
|
// Household Knowledge
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_household_item",
|
|
Description: "Store a household fact (paint color, appliance details, measurement, document, etc.).",
|
|
}, toolSet.Household.AddItem)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_household_items",
|
|
Description: "Search household items by name, category, or location.",
|
|
}, toolSet.Household.SearchItems)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_household_item",
|
|
Description: "Retrieve a household item by id.",
|
|
}, toolSet.Household.GetItem)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_vendor",
|
|
Description: "Add a service provider (plumber, electrician, landscaper, etc.).",
|
|
}, toolSet.Household.AddVendor)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "list_vendors",
|
|
Description: "List household service vendors, optionally filtered by service type.",
|
|
}, toolSet.Household.ListVendors)
|
|
|
|
// Home Maintenance
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_maintenance_task",
|
|
Description: "Create a recurring or one-time home maintenance task.",
|
|
}, toolSet.Maintenance.AddTask)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "log_maintenance",
|
|
Description: "Log completed maintenance work; automatically updates the task's next due date.",
|
|
}, toolSet.Maintenance.LogWork)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_upcoming_maintenance",
|
|
Description: "List maintenance tasks due within the next N days.",
|
|
}, toolSet.Maintenance.GetUpcoming)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_maintenance_history",
|
|
Description: "Search the maintenance log by task name, category, or date range.",
|
|
}, toolSet.Maintenance.SearchHistory)
|
|
|
|
// Family Calendar
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_family_member",
|
|
Description: "Add a family member to the household.",
|
|
}, toolSet.Calendar.AddMember)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "list_family_members",
|
|
Description: "List all family members.",
|
|
}, toolSet.Calendar.ListMembers)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_activity",
|
|
Description: "Schedule a one-time or recurring family activity.",
|
|
}, toolSet.Calendar.AddActivity)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_week_schedule",
|
|
Description: "Get all activities scheduled for a given week.",
|
|
}, toolSet.Calendar.GetWeekSchedule)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_activities",
|
|
Description: "Search activities by title, type, or family member.",
|
|
}, toolSet.Calendar.SearchActivities)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_important_date",
|
|
Description: "Track a birthday, anniversary, deadline, or other important date.",
|
|
}, toolSet.Calendar.AddImportantDate)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_upcoming_dates",
|
|
Description: "Get important dates coming up in the next N days.",
|
|
}, toolSet.Calendar.GetUpcomingDates)
|
|
|
|
// Meal Planning
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_recipe",
|
|
Description: "Save a recipe with ingredients and instructions.",
|
|
}, toolSet.Meals.AddRecipe)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_recipes",
|
|
Description: "Search recipes by name, cuisine, tags, or ingredient.",
|
|
}, toolSet.Meals.SearchRecipes)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "update_recipe",
|
|
Description: "Update an existing recipe.",
|
|
}, toolSet.Meals.UpdateRecipe)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "create_meal_plan",
|
|
Description: "Set the meal plan for a week; replaces any existing plan for that week.",
|
|
}, toolSet.Meals.CreateMealPlan)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_meal_plan",
|
|
Description: "Get the meal plan for a given week.",
|
|
}, toolSet.Meals.GetMealPlan)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "generate_shopping_list",
|
|
Description: "Auto-generate a shopping list from the meal plan for a given week.",
|
|
}, toolSet.Meals.GenerateShoppingList)
|
|
|
|
// Professional CRM
|
|
addTool(server, &mcp.Tool{
|
|
Name: "add_professional_contact",
|
|
Description: "Add a professional contact to the CRM.",
|
|
}, toolSet.CRM.AddContact)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "search_contacts",
|
|
Description: "Search professional contacts by name, company, title, notes, or tags.",
|
|
}, toolSet.CRM.SearchContacts)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "log_interaction",
|
|
Description: "Log an interaction with a professional contact.",
|
|
}, toolSet.CRM.LogInteraction)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_contact_history",
|
|
Description: "Get full history (interactions and opportunities) for a contact.",
|
|
}, toolSet.CRM.GetHistory)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "create_opportunity",
|
|
Description: "Create a deal, project, or opportunity linked to a contact.",
|
|
}, toolSet.CRM.CreateOpportunity)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "get_follow_ups_due",
|
|
Description: "List contacts with a follow-up date due within the next N days.",
|
|
}, toolSet.CRM.GetFollowUpsDue)
|
|
|
|
addTool(server, &mcp.Tool{
|
|
Name: "link_thought_to_contact",
|
|
Description: "Append a stored thought to a contact's notes.",
|
|
}, toolSet.CRM.LinkThought)
|
|
|
|
return mcp.NewStreamableHTTPHandler(func(*http.Request) *mcp.Server {
|
|
return server
|
|
}, &mcp.StreamableHTTPOptions{
|
|
JSONResponse: true,
|
|
SessionTimeout: 10 * time.Minute,
|
|
})
|
|
}
|