wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-18 20:14:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add ToggleGroup

Browse Source
This commit is contained in:
Henry Li
2026-01-19 19:41:46 +08:00
parent 74d4a16492
commit 24ca87d650
9 changed files with 316 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
frontend/src/core/artifacts/hooks.ts
View File

@@ -3,7 +3,7 @@ import { useMemo } from "react";
import { useThread } from "@/components/workspace/messages/context";
import { loadArtifactContent } from "./loader";
import { loadArtifactContent, loadArtifactContentFromToolCall } from "./loader";
export function useArtifactContent({
filepath,
@@ -20,25 +20,10 @@ export function useArtifactContent({
const { thread } = useThread();
const content = useMemo(() => {
if (isWriteFile) {
const url = new URL(filepath);
const toolCallId = url.searchParams.get("tool_call_id");
const messageId = url.searchParams.get("message_id");
if (messageId && toolCallId) {
const message = thread.messages.find(
(message) => message.id === messageId,
);
if (message?.type === "ai" && message.tool_calls) {
const toolCall = message.tool_calls.find(
(toolCall) => toolCall.id === toolCallId,
);
if (toolCall) {
return toolCall.args.content;
}
}
}
return loadArtifactContentFromToolCall({ url: filepath, thread });
}
return null;
}, [filepath, isWriteFile, thread.messages]);
}, [filepath, isWriteFile, thread]);
const { data, isLoading, error } = useQuery({
queryKey: ["artifact", filepath, threadId],
queryFn: () => {

27
frontend/src/core/artifacts/loader.ts
View File

@@ -1,3 +1,7 @@
import type { UseStream } from "@langchain/langgraph-sdk/react";
import type { AgentThreadState } from "../threads";
import { urlOfArtifact } from "./utils";
export async function loadArtifactContent({
@@ -12,3 +16,26 @@ export async function loadArtifactContent({
const text = await response.text();
return text;
}
export function loadArtifactContentFromToolCall({
url: urlString,
thread,
}: {
url: string;
thread: UseStream<AgentThreadState>;
}) {
const url = new URL(urlString);
const toolCallId = url.searchParams.get("tool_call_id");
const messageId = url.searchParams.get("message_id");
if (messageId && toolCallId) {
const message = thread.messages.find((message) => message.id === messageId);
if (message?.type === "ai" && message.tool_calls) {
const toolCall = message.tool_calls.find(
(toolCall) => toolCall.id === toolCallId,
);
if (toolCall) {
return toolCall.args.content;
}
}
}
}