diff options
Diffstat (limited to 'packages/server/src/utils/nextSpeakerChecker.ts')
| -rw-r--r-- | packages/server/src/utils/nextSpeakerChecker.ts | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/packages/server/src/utils/nextSpeakerChecker.ts b/packages/server/src/utils/nextSpeakerChecker.ts index fb00b39c..66fa4395 100644 --- a/packages/server/src/utils/nextSpeakerChecker.ts +++ b/packages/server/src/utils/nextSpeakerChecker.ts @@ -61,6 +61,7 @@ export interface NextSpeakerResponse { export async function checkNextSpeaker( chat: GeminiChat, geminiClient: GeminiClient, + abortSignal: AbortSignal, ): Promise<NextSpeakerResponse | null> { // We need to capture the curated history because there are many moments when the model will return invalid turns // that when passed back up to the endpoint will break subsequent calls. An example of this is when the model decides @@ -129,6 +130,7 @@ export async function checkNextSpeaker( const parsedResponse = (await geminiClient.generateJson( contents, RESPONSE_SCHEMA, + abortSignal, )) as unknown as NextSpeakerResponse; if ( |