chore: playground api request

2024-06-12 18:13:23 +08:00
parent 405d95c81f
commit c9ea96e078
25 changed files with 594 additions and 147 deletions
@@ -1,11 +1,31 @@
+import { Space, Tooltip } from 'antd';
 import '../style/reference-params.less';

-const ReferenceParams = () => {
+interface ReferenceParamsProps {
+  usage: {
+    completion_tokens: number;
+    prompt_tokens: number;
+    total_tokens: number;
+  };
+}
+
+const ReferenceParams = (props: ReferenceParamsProps) => {
+  const { usage } = props;
+  if (!usage) {
+    return null;
+  }
  return (
    <div className="reference-params">
-      <span>Inference: 597 ms</span>
-      <span style={{ padding: '10px' }}></span>
-      <span>Tokens/s: 561</span>
+      <Tooltip
+        title={
+          <Space>
+            <span>Completion: {usage.completion_tokens}</span>
+            <span>Prompt: {usage.prompt_tokens}</span>
+          </Space>
+        }
+      >
+        <span>Token Usage: {usage.total_tokens}</span>
+      </Tooltip>
    </div>
  );
 };