Skip to content

Commit

Permalink
fix testmodel class
Browse files Browse the repository at this point in the history
  • Loading branch information
lilyydu committed Sep 18, 2024
1 parent fc16d98 commit 14ccd91
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions js/packages/teams-ai/src/models/TestModel.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ import { Tokenizer } from '../tokenizers';
import { TurnContext } from 'botbuilder';
import { Memory } from '../MemoryFork';
import EventEmitter from 'events';
import { StreamingResponse } from '../StreamingResponse';

/**
* A `PromptCompletionModel` used for testing.
Expand Down Expand Up @@ -90,7 +91,8 @@ export class TestModel implements PromptCompletionModel {
return new TestModel(async (model, context, memory, functions, tokenizer, template) => {
model.events.emit('beforeCompletion', context, memory, functions, tokenizer, template, false);
await new Promise((resolve) => setTimeout(resolve, delay));
model.events.emit('responseReceived', context, memory, response);
const streamer = new StreamingResponse(context);
model.events.emit('responseReceived', context, memory, response, streamer);
return response;
});
}
Expand Down Expand Up @@ -127,7 +129,8 @@ export class TestModel implements PromptCompletionModel {
// Finalize the response.
await new Promise((resolve) => setTimeout(resolve, delay));
const response: PromptResponse<string> = { status: 'success', message: { role: 'assistant', content } };
model.events.emit('responseReceived', context, memory, response);
const streamer = new StreamingResponse(context);
model.events.emit('responseReceived', context, memory, response, streamer);
return response;
});
}
Expand Down

0 comments on commit 14ccd91

Please sign in to comment.