diff options
Diffstat (limited to 'packages/core/src/prompts/prompt-registry.ts')
| -rw-r--r-- | packages/core/src/prompts/prompt-registry.ts | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/packages/core/src/prompts/prompt-registry.ts b/packages/core/src/prompts/prompt-registry.ts index 56699130..a94183ac 100644 --- a/packages/core/src/prompts/prompt-registry.ts +++ b/packages/core/src/prompts/prompt-registry.ts @@ -53,4 +53,22 @@ export class PromptRegistry { } return serverPrompts.sort((a, b) => a.name.localeCompare(b.name)); } + + /** + * Clears all the prompts from the registry. + */ + clear(): void { + this.prompts.clear(); + } + + /** + * Removes all prompts from a specific server. + */ + removePromptsByServer(serverName: string): void { + for (const [name, prompt] of this.prompts.entries()) { + if (prompt.serverName === serverName) { + this.prompts.delete(name); + } + } + } } |
