diff options
Diffstat (limited to 'packages/session-orchestrator/src')
| -rw-r--r-- | packages/session-orchestrator/src/extension.ts | 30 |
1 files changed, 3 insertions, 27 deletions
diff --git a/packages/session-orchestrator/src/extension.ts b/packages/session-orchestrator/src/extension.ts index c75bf29..8cd4c44 100644 --- a/packages/session-orchestrator/src/extension.ts +++ b/packages/session-orchestrator/src/extension.ts @@ -1,11 +1,5 @@ import { conversationStoreHandle } from "@dispatch/conversation-store"; -import type { - Extension, - HostAPI, - Manifest, - ProviderContract, - ToolContract, -} from "@dispatch/kernel"; +import type { Extension, HostAPI, Manifest } from "@dispatch/kernel"; import { runTurn } from "@dispatch/kernel"; import { createSessionOrchestrator, @@ -27,31 +21,13 @@ export const manifest: Manifest = { }, }; -interface ProviderResolvingHostAPI extends HostAPI { - readonly getProviders?: () => ReadonlyMap<string, ProviderContract>; - readonly getTools?: () => ReadonlyMap<string, ToolContract>; -} - export function activate(host: HostAPI): void { const conversationStore = host.getService(conversationStoreHandle); - const extendedHost = host as ProviderResolvingHostAPI; const orchestrator: SessionOrchestrator = createSessionOrchestrator({ conversationStore, - resolveProvider: () => { - if (extendedHost.getProviders !== undefined) { - return selectFirstProvider(extendedHost.getProviders()); - } - throw new Error( - "HostAPI does not expose getProviders() — change-request: add provider resolution to HostAPI", - ); - }, - resolveTools: () => { - if (extendedHost.getTools !== undefined) { - return [...extendedHost.getTools().values()]; - } - return []; - }, + resolveProvider: () => selectFirstProvider(host.getProviders()), + resolveTools: () => [...host.getTools().values()], runTurn, }); |
