mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-19 04:14:46 +08:00
feat: support basic file presenting
This commit is contained in:
@@ -46,9 +46,14 @@ export function groupMessages<T>(
|
||||
messageIndex === messages.length - 1 &&
|
||||
isLoading)
|
||||
) {
|
||||
// Assistant messages without any content are folded into the previous group
|
||||
// Normally, these are tool calls (with or without thinking)
|
||||
currentGroup.push(message);
|
||||
if (message.tool_calls?.[0]?.name === "present_files") {
|
||||
yieldCurrentGroup();
|
||||
currentGroup.push(message);
|
||||
} else {
|
||||
// Assistant messages without any content are folded into the previous group
|
||||
// Normally, these are tool calls (with or without thinking)
|
||||
currentGroup.push(message);
|
||||
}
|
||||
} else {
|
||||
// Assistant messages with content (text or images) are shown as a group if they have content
|
||||
// No matter whether it has tool calls or not
|
||||
@@ -144,6 +149,25 @@ export function hasToolCalls(message: Message) {
|
||||
);
|
||||
}
|
||||
|
||||
export function hasPresentFiles(message: Message) {
|
||||
return (
|
||||
message.type === "ai" && message.tool_calls?.[0]?.name === "present_files"
|
||||
);
|
||||
}
|
||||
|
||||
export function extractPresentFilesFromMessage(message: Message) {
|
||||
if (message.type !== "ai" || !hasPresentFiles(message)) {
|
||||
return [];
|
||||
}
|
||||
const files = [];
|
||||
for (const toolCall of message.tool_calls ?? []) {
|
||||
if (toolCall.name === "present_files") {
|
||||
files.push(...(toolCall.args.filepaths as string[]));
|
||||
}
|
||||
}
|
||||
return files;
|
||||
}
|
||||
|
||||
export function findToolCallResult(toolCallId: string, messages: Message[]) {
|
||||
for (const message of messages) {
|
||||
if (message.type === "tool" && message.tool_call_id === toolCallId) {
|
||||
|
||||
25
frontend/src/core/utils/files.ts
Normal file
25
frontend/src/core/utils/files.ts
Normal file
@@ -0,0 +1,25 @@
|
||||
export function getFileName(filepath: string) {
|
||||
return filepath.split("/").pop()!;
|
||||
}
|
||||
|
||||
export function getFileExtension(filepath: string) {
|
||||
const fileName = getFileName(filepath);
|
||||
const extension = fileName.split(".").pop()!.toLocaleLowerCase();
|
||||
switch (extension) {
|
||||
case "doc":
|
||||
case "docx":
|
||||
return "Word";
|
||||
case "md":
|
||||
return "Markdown";
|
||||
case "txt":
|
||||
return "Text";
|
||||
case "ppt":
|
||||
case "pptx":
|
||||
return "PowerPoint";
|
||||
case "xls":
|
||||
case "xlsx":
|
||||
return "Excel";
|
||||
default:
|
||||
return extension.toUpperCase();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user