File size: 3,292 Bytes
5da4770 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import React from 'react';
import { ToolViewProps } from '../types';
import { formatTimestamp, getToolTitle } from '../utils';
import { getToolIcon } from '../../utils';
import { CircleDashed, CheckCircle, AlertTriangle } from 'lucide-react';
import { cn } from '@/lib/utils';
export interface ToolViewWrapperProps extends ToolViewProps {
children: React.ReactNode;
headerContent?: React.ReactNode;
footerContent?: React.ReactNode;
className?: string;
contentClassName?: string;
headerClassName?: string;
footerClassName?: string;
showStatus?: boolean;
customStatus?: {
success?: string;
failure?: string;
streaming?: string;
};
}
export function ToolViewWrapper({
name = 'unknown',
isSuccess = true,
isStreaming = false,
assistantTimestamp,
toolTimestamp,
children,
headerContent,
footerContent,
className,
contentClassName,
headerClassName,
footerClassName,
showStatus = true,
customStatus,
}: ToolViewWrapperProps) {
const toolTitle = getToolTitle(name);
const Icon = getToolIcon(name);
return (
<div className={cn("flex flex-col h-full", className)}>
{(headerContent || showStatus) && (
<div className={cn(
"flex items-center p-2 bg-zinc-100 dark:bg-zinc-900 justify-between border-zinc-200 dark:border-zinc-800",
headerClassName
)}>
<div className="flex ml-1 items-center">
{Icon && <Icon className="h-4 w-4 mr-2 text-zinc-600 dark:text-zinc-400" />}
<span className="text-xs font-medium text-zinc-700 dark:text-zinc-300">
{toolTitle}
</span>
</div>
{headerContent}
</div>
)}
<div className={cn("flex-1 overflow-auto", contentClassName)}>
{children}
</div>
{(footerContent || showStatus) && (
<div className={cn(
"p-4 border-t border-zinc-200 dark:border-zinc-800",
footerClassName
)}>
<div className="flex items-center justify-between text-xs text-zinc-500 dark:text-zinc-400">
{!isStreaming && showStatus && (
<div className="flex items-center gap-2">
{isSuccess ? (
<CheckCircle className="h-3.5 w-3.5 text-emerald-500" />
) : (
<AlertTriangle className="h-3.5 w-3.5 text-red-500" />
)}
<span>
{isSuccess
? customStatus?.success || "Completed successfully"
: customStatus?.failure || "Execution failed"}
</span>
</div>
)}
{isStreaming && showStatus && (
<div className="flex items-center gap-2">
<CircleDashed className="h-3.5 w-3.5 text-blue-500 animate-spin" />
<span>{customStatus?.streaming || "Processing..."}</span>
</div>
)}
<div className="text-xs">
{toolTimestamp && !isStreaming
? formatTimestamp(toolTimestamp)
: assistantTimestamp
? formatTimestamp(assistantTimestamp)
: ""}
</div>
{footerContent}
</div>
</div>
)}
</div>
);
}
|