20 lines
597 B
TypeScript
20 lines
597 B
TypeScript
import { Scenario } from '../types.js';
|
|
import { InitializeScenario } from './initialize.js';
|
|
import { ToolsCallScenario } from './tools_call.js';
|
|
|
|
export const scenarios = new Map<string, Scenario>([
|
|
['initialize', new InitializeScenario()],
|
|
['tools-call', new ToolsCallScenario()]
|
|
]);
|
|
|
|
export function registerScenario(name: string, scenario: Scenario): void {
|
|
scenarios.set(name, scenario);
|
|
}
|
|
|
|
export function getScenario(name: string): Scenario | undefined {
|
|
return scenarios.get(name);
|
|
}
|
|
|
|
export function listScenarios(): string[] {
|
|
return Array.from(scenarios.keys());
|
|
}
|