⬆️ Upgrade AI SDK (#1641)
This commit is contained in:
28
packages/ai/appendToolResultsToMessages.ts
Normal file
28
packages/ai/appendToolResultsToMessages.ts
Normal file
@@ -0,0 +1,28 @@
|
||||
import { CoreMessage, ToolCallPart, ToolResultPart } from 'ai'
|
||||
|
||||
type Props = {
|
||||
messages: CoreMessage[]
|
||||
toolCalls: ToolCallPart[]
|
||||
toolResults: ToolResultPart[]
|
||||
}
|
||||
export const appendToolResultsToMessages = ({
|
||||
messages,
|
||||
toolCalls,
|
||||
toolResults,
|
||||
}: Props): CoreMessage[] => {
|
||||
if (toolCalls.length > 0) {
|
||||
messages.push({
|
||||
role: 'assistant',
|
||||
content: toolCalls,
|
||||
})
|
||||
}
|
||||
|
||||
if (toolResults.length > 0) {
|
||||
messages.push({
|
||||
role: 'tool',
|
||||
content: toolResults,
|
||||
})
|
||||
}
|
||||
|
||||
return messages
|
||||
}
|
||||
Reference in New Issue
Block a user