Skip to content

Replace Prompts/Resources/Resource Templates #518

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Jul 30, 2025
Merged

Replace Prompts/Resources/Resource Templates #518

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
c40062a
Replace all Prompts
dgageot Jul 22, 2025
ebba84e
Replace all Resources
dgageot Jul 22, 2025
6780f8e
Replace all Resource Templates
dgageot Jul 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 25 additions & 0 deletions server/server.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -348,6 +348,14 @@ func (s *MCPServer) AddResources(resources ...ServerResource) {
}
}

// SetResources replaces all existing resources with the provided list
func (s *MCPServer) SetResources(resources ...ServerResource) {
s.resourcesMu.Lock()
s.resources = make(map[string]resourceEntry, len(resources))
s.resourcesMu.Unlock()
s.AddResources(resources...)
}

// AddResource registers a new resource and its handler
func (s *MCPServer) AddResource(
resource mcp.Resource,
Expand Down Expand Up @@ -391,6 +399,14 @@ func (s *MCPServer) AddResourceTemplates(resourceTemplates ...ServerResourceTemp
}
}

// SetResourceTemplates replaces all existing resource templates with the provided list
func (s *MCPServer) SetResourceTemplates(templates ...ServerResourceTemplate) {
s.resourcesMu.Lock()
s.resourceTemplates = make(map[string]resourceTemplateEntry, len(templates))
s.resourcesMu.Unlock()
s.AddResourceTemplates(templates...)
}

// AddResourceTemplate registers a new resource template and its handler
func (s *MCPServer) AddResourceTemplate(
template mcp.ResourceTemplate,
Expand Down Expand Up @@ -422,6 +438,15 @@ func (s *MCPServer) AddPrompt(prompt mcp.Prompt, handler PromptHandlerFunc) {
s.AddPrompts(ServerPrompt{Prompt: prompt, Handler: handler})
}

// SetPrompts replaces all existing prompts with the provided list
func (s *MCPServer) SetPrompts(prompts ...ServerPrompt) {
s.promptsMu.Lock()
s.prompts = make(map[string]mcp.Prompt, len(prompts))
s.promptHandlers = make(map[string]PromptHandlerFunc, len(prompts))
s.promptsMu.Unlock()
s.AddPrompts(prompts...)
}

// DeletePrompts removes prompts from the server
func (s *MCPServer) DeletePrompts(names ...string) {
s.promptsMu.Lock()
Expand Down