feat: support basic file presenting

This commit is contained in:
Henry Li
2026-01-16 22:35:20 +08:00
parent 50a1e407cf
commit 83f367b98a
4 changed files with 102 additions and 4 deletions

View File

@@ -4,7 +4,12 @@ import {
Conversation,
ConversationContent,
} from "@/components/ai-elements/conversation";
import { groupMessages, hasContent } from "@/core/messages/utils";
import {
extractPresentFilesFromMessage,
groupMessages,
hasContent,
hasPresentFiles,
} from "@/core/messages/utils";
import type { AgentThreadState } from "@/core/threads";
import { cn } from "@/lib/utils";
@@ -12,6 +17,7 @@ import { StreamingIndicator } from "../streaming-indicator";
import { MessageGroup } from "./message-group";
import { MessageListItem } from "./message-list-item";
import { PresentFileList } from "./present-file-list";
import { MessageListSkeleton } from "./skeleton";
export function MessageList({
@@ -43,6 +49,17 @@ export function MessageList({
/>
);
}
if (groupedMessages[0] && hasPresentFiles(groupedMessages[0])) {
const files = [];
for (const message of groupedMessages) {
if (hasPresentFiles(message)) {
files.push(...extractPresentFilesFromMessage(message));
}
}
return (
<PresentFileList key={groupedMessages[0].id} files={files} />
);
}
return (
<MessageGroup
key={groupedMessages[0]!.id}

View File

@@ -0,0 +1,32 @@
import { DownloadIcon } from "lucide-react";
import { Button } from "@/components/ui/button";
import {
Card,
CardAction,
CardDescription,
CardHeader,
CardTitle,
} from "@/components/ui/card";
import { getFileExtension, getFileName } from "@/core/utils/files";
export function PresentFileList({ files }: { files: string[] }) {
return (
<ul className="w-full">
{files.map((file) => (
<Card key={file}>
<CardHeader>
<CardTitle>{getFileName(file)}</CardTitle>
<CardDescription>{getFileExtension(file)} file</CardDescription>
<CardAction>
<Button variant="ghost">
<DownloadIcon className="size-4" />
Download
</Button>
</CardAction>
</CardHeader>
</Card>
))}
</ul>
);
}

View File

@@ -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) {

View 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();
}
}