feat: 预估提示词token大小更适合人类阅读

Author: CC11001100

src/pages/CopyButton.tsx

@@ -9,6 +9,16 @@ interface CopyButtonProps {
     files: FileInfo[];
 }
 
+const formatTokenCount = (count: number): string => {
+    if (count >= 10000) {
+        return `${(count / 10000).toFixed(1).replace(/\.0$/, '')}万`;
+    }
+    if (count >= 1000) {
+        return `${(count / 1000).toFixed(1).replace(/\.0$/, '')}k`;
+    }
+    return count.toString();
+};
+
 const CopyButton: React.FC<CopyButtonProps> = ({content, files}) => {
     const [open, setOpen] = useState(false);
     const [message, setMessage] = useState<{ text: string; severity: 'success' | 'error' }>({
@@ -25,6 +35,8 @@ const CopyButton: React.FC<CopyButtonProps> = ({content, files}) => {
         return Math.ceil(totalContent.length / 4);
     }, [content, files]);
 
+    const formattedToken = useMemo(() => formatTokenCount(tokenCount), [tokenCount]);
+
     const handleCopy = async () => {
         const filesContent = files.map(file =>
             `\`\`\`${file.path}\n${file.content}\n\`\`\``
@@ -50,7 +62,7 @@ const CopyButton: React.FC<CopyButtonProps> = ({content, files}) => {
                     padding: '12px 24px',
                     width: '100%'
                 }}>
-                构造提示词并复制到剪切板 (约 {tokenCount} Tokens)
+                构造提示词并复制到剪切板 (约 {formattedToken} Tokens)
             </Button>
             <Snackbar
                 open={open}