i/next-ai-draw-io
1
0
Fork 0
mirror of https://github.com/DayuanJiang/next-ai-draw-io.git synced 2026-01-02 22:32:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: enhance ChatPanel to fetch chart data before submission and track update steps

Browse Source
This commit is contained in:
dayuan.jiang
2025-03-22 16:03:03 +00:00
parent a27c94f798
commit df483ae647
1 changed files with 39 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
components/chatPanel.tsx
View File

@@ -8,14 +8,16 @@ import { ScrollArea } from "@/components/ui/scroll-area"
import { Button } from "@/components/ui/button" import { Button } from "@/components/ui/button"
import { useChat } from '@ai-sdk/react'; import { useChat } from '@ai-sdk/react';
import { ChatInput } from "@/components/chat-input" import { ChatInput } from "@/components/chat-input"
import { convertToLegalXml } from "@/lib/utils"
interface ChatPanelProps { interface ChatPanelProps {
onDisplayChart: (xml: string) => void; onDisplayChart: (xml: string) => void;
onFetchChart: () => Promise<string>; onFetchChart: () => Promise<string>;
} }
export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelProps) { export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelProps) {
const [previousXml, setPreviousXml] = useState<string>(""); // Add a step counter to track updates
const stepCounterRef = useRef<number>(0);
// Remove the currentXmlRef and related useEffect
const { messages, input, handleInputChange, handleSubmit, status, error, setInput, setMessages, data } = useChat({ const { messages, input, handleInputChange, handleSubmit, status, error, setInput, setMessages, data } = useChat({
maxSteps: 5, maxSteps: 5,
async onToolCall({ toolCall }) { async onToolCall({ toolCall }) {
@@ -41,14 +43,19 @@ export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelPro
console.log("Data updated:", data); console.log("Data updated:", data);
}, [messages]) }, [messages])
const onFormSubmit = (e: React.FormEvent<HTMLFormElement>) => { const onFormSubmit = async (e: React.FormEvent<HTMLFormElement>) => {
e.preventDefault() e.preventDefault()
if (input.trim() && status !== "streaming") { if (input.trim() && status !== "streaming") {
try {
// Fetch chart data before setting input
const chartXml = await onFetchChart();
// Now use the fetched data to set input
setInput( setInput(
` `
Current diagram XML: Current diagram XML:
"""xml """xml
${onFetchChart()} ${chartXml}
""" """
User input: User input:
"""md """md
@@ -57,12 +64,11 @@ export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelPro
` `
) )
handleSubmit(e) handleSubmit(e)
} catch (error) {
console.error("Error fetching chart data:", error);
}
} }
} }
// Helper function to render tool invocations // Helper function to render tool invocations
const renderToolInvocation = (toolInvocation: any) => { const renderToolInvocation = (toolInvocation: any) => {
@@ -73,24 +79,24 @@ export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelPro
switch (toolInvocation.state) { switch (toolInvocation.state) {
case 'partial-call': { case 'partial-call': {
const currentXml = toolInvocation.args?.xml || ""; const currentXml = toolInvocation.args?.xml || "";
const shouldShowPartialArgs = currentXml &&
(!previousXml || Math.abs(currentXml.length - previousXml.length) > 50);
// Update previous XML for next comparison // Increment the step counter
if (currentXml) { stepCounterRef.current += 1;
setPreviousXml(currentXml); // Log the current step
// Determine whether to show details based on a simple threshold
// Rather than comparing lengths (which can cause re-renders)
if (stepCounterRef.current >= 20 && stepCounterRef.current % 20 === 0) {
onDisplayChart(convertToLegalXml(currentXml));
} }
return ( return (
<div key={callId} className="mt-2 text-sm bg-blue-50 p-2 rounded border border-blue-200"> <div key={callId} className="mt-2 text-sm bg-blue-50 p-2 rounded border border-blue-200">
<div className="font-medium">Preparing to display diagram...</div> <div className="font-medium">Generating diagram...</div>
<div className="text-xs text-gray-500 mt-1"> <div className="text-xs text-gray-500 mt-1">
Tool: display_diagram Tool: display_diagram
{shouldShowPartialArgs && toolInvocation.args && (
<div className="mt-1 font-mono text-xs"> <div className="mt-1 font-mono text-xs">
Partial args: {JSON.stringify(toolInvocation.args, null, 2)} onDisplayChart
</div> </div>
)}
</div> </div>
</div> </div>
); );
@@ -110,7 +116,7 @@ export default function ChatPanel({ onDisplayChart, onFetchChart }: ChatPanelPro
case 'result': case 'result':
return ( return (
<div key={callId} className="mt-2 text-sm bg-green-50 p-2 rounded border border-green-200"> <div key={callId} className="mt-2 text-sm bg-green-50 p-2 rounded border border-green-200">
<div className="font-medium">Diagram displayed</div> <div className="font-medium">Diagram generated</div>
<div className="text-xs text-gray-500 mt-1"> <div className="text-xs text-gray-500 mt-1">
Result: {toolInvocation.result} Result: {toolInvocation.result}
</div> </div>