summaryrefslogtreecommitdiff
path: root/packages/core/src/prompts/prompt-registry.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src/prompts/prompt-registry.ts')
-rw-r--r--packages/core/src/prompts/prompt-registry.ts18
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);
+ }
+ }
+ }
}