fix: show embedding token usage

2025-06-06 14:57:40 +08:00
parent d5059f12d5
commit a2231e6b42
4 changed files with 128 additions and 92 deletions
@@ -0,0 +1,36 @@
+import { useIntl } from '@umijs/max';
+import React from 'react';
+import styled from 'styled-components';
+
+const TokenUsageWrapper = styled.div`
+  font-size: var(--font-size-small);
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  padding: 2px 5px;
+  .text {
+    color: var(--ant-orange);
+  }
+`;
+
+const TokenUsage: React.FC<{
+  tokenResult?: any;
+  [key: string]: any;
+}> = ({ tokenResult, ...rest }) => {
+  const intl = useIntl();
+  if (!tokenResult) {
+    return null;
+  }
+  return (
+    <TokenUsageWrapper {...rest}>
+      {tokenResult?.total_tokens && (
+        <span className="text">
+          {intl.formatMessage({ id: 'playground.tokenusage' })}:{' '}
+          {tokenResult?.total_tokens}
+        </span>
+      )}
+    </TokenUsageWrapper>
+  );
+};
+
+export default TokenUsage;