| import type { StudioModelPort, StudioModelRequest, StudioModelResponse } from '../../model/studio-model-port' | |
| export class ScriptedStudioModel implements StudioModelPort { | |
| readonly requests: StudioModelRequest[] = [] | |
| private readonly responses: StudioModelResponse[] | |
| constructor(responses: StudioModelResponse[]) { | |
| this.responses = [...responses] | |
| } | |
| async complete(request: StudioModelRequest): Promise<StudioModelResponse> { | |
| this.requests.push(request) | |
| const response = this.responses.shift() | |
| if (!response) { | |
| throw new Error('ScriptedStudioModel ran out of responses') | |
| } | |
| return response | |
| } | |
| } | |