Skip to content

Commit 4162ba9

Browse files
committed
fix: runTools without stream should not emit user message events
1 parent f1a4a68 commit 4162ba9

File tree

2 files changed

+6
-14
lines changed

2 files changed

+6
-14
lines changed

src/lib/ChatCompletionRunFunctions.test.ts

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,6 @@ describe('resource completions', () => {
663663
await runner.done();
664664

665665
expect(listener.messages).toEqual([
666-
{ role: 'user', content: 'tell me what the weather is like' },
667666
{
668667
role: 'assistant',
669668
content: null,
@@ -747,7 +746,6 @@ describe('resource completions', () => {
747746
await runner.done().catch(() => {});
748747

749748
expect(listener.messages).toEqual([
750-
{ role: 'user', content: 'tell me what the weather is like' },
751749
{
752750
role: 'assistant',
753751
content: null,
@@ -893,10 +891,6 @@ describe('resource completions', () => {
893891
await runner.done();
894892

895893
expect(listener.messages).toEqual([
896-
{
897-
role: 'user',
898-
content: 'can you tell me how many properties are in {"a": 1, "b": 2, "c": 3}',
899-
},
900894
{
901895
role: 'assistant',
902896
content: null,
@@ -1102,10 +1096,6 @@ describe('resource completions', () => {
11021096
]);
11031097

11041098
expect(listener.messages).toEqual([
1105-
{
1106-
role: 'user',
1107-
content: 'can you tell me how many properties are in {"a": 1, "b": 2, "c": 3}',
1108-
},
11091099
{
11101100
role: 'assistant',
11111101
content: null,
@@ -1199,7 +1189,6 @@ describe('resource completions', () => {
11991189
]);
12001190

12011191
expect(listener.messages).toEqual([
1202-
{ role: 'user', content: 'tell me what the weather is like' },
12031192
{
12041193
role: 'assistant',
12051194
content: null,
@@ -1388,7 +1377,6 @@ describe('resource completions', () => {
13881377
]);
13891378

13901379
expect(listener.messages).toEqual([
1391-
{ role: 'user', content: 'tell me what the weather is like' },
13921380
{
13931381
role: 'assistant',
13941382
content: null,

src/lib/ChatCompletionRunner.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,12 @@ export class ChatCompletionRunner extends AbstractChatCompletionRunner<ChatCompl
5959
return runner;
6060
}
6161

62-
override _addMessage(this: ChatCompletionRunner, message: ChatCompletionMessageParam) {
63-
super._addMessage(message);
62+
override _addMessage(
63+
this: ChatCompletionRunner,
64+
message: ChatCompletionMessageParam,
65+
emit: boolean = true,
66+
) {
67+
super._addMessage(message, emit);
6468
if (isAssistantMessage(message) && message.content) {
6569
this._emit('content', message.content as string);
6670
}

0 commit comments

Comments
 (0)