mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-14 18:54:46 +08:00
feat: integrated with artifacts
This commit is contained in:
@@ -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>
|
||||
);
|
||||
}
|
||||
@@ -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>
|
||||
58
frontend/src/components/workspace/artifacts/context.tsx
Normal file
58
frontend/src/components/workspace/artifacts/context.tsx
Normal 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;
|
||||
}
|
||||
3
frontend/src/components/workspace/artifacts/index.ts
Normal file
3
frontend/src/components/workspace/artifacts/index.ts
Normal file
@@ -0,0 +1,3 @@
|
||||
export * from "./artifact-file-detail";
|
||||
export * from "./artifact-file-list";
|
||||
export * from "./context";
|
||||
@@ -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 (
|
||||
|
||||
Reference in New Issue
Block a user