diff --git a/public/scripts/kai-settings.js b/public/scripts/kai-settings.js index 27a204c42..6efadce87 100644 --- a/public/scripts/kai-settings.js +++ b/public/scripts/kai-settings.js @@ -188,7 +188,7 @@ export async function generateKoboldWithStreaming(generate_data, signal) { if (data?.token) { text += data.token; } - yield { text, swipes: [] }; + yield { text, swipes: [], toolCalls: [] }; } }; } diff --git a/public/scripts/nai-settings.js b/public/scripts/nai-settings.js index fc4a042c5..040c6f203 100644 --- a/public/scripts/nai-settings.js +++ b/public/scripts/nai-settings.js @@ -746,7 +746,7 @@ export async function generateNovelWithStreaming(generate_data, signal) { text += data.token; } - yield { text, swipes: [], logprobs: parseNovelAILogprobs(data.logprobs) }; + yield { text, swipes: [], logprobs: parseNovelAILogprobs(data.logprobs), toolCalls: [] }; } }; } diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js index 9403555bd..c021c9682 100644 --- a/public/scripts/textgen-settings.js +++ b/public/scripts/textgen-settings.js @@ -916,6 +916,7 @@ async function generateTextGenWithStreaming(generate_data, signal) { /** @type {import('./logprobs.js').TokenLogprobs | null} */ let logprobs = null; const swipes = []; + const toolCalls = []; while (true) { const { done, value } = await reader.read(); if (done) return; @@ -934,7 +935,7 @@ async function generateTextGenWithStreaming(generate_data, signal) { logprobs = parseTextgenLogprobs(newText, data.choices?.[0]?.logprobs || data?.completion_probabilities); } - yield { text, swipes, logprobs }; + yield { text, swipes, logprobs, toolCalls }; } }; }