feat: integrated with artifacts

This commit is contained in:
Henry Li
2026-01-17 00:02:03 +08:00
parent 34ca58ed1b
commit 9d64c7e076
6 changed files with 174 additions and 39 deletions

View File

@@ -0,0 +1,14 @@
import { FileIcon } from "lucide-react";
export function ArtifactFileDetail({ filepath }: { filepath: string }) {
return (
<div className="flex size-full items-center justify-center">
<div className="flex items-center gap-2">
<div>
<FileIcon />
</div>
<div>{filepath}</div>
</div>
</div>
);
}

View File

@@ -1,4 +1,5 @@
import { DownloadIcon } from "lucide-react";
import { useCallback } from "react";
import { Button } from "@/components/ui/button";
import {
@@ -9,12 +10,32 @@ import {
CardTitle,
} from "@/components/ui/card";
import { getFileExtension, getFileName } from "@/core/utils/files";
import { cn } from "@/lib/utils";
export function PresentFileList({ files }: { files: string[] }) {
import { useArtifacts } from "./context";
export function ArtifactFileList({
className,
files,
}: {
className?: string;
files: string[];
}) {
const { openArtifact } = useArtifacts();
const handleClick = useCallback(
(filepath: string) => {
openArtifact(filepath);
},
[openArtifact],
);
return (
<ul className="flex w-full flex-col gap-4">
<ul className={cn("flex w-full flex-col gap-4", className)}>
{files.map((file) => (
<Card key={file}>
<Card
className="cursor-pointer"
key={file}
onClick={() => handleClick(file)}
>
<CardHeader>
<CardTitle>{getFileName(file)}</CardTitle>
<CardDescription>{getFileExtension(file)} file</CardDescription>

View File

@@ -0,0 +1,58 @@
import { createContext, useContext, useState, type ReactNode } from "react";
export interface ArtifactsContextType {
artifacts: string[];
selectedArtifact: string | null;
open: boolean;
setOpen: (open: boolean) => void;
addArtifacts: (artifacts: string[]) => void;
openArtifact: (artifact: string) => void;
}
const ArtifactsContext = createContext<ArtifactsContextType | undefined>(
undefined,
);
interface ArtifactsProviderProps {
children: ReactNode;
}
export function ArtifactsProvider({ children }: ArtifactsProviderProps) {
const [artifacts, setArtifacts] = useState<string[]>([]);
const [selectedArtifact, setSelectedArtifact] = useState<string | null>(null);
const [open, setOpen] = useState(false);
const addArtifacts = (newArtifacts: string[]) => {
setArtifacts((prev) => [...prev, ...newArtifacts]);
};
const openArtifact = (artifact: string) => {
setSelectedArtifact(artifact);
setOpen(true);
};
const value: ArtifactsContextType = {
artifacts,
selectedArtifact,
open,
setOpen,
addArtifacts,
openArtifact,
};
return (
<ArtifactsContext.Provider value={value}>
{children}
</ArtifactsContext.Provider>
);
}
export function useArtifacts() {
const context = useContext(ArtifactsContext);
if (context === undefined) {
throw new Error("useArtifacts must be used within an ArtifactsProvider");
}
return context;
}

View File

@@ -0,0 +1,3 @@
export * from "./artifact-file-detail";
export * from "./artifact-file-list";
export * from "./context";

View File

@@ -13,11 +13,11 @@ import {
import type { AgentThreadState } from "@/core/threads";
import { cn } from "@/lib/utils";
import { ArtifactFileList } from "../artifacts/artifact-file-list";
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({
@@ -57,7 +57,7 @@ export function MessageList({
}
}
return (
<PresentFileList key={groupedMessages[0].id} files={files} />
<ArtifactFileList key={groupedMessages[0].id} files={files} />
);
}
return (