xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-12 04:03:42 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat:add amazon.nova model tool use support.

Browse Source
This commit is contained in:
glay
2024-12-14 11:01:10 +08:00
parent 0ec1ae6276
commit e839940a26
3 changed files with 57 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
app/utils/aws.ts
View File

@@ -435,25 +435,31 @@ export function processMessage(
if (!data) return { remainText, index };
try {
// Handle Nova's tool calls
// Handle Nova's tool calls with exact schema match
// console.log("processMessage data=========================",data);
if (
data.stopReason === "tool_use" &&
data.output?.message?.content?.[0]?.toolUse
) {
const toolUse = data.output.message.content[0].toolUse;
if (data.contentBlockStart?.start?.toolUse) {
const toolUse = data.contentBlockStart.start.toolUse;
index += 1;
runTools.push({
id: `tool-${Date.now()}`,
id: toolUse.toolUseId,
type: "function",
function: {
name: toolUse.name,
arguments: JSON.stringify(toolUse.input),
name: toolUse.name || "", // Ensure name is always present
arguments: "{}", // Initialize empty arguments
},
});
return { remainText, index };
}
// Handle Nova's tool input in contentBlockDelta
if (data.contentBlockDelta?.delta?.toolUse?.input) {
if (runTools[index]) {
runTools[index].function.arguments =
data.contentBlockDelta.delta.toolUse.input;
}
return { remainText, index };
}
// Handle Nova's text content
if (data.output?.message?.content?.[0]?.text) {
remainText += data.output.message.content[0].text;
@@ -465,11 +471,9 @@ export function processMessage(
return { remainText, index };
}
// Handle Nova's contentBlockDelta event
if (data.contentBlockDelta) {
if (data.contentBlockDelta.delta?.text) {
remainText += data.contentBlockDelta.delta.text;
}
// Handle Nova's text delta
if (data.contentBlockDelta?.delta?.text) {
remainText += data.contentBlockDelta.delta.text;
return { remainText, index };
}
@@ -496,7 +500,7 @@ export function processMessage(
id: data.content_block.id,
type: "function",
function: {
name: data.content_block.name,
name: data.content_block.name || "", // Ensure name is always present
arguments: "",
},
});
@@ -523,7 +527,7 @@ export function processMessage(
id: toolCall.id || `tool-${Date.now()}`,
type: "function",
function: {
name: toolCall.function?.name,
name: toolCall.function?.name || "", // Ensure name is always present
arguments: toolCall.function?.arguments || "",
},
});
@@ -554,7 +558,7 @@ export function processMessage(
remainText += newText;
}
} catch (e) {
console.warn("Failed to process Bedrock message");
console.warn("Failed to process Bedrock message:");
}
return { remainText, index };