LogsTable.js 33 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135
  1. import React, { useContext, useEffect, useState } from 'react';
  2. import { useTranslation } from 'react-i18next';
  3. import {
  4. API,
  5. copy,
  6. getTodayStartTimestamp,
  7. isAdmin,
  8. showError,
  9. showSuccess,
  10. timestamp2string,
  11. } from '../helpers';
  12. import {
  13. Avatar,
  14. Button, Descriptions,
  15. Form,
  16. Layout,
  17. Modal, Popover,
  18. Select,
  19. Space,
  20. Spin,
  21. Table,
  22. Tag,
  23. Tooltip,
  24. Checkbox
  25. } from '@douyinfe/semi-ui';
  26. import { ITEMS_PER_PAGE } from '../constants';
  27. import {
  28. renderAudioModelPrice, renderGroup,
  29. renderModelPrice, renderModelPriceSimple,
  30. renderNumber,
  31. renderQuota,
  32. stringToColor
  33. } from '../helpers/render';
  34. import Paragraph from '@douyinfe/semi-ui/lib/es/typography/paragraph';
  35. import { getLogOther } from '../helpers/other.js';
  36. import { StyleContext } from '../context/Style/index.js';
  37. import { IconInherit, IconRefresh, IconSetting } from '@douyinfe/semi-icons';
  38. const { Header } = Layout;
  39. function renderTimestamp(timestamp) {
  40. return <>{timestamp2string(timestamp)}</>;
  41. }
  42. const MODE_OPTIONS = [
  43. { key: 'all', text: 'all', value: 'all' },
  44. { key: 'self', text: 'current user', value: 'self' },
  45. ];
  46. const colors = [
  47. 'amber',
  48. 'blue',
  49. 'cyan',
  50. 'green',
  51. 'grey',
  52. 'indigo',
  53. 'light-blue',
  54. 'lime',
  55. 'orange',
  56. 'pink',
  57. 'purple',
  58. 'red',
  59. 'teal',
  60. 'violet',
  61. 'yellow',
  62. ];
  63. const LogsTable = () => {
  64. const { t } = useTranslation();
  65. function renderType(type) {
  66. switch (type) {
  67. case 1:
  68. return <Tag color='cyan' size='large'>{t('充值')}</Tag>;
  69. case 2:
  70. return <Tag color='lime' size='large'>{t('消费')}</Tag>;
  71. case 3:
  72. return <Tag color='orange' size='large'>{t('管理')}</Tag>;
  73. case 4:
  74. return <Tag color='purple' size='large'>{t('系统')}</Tag>;
  75. default:
  76. return <Tag color='black' size='large'>{t('未知')}</Tag>;
  77. }
  78. }
  79. function renderIsStream(bool) {
  80. if (bool) {
  81. return <Tag color='blue' size='large'>{t('流')}</Tag>;
  82. } else {
  83. return <Tag color='purple' size='large'>{t('非流')}</Tag>;
  84. }
  85. }
  86. function renderUseTime(type) {
  87. const time = parseInt(type);
  88. if (time < 101) {
  89. return (
  90. <Tag color='green' size='large'>
  91. {' '}
  92. {time} s{' '}
  93. </Tag>
  94. );
  95. } else if (time < 300) {
  96. return (
  97. <Tag color='orange' size='large'>
  98. {' '}
  99. {time} s{' '}
  100. </Tag>
  101. );
  102. } else {
  103. return (
  104. <Tag color='red' size='large'>
  105. {' '}
  106. {time} s{' '}
  107. </Tag>
  108. );
  109. }
  110. }
  111. function renderFirstUseTime(type) {
  112. let time = parseFloat(type) / 1000.0;
  113. time = time.toFixed(1);
  114. if (time < 3) {
  115. return (
  116. <Tag color='green' size='large'>
  117. {' '}
  118. {time} s{' '}
  119. </Tag>
  120. );
  121. } else if (time < 10) {
  122. return (
  123. <Tag color='orange' size='large'>
  124. {' '}
  125. {time} s{' '}
  126. </Tag>
  127. );
  128. } else {
  129. return (
  130. <Tag color='red' size='large'>
  131. {' '}
  132. {time} s{' '}
  133. </Tag>
  134. );
  135. }
  136. }
  137. function renderModelName(record) {
  138. let other = getLogOther(record.other);
  139. let modelMapped = other?.is_model_mapped && other?.upstream_model_name && other?.upstream_model_name !== '';
  140. if (!modelMapped) {
  141. return <Tag
  142. color={stringToColor(record.model_name)}
  143. size='large'
  144. onClick={(event) => {
  145. copyText(event, record.model_name).then(r => {});
  146. }}
  147. >
  148. {' '}{record.model_name}{' '}
  149. </Tag>;
  150. } else {
  151. return (
  152. <>
  153. <Space vertical align={'start'}>
  154. <Popover content={
  155. <div style={{padding: 10}}>
  156. <Space vertical align={'start'}>
  157. <Tag
  158. color={stringToColor(record.model_name)}
  159. size='large'
  160. onClick={(event) => {
  161. copyText(event, record.model_name).then(r => {});
  162. }}
  163. >
  164. {t('请求并计费模型')}{' '}{record.model_name}{' '}
  165. </Tag>
  166. <Tag
  167. color={stringToColor(other.upstream_model_name)}
  168. size='large'
  169. onClick={(event) => {
  170. copyText(event, other.upstream_model_name).then(r => {});
  171. }}
  172. >
  173. {t('实际模型')}{' '}{other.upstream_model_name}{' '}
  174. </Tag>
  175. </Space>
  176. </div>
  177. }>
  178. <Tag
  179. color={stringToColor(record.model_name)}
  180. size='large'
  181. onClick={(event) => {
  182. copyText(event, record.model_name).then(r => {});
  183. }}
  184. suffixIcon={<IconRefresh style={{width: '0.8em', height: '0.8em', opacity: 0.6}} />}
  185. >
  186. {' '}{record.model_name}{' '}
  187. </Tag>
  188. </Popover>
  189. {/*<Tooltip content={t('实际模型')}>*/}
  190. {/* <Tag*/}
  191. {/* color={stringToColor(other.upstream_model_name)}*/}
  192. {/* size='large'*/}
  193. {/* onClick={(event) => {*/}
  194. {/* copyText(event, other.upstream_model_name).then(r => {});*/}
  195. {/* }}*/}
  196. {/* >*/}
  197. {/* {' '}{other.upstream_model_name}{' '}*/}
  198. {/* </Tag>*/}
  199. {/*</Tooltip>*/}
  200. </Space>
  201. </>
  202. );
  203. }
  204. }
  205. // Define column keys for selection
  206. const COLUMN_KEYS = {
  207. TIME: 'time',
  208. CHANNEL: 'channel',
  209. USERNAME: 'username',
  210. TOKEN: 'token',
  211. GROUP: 'group',
  212. TYPE: 'type',
  213. MODEL: 'model',
  214. USE_TIME: 'use_time',
  215. PROMPT: 'prompt',
  216. COMPLETION: 'completion',
  217. COST: 'cost',
  218. RETRY: 'retry',
  219. DETAILS: 'details'
  220. };
  221. // State for column visibility
  222. const [visibleColumns, setVisibleColumns] = useState({});
  223. const [showColumnSelector, setShowColumnSelector] = useState(false);
  224. // Load saved column preferences from localStorage
  225. useEffect(() => {
  226. const savedColumns = localStorage.getItem('logs-table-columns');
  227. if (savedColumns) {
  228. try {
  229. const parsed = JSON.parse(savedColumns);
  230. // Make sure all columns are accounted for
  231. const defaults = getDefaultColumnVisibility();
  232. const merged = { ...defaults, ...parsed };
  233. setVisibleColumns(merged);
  234. } catch (e) {
  235. console.error('Failed to parse saved column preferences', e);
  236. initDefaultColumns();
  237. }
  238. } else {
  239. initDefaultColumns();
  240. }
  241. }, []);
  242. // Get default column visibility based on user role
  243. const getDefaultColumnVisibility = () => {
  244. return {
  245. [COLUMN_KEYS.TIME]: true,
  246. [COLUMN_KEYS.CHANNEL]: isAdminUser,
  247. [COLUMN_KEYS.USERNAME]: isAdminUser,
  248. [COLUMN_KEYS.TOKEN]: true,
  249. [COLUMN_KEYS.GROUP]: true,
  250. [COLUMN_KEYS.TYPE]: true,
  251. [COLUMN_KEYS.MODEL]: true,
  252. [COLUMN_KEYS.USE_TIME]: true,
  253. [COLUMN_KEYS.PROMPT]: true,
  254. [COLUMN_KEYS.COMPLETION]: true,
  255. [COLUMN_KEYS.COST]: true,
  256. [COLUMN_KEYS.RETRY]: isAdminUser,
  257. [COLUMN_KEYS.DETAILS]: true
  258. };
  259. };
  260. // Initialize default column visibility
  261. const initDefaultColumns = () => {
  262. const defaults = getDefaultColumnVisibility();
  263. setVisibleColumns(defaults);
  264. localStorage.setItem('logs-table-columns', JSON.stringify(defaults));
  265. };
  266. // Handle column visibility change
  267. const handleColumnVisibilityChange = (columnKey, checked) => {
  268. const updatedColumns = { ...visibleColumns, [columnKey]: checked };
  269. setVisibleColumns(updatedColumns);
  270. };
  271. // Handle "Select All" checkbox
  272. const handleSelectAll = (checked) => {
  273. const allKeys = Object.keys(COLUMN_KEYS).map(key => COLUMN_KEYS[key]);
  274. const updatedColumns = {};
  275. allKeys.forEach(key => {
  276. // For admin-only columns, only enable them if user is admin
  277. if ((key === COLUMN_KEYS.CHANNEL || key === COLUMN_KEYS.USERNAME || key === COLUMN_KEYS.RETRY) && !isAdminUser) {
  278. updatedColumns[key] = false;
  279. } else {
  280. updatedColumns[key] = checked;
  281. }
  282. });
  283. setVisibleColumns(updatedColumns);
  284. };
  285. // Define all columns
  286. const allColumns = [
  287. {
  288. key: COLUMN_KEYS.TIME,
  289. title: t('时间'),
  290. dataIndex: 'timestamp2string',
  291. },
  292. {
  293. key: COLUMN_KEYS.CHANNEL,
  294. title: t('渠道'),
  295. dataIndex: 'channel',
  296. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  297. render: (text, record, index) => {
  298. return isAdminUser ? (
  299. record.type === 0 || record.type === 2 ? (
  300. <div>
  301. {
  302. <Tooltip content={record.channel_name || '[未知]'}>
  303. <Tag
  304. color={colors[parseInt(text) % colors.length]}
  305. size='large'
  306. >
  307. {' '}
  308. {text}{' '}
  309. </Tag>
  310. </Tooltip>
  311. }
  312. </div>
  313. ) : (
  314. <></>
  315. )
  316. ) : (
  317. <></>
  318. );
  319. },
  320. },
  321. {
  322. key: COLUMN_KEYS.USERNAME,
  323. title: t('用户'),
  324. dataIndex: 'username',
  325. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  326. render: (text, record, index) => {
  327. return isAdminUser ? (
  328. <div>
  329. <Avatar
  330. size='small'
  331. color={stringToColor(text)}
  332. style={{ marginRight: 4 }}
  333. onClick={(event) => {
  334. event.stopPropagation();
  335. showUserInfo(record.user_id)
  336. }}
  337. >
  338. {typeof text === 'string' && text.slice(0, 1)}
  339. </Avatar>
  340. {text}
  341. </div>
  342. ) : (
  343. <></>
  344. );
  345. },
  346. },
  347. {
  348. key: COLUMN_KEYS.TOKEN,
  349. title: t('令牌'),
  350. dataIndex: 'token_name',
  351. render: (text, record, index) => {
  352. return record.type === 0 || record.type === 2 ? (
  353. <div>
  354. <Tag
  355. color='grey'
  356. size='large'
  357. onClick={(event) => {
  358. //cancel the row click event
  359. copyText(event, text);
  360. }}
  361. >
  362. {' '}
  363. {t(text)}{' '}
  364. </Tag>
  365. </div>
  366. ) : (
  367. <></>
  368. );
  369. },
  370. },
  371. {
  372. key: COLUMN_KEYS.GROUP,
  373. title: t('分组'),
  374. dataIndex: 'group',
  375. render: (text, record, index) => {
  376. if (record.type === 0 || record.type === 2) {
  377. if (record.group) {
  378. return (
  379. <>
  380. {renderGroup(record.group)}
  381. </>
  382. );
  383. } else {
  384. let other = null;
  385. try {
  386. other = JSON.parse(record.other);
  387. } catch (e) {
  388. console.error(`Failed to parse record.other: "${record.other}".`, e);
  389. }
  390. if (other === null) {
  391. return <></>;
  392. }
  393. if (other.group !== undefined) {
  394. return (
  395. <>
  396. {renderGroup(other.group)}
  397. </>
  398. );
  399. } else {
  400. return <></>;
  401. }
  402. }
  403. } else {
  404. return <></>;
  405. }
  406. },
  407. },
  408. {
  409. key: COLUMN_KEYS.TYPE,
  410. title: t('类型'),
  411. dataIndex: 'type',
  412. render: (text, record, index) => {
  413. return <>{renderType(text)}</>;
  414. },
  415. },
  416. {
  417. key: COLUMN_KEYS.MODEL,
  418. title: t('模型'),
  419. dataIndex: 'model_name',
  420. render: (text, record, index) => {
  421. return record.type === 0 || record.type === 2 ? (
  422. <>{renderModelName(record)}</>
  423. ) : (
  424. <></>
  425. );
  426. },
  427. },
  428. {
  429. key: COLUMN_KEYS.USE_TIME,
  430. title: t('用时/首字'),
  431. dataIndex: 'use_time',
  432. render: (text, record, index) => {
  433. if (record.is_stream) {
  434. let other = getLogOther(record.other);
  435. return (
  436. <>
  437. <Space>
  438. {renderUseTime(text)}
  439. {renderFirstUseTime(other.frt)}
  440. {renderIsStream(record.is_stream)}
  441. </Space>
  442. </>
  443. );
  444. } else {
  445. return (
  446. <>
  447. <Space>
  448. {renderUseTime(text)}
  449. {renderIsStream(record.is_stream)}
  450. </Space>
  451. </>
  452. );
  453. }
  454. },
  455. },
  456. {
  457. key: COLUMN_KEYS.PROMPT,
  458. title: t('提示'),
  459. dataIndex: 'prompt_tokens',
  460. render: (text, record, index) => {
  461. return record.type === 0 || record.type === 2 ? (
  462. <>{<span> {text} </span>}</>
  463. ) : (
  464. <></>
  465. );
  466. },
  467. },
  468. {
  469. key: COLUMN_KEYS.COMPLETION,
  470. title: t('补全'),
  471. dataIndex: 'completion_tokens',
  472. render: (text, record, index) => {
  473. return parseInt(text) > 0 &&
  474. (record.type === 0 || record.type === 2) ? (
  475. <>{<span> {text} </span>}</>
  476. ) : (
  477. <></>
  478. );
  479. },
  480. },
  481. {
  482. key: COLUMN_KEYS.COST,
  483. title: t('花费'),
  484. dataIndex: 'quota',
  485. render: (text, record, index) => {
  486. return record.type === 0 || record.type === 2 ? (
  487. <>{renderQuota(text, 6)}</>
  488. ) : (
  489. <></>
  490. );
  491. },
  492. },
  493. {
  494. key: COLUMN_KEYS.RETRY,
  495. title: t('重试'),
  496. dataIndex: 'retry',
  497. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  498. render: (text, record, index) => {
  499. let content = t('渠道') + `:${record.channel}`;
  500. if (record.other !== '') {
  501. let other = JSON.parse(record.other);
  502. if (other === null) {
  503. return <></>;
  504. }
  505. if (other.admin_info !== undefined) {
  506. if (
  507. other.admin_info.use_channel !== null &&
  508. other.admin_info.use_channel !== undefined &&
  509. other.admin_info.use_channel !== ''
  510. ) {
  511. // channel id array
  512. let useChannel = other.admin_info.use_channel;
  513. let useChannelStr = useChannel.join('->');
  514. content = t('渠道') + `:${useChannelStr}`;
  515. }
  516. }
  517. }
  518. return isAdminUser ? <div>{content}</div> : <></>;
  519. },
  520. },
  521. {
  522. key: COLUMN_KEYS.DETAILS,
  523. title: t('详情'),
  524. dataIndex: 'content',
  525. render: (text, record, index) => {
  526. let other = getLogOther(record.other);
  527. if (other == null || record.type !== 2) {
  528. return (
  529. <Paragraph
  530. ellipsis={{
  531. rows: 2,
  532. showTooltip: {
  533. type: 'popover',
  534. opts: { style: { width: 240 } },
  535. },
  536. }}
  537. style={{ maxWidth: 240 }}
  538. >
  539. {text}
  540. </Paragraph>
  541. );
  542. }
  543. let content = renderModelPriceSimple(
  544. other.model_ratio,
  545. other.model_price,
  546. other.group_ratio,
  547. other.cache_tokens || 0,
  548. other.cache_ratio || 1.0,
  549. );
  550. return (
  551. <Paragraph
  552. ellipsis={{
  553. rows: 2,
  554. }}
  555. style={{ maxWidth: 240 }}
  556. >
  557. {content}
  558. </Paragraph>
  559. );
  560. },
  561. },
  562. ];
  563. // Update table when column visibility changes
  564. useEffect(() => {
  565. if (Object.keys(visibleColumns).length > 0) {
  566. // Save to localStorage
  567. localStorage.setItem('logs-table-columns', JSON.stringify(visibleColumns));
  568. }
  569. }, [visibleColumns]);
  570. // Filter columns based on visibility settings
  571. const getVisibleColumns = () => {
  572. return allColumns.filter(column => visibleColumns[column.key]);
  573. };
  574. // Column selector modal
  575. const renderColumnSelector = () => {
  576. return (
  577. <Modal
  578. title={t('列设置')}
  579. visible={showColumnSelector}
  580. onCancel={() => setShowColumnSelector(false)}
  581. footer={
  582. <>
  583. <Button onClick={() => initDefaultColumns()}>{t('重置')}</Button>
  584. <Button onClick={() => setShowColumnSelector(false)}>{t('取消')}</Button>
  585. <Button type="primary" onClick={() => setShowColumnSelector(false)}>{t('确定')}</Button>
  586. </>
  587. }
  588. >
  589. <div style={{ marginBottom: 20 }}>
  590. <Checkbox
  591. checked={Object.values(visibleColumns).every(v => v === true)}
  592. indeterminate={Object.values(visibleColumns).some(v => v === true) && !Object.values(visibleColumns).every(v => v === true)}
  593. onChange={e => handleSelectAll(e.target.checked)}
  594. >
  595. {t('全选')}
  596. </Checkbox>
  597. </div>
  598. <div style={{
  599. display: 'flex',
  600. flexWrap: 'wrap',
  601. maxHeight: '400px',
  602. overflowY: 'auto',
  603. border: '1px solid var(--semi-color-border)',
  604. borderRadius: '6px',
  605. padding: '16px'
  606. }}>
  607. {allColumns.map(column => {
  608. // Skip admin-only columns for non-admin users
  609. if (!isAdminUser && (column.key === COLUMN_KEYS.CHANNEL ||
  610. column.key === COLUMN_KEYS.USERNAME ||
  611. column.key === COLUMN_KEYS.RETRY)) {
  612. return null;
  613. }
  614. return (
  615. <div key={column.key} style={{ width: '50%', marginBottom: 16, paddingRight: 8 }}>
  616. <Checkbox
  617. checked={!!visibleColumns[column.key]}
  618. onChange={e => handleColumnVisibilityChange(column.key, e.target.checked)}
  619. >
  620. {column.title}
  621. </Checkbox>
  622. </div>
  623. );
  624. })}
  625. </div>
  626. </Modal>
  627. );
  628. };
  629. const [styleState, styleDispatch] = useContext(StyleContext);
  630. const [logs, setLogs] = useState([]);
  631. const [expandData, setExpandData] = useState({});
  632. const [showStat, setShowStat] = useState(false);
  633. const [loading, setLoading] = useState(false);
  634. const [loadingStat, setLoadingStat] = useState(false);
  635. const [activePage, setActivePage] = useState(1);
  636. const [logCount, setLogCount] = useState(ITEMS_PER_PAGE);
  637. const [pageSize, setPageSize] = useState(ITEMS_PER_PAGE);
  638. const [logType, setLogType] = useState(0);
  639. const isAdminUser = isAdmin();
  640. let now = new Date();
  641. // 初始化start_timestamp为今天0点
  642. const [inputs, setInputs] = useState({
  643. username: '',
  644. token_name: '',
  645. model_name: '',
  646. start_timestamp: timestamp2string(getTodayStartTimestamp()),
  647. end_timestamp: timestamp2string(now.getTime() / 1000 + 3600),
  648. channel: '',
  649. group: '',
  650. });
  651. const {
  652. username,
  653. token_name,
  654. model_name,
  655. start_timestamp,
  656. end_timestamp,
  657. channel,
  658. group,
  659. } = inputs;
  660. const [stat, setStat] = useState({
  661. quota: 0,
  662. token: 0,
  663. });
  664. const handleInputChange = (value, name) => {
  665. setInputs(inputs => ({ ...inputs, [name]: value }));
  666. };
  667. const getLogSelfStat = async () => {
  668. let localStartTimestamp = Date.parse(start_timestamp) / 1000;
  669. let localEndTimestamp = Date.parse(end_timestamp) / 1000;
  670. let url = `/api/log/self/stat?type=${logType}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&group=${group}`;
  671. url = encodeURI(url);
  672. let res = await API.get(url);
  673. const { success, message, data } = res.data;
  674. if (success) {
  675. setStat(data);
  676. } else {
  677. showError(message);
  678. }
  679. };
  680. const getLogStat = async () => {
  681. let localStartTimestamp = Date.parse(start_timestamp) / 1000;
  682. let localEndTimestamp = Date.parse(end_timestamp) / 1000;
  683. let url = `/api/log/stat?type=${logType}&username=${username}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&channel=${channel}&group=${group}`;
  684. url = encodeURI(url);
  685. let res = await API.get(url);
  686. const { success, message, data } = res.data;
  687. if (success) {
  688. setStat(data);
  689. } else {
  690. showError(message);
  691. }
  692. };
  693. const handleEyeClick = async () => {
  694. if (loadingStat) {
  695. return;
  696. }
  697. setLoadingStat(true);
  698. if (isAdminUser) {
  699. await getLogStat();
  700. } else {
  701. await getLogSelfStat();
  702. }
  703. setShowStat(true);
  704. setLoadingStat(false);
  705. };
  706. const showUserInfo = async (userId) => {
  707. if (!isAdminUser) {
  708. return;
  709. }
  710. const res = await API.get(`/api/user/${userId}`);
  711. const { success, message, data } = res.data;
  712. if (success) {
  713. Modal.info({
  714. title: t('用户信息'),
  715. content: (
  716. <div style={{ padding: 12 }}>
  717. <p>{t('用户名')}: {data.username}</p>
  718. <p>{t('余额')}: {renderQuota(data.quota)}</p>
  719. <p>{t('已用额度')}:{renderQuota(data.used_quota)}</p>
  720. <p>{t('请求次数')}:{renderNumber(data.request_count)}</p>
  721. </div>
  722. ),
  723. centered: true,
  724. });
  725. } else {
  726. showError(message);
  727. }
  728. };
  729. const setLogsFormat = (logs) => {
  730. let expandDatesLocal = {};
  731. for (let i = 0; i < logs.length; i++) {
  732. logs[i].timestamp2string = timestamp2string(logs[i].created_at);
  733. logs[i].key = logs[i].id;
  734. let other = getLogOther(logs[i].other);
  735. let expandDataLocal = [];
  736. if (isAdmin()) {
  737. // let content = '渠道:' + logs[i].channel;
  738. // if (other.admin_info !== undefined) {
  739. // if (
  740. // other.admin_info.use_channel !== null &&
  741. // other.admin_info.use_channel !== undefined &&
  742. // other.admin_info.use_channel !== ''
  743. // ) {
  744. // // channel id array
  745. // let useChannel = other.admin_info.use_channel;
  746. // let useChannelStr = useChannel.join('->');
  747. // content = `渠道:${useChannelStr}`;
  748. // }
  749. // }
  750. // expandDataLocal.push({
  751. // key: '渠道重试',
  752. // value: content,
  753. // })
  754. }
  755. if (isAdminUser && (logs[i].type === 0 || logs[i].type === 2)) {
  756. expandDataLocal.push({
  757. key: t('渠道信息'),
  758. value: `${logs[i].channel} - ${logs[i].channel_name || '[未知]'}`
  759. });
  760. }
  761. if (other?.ws || other?.audio) {
  762. expandDataLocal.push({
  763. key: t('语音输入'),
  764. value: other.audio_input,
  765. });
  766. expandDataLocal.push({
  767. key: t('语音输出'),
  768. value: other.audio_output,
  769. });
  770. expandDataLocal.push({
  771. key: t('文字输入'),
  772. value: other.text_input,
  773. });
  774. expandDataLocal.push({
  775. key: t('文字输出'),
  776. value: other.text_output,
  777. });
  778. }
  779. if (other?.cache_tokens > 0) {
  780. expandDataLocal.push({
  781. key: t('缓存 Tokens'),
  782. value: other.cache_tokens,
  783. });
  784. }
  785. expandDataLocal.push({
  786. key: t('日志详情'),
  787. value: logs[i].content,
  788. });
  789. if (logs[i].type === 2) {
  790. let modelMapped = other?.is_model_mapped && other?.upstream_model_name && other?.upstream_model_name !== '';
  791. if (modelMapped) {
  792. expandDataLocal.push({
  793. key: t('请求并计费模型'),
  794. value: logs[i].model_name,
  795. });
  796. expandDataLocal.push({
  797. key: t('实际模型'),
  798. value: other.upstream_model_name,
  799. });
  800. }
  801. let content = '';
  802. if (other?.ws || other?.audio) {
  803. content = renderAudioModelPrice(
  804. other.text_input,
  805. other.text_output,
  806. other.model_ratio,
  807. other.model_price,
  808. other.completion_ratio,
  809. other.audio_input,
  810. other.audio_output,
  811. other?.audio_ratio,
  812. other?.audio_completion_ratio,
  813. other.group_ratio,
  814. other.cache_tokens || 0,
  815. other.cache_ratio || 1.0,
  816. );
  817. } else {
  818. content = renderModelPrice(
  819. logs[i].prompt_tokens,
  820. logs[i].completion_tokens,
  821. other.model_ratio,
  822. other.model_price,
  823. other.completion_ratio,
  824. other.group_ratio,
  825. other.cache_tokens || 0,
  826. other.cache_ratio || 1.0,
  827. );
  828. }
  829. expandDataLocal.push({
  830. key: t('计费过程'),
  831. value: content,
  832. });
  833. if (other?.reasoning_effort) {
  834. expandDataLocal.push({
  835. key: t('Reasoning Effort'),
  836. value: other.reasoning_effort,
  837. });
  838. }
  839. }
  840. expandDatesLocal[logs[i].key] = expandDataLocal;
  841. }
  842. setExpandData(expandDatesLocal);
  843. setLogs(logs);
  844. };
  845. const loadLogs = async (startIdx, pageSize, logType = 0) => {
  846. setLoading(true);
  847. let url = '';
  848. let localStartTimestamp = Date.parse(start_timestamp) / 1000;
  849. let localEndTimestamp = Date.parse(end_timestamp) / 1000;
  850. if (isAdminUser) {
  851. url = `/api/log/?p=${startIdx}&page_size=${pageSize}&type=${logType}&username=${username}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&channel=${channel}&group=${group}`;
  852. } else {
  853. url = `/api/log/self/?p=${startIdx}&page_size=${pageSize}&type=${logType}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&group=${group}`;
  854. }
  855. url = encodeURI(url);
  856. const res = await API.get(url);
  857. const { success, message, data } = res.data;
  858. if (success) {
  859. const newPageData = data.items;
  860. setActivePage(data.page);
  861. setPageSize(data.page_size);
  862. setLogCount(data.total);
  863. setLogsFormat(newPageData);
  864. } else {
  865. showError(message);
  866. }
  867. setLoading(false);
  868. };
  869. const handlePageChange = (page) => {
  870. setActivePage(page);
  871. loadLogs(page, pageSize, logType).then((r) => {});
  872. };
  873. const handlePageSizeChange = async (size) => {
  874. localStorage.setItem('page-size', size + '');
  875. setPageSize(size);
  876. setActivePage(1);
  877. loadLogs(activePage, size)
  878. .then()
  879. .catch((reason) => {
  880. showError(reason);
  881. });
  882. };
  883. const refresh = async () => {
  884. setActivePage(1);
  885. handleEyeClick();
  886. await loadLogs(activePage, pageSize, logType);
  887. };
  888. const copyText = async (e, text) => {
  889. e.stopPropagation();
  890. if (await copy(text)) {
  891. showSuccess('已复制:' + text);
  892. } else {
  893. Modal.error({ title: t('无法复制到剪贴板,请手动复制'), content: text });
  894. }
  895. };
  896. useEffect(() => {
  897. const localPageSize =
  898. parseInt(localStorage.getItem('page-size')) || ITEMS_PER_PAGE;
  899. setPageSize(localPageSize);
  900. loadLogs(activePage, localPageSize)
  901. .then()
  902. .catch((reason) => {
  903. showError(reason);
  904. });
  905. handleEyeClick();
  906. }, []);
  907. const expandRowRender = (record, index) => {
  908. return <Descriptions data={expandData[record.key]} />;
  909. };
  910. return (
  911. <>
  912. {renderColumnSelector()}
  913. <Layout>
  914. <Header style={{ backgroundColor: 'var(--semi-color-bg-1)' }}>
  915. <Spin spinning={loadingStat}>
  916. <Space>
  917. <Tag color='green' size='large' style={{ padding: 15 }}>
  918. {t('总消耗额度')}: {renderQuota(stat.quota)}
  919. </Tag>
  920. <Tag color='blue' size='large' style={{ padding: 15 }}>
  921. RPM: {stat.rpm}
  922. </Tag>
  923. <Tag color='purple' size='large' style={{ padding: 15 }}>
  924. TPM: {stat.tpm}
  925. </Tag>
  926. </Space>
  927. </Spin>
  928. </Header>
  929. <Form layout='horizontal' style={{ marginTop: 10 }}>
  930. <>
  931. <Form.Section>
  932. <div style={{ marginBottom: 10 }}>
  933. {
  934. styleState.isMobile ? (
  935. <div>
  936. <Form.DatePicker
  937. field='start_timestamp'
  938. label={t('起始时间')}
  939. style={{ width: 272 }}
  940. initValue={start_timestamp}
  941. type='dateTime'
  942. onChange={(value) => {
  943. console.log(value);
  944. handleInputChange(value, 'start_timestamp')
  945. }}
  946. />
  947. <Form.DatePicker
  948. field='end_timestamp'
  949. fluid
  950. label={t('结束时间')}
  951. style={{ width: 272 }}
  952. initValue={end_timestamp}
  953. type='dateTime'
  954. onChange={(value) => handleInputChange(value, 'end_timestamp')}
  955. />
  956. </div>
  957. ) : (
  958. <Form.DatePicker
  959. field="range_timestamp"
  960. label={t('时间范围')}
  961. initValue={[start_timestamp, end_timestamp]}
  962. type="dateTimeRange"
  963. name="range_timestamp"
  964. onChange={(value) => {
  965. if (Array.isArray(value) && value.length === 2) {
  966. handleInputChange(value[0], 'start_timestamp');
  967. handleInputChange(value[1], 'end_timestamp');
  968. }
  969. }}
  970. />
  971. )
  972. }
  973. </div>
  974. </Form.Section>
  975. <Form.Input
  976. field='token_name'
  977. label={t('令牌名称')}
  978. value={token_name}
  979. placeholder={t('可选值')}
  980. name='token_name'
  981. onChange={(value) => handleInputChange(value, 'token_name')}
  982. />
  983. <Form.Input
  984. field='model_name'
  985. label={t('模型名称')}
  986. value={model_name}
  987. placeholder={t('可选值')}
  988. name='model_name'
  989. onChange={(value) => handleInputChange(value, 'model_name')}
  990. />
  991. <Form.Input
  992. field='group'
  993. label={t('分组')}
  994. value={group}
  995. placeholder={t('可选值')}
  996. name='group'
  997. onChange={(value) => handleInputChange(value, 'group')}
  998. />
  999. {isAdminUser && (
  1000. <>
  1001. <Form.Input
  1002. field='channel'
  1003. label={t('渠道 ID')}
  1004. value={channel}
  1005. placeholder={t('可选值')}
  1006. name='channel'
  1007. onChange={(value) => handleInputChange(value, 'channel')}
  1008. />
  1009. <Form.Input
  1010. field='username'
  1011. label={t('用户名称')}
  1012. value={username}
  1013. placeholder={t('可选值')}
  1014. name='username'
  1015. onChange={(value) => handleInputChange(value, 'username')}
  1016. />
  1017. </>
  1018. )}
  1019. <Button
  1020. label={t('查询')}
  1021. type='primary'
  1022. htmlType='submit'
  1023. className='btn-margin-right'
  1024. onClick={refresh}
  1025. loading={loading}
  1026. style={{ marginTop: 24 }}
  1027. >
  1028. {t('查询')}
  1029. </Button>
  1030. <Form.Section></Form.Section>
  1031. </>
  1032. </Form>
  1033. <div style={{marginTop:10}}>
  1034. <Select
  1035. defaultValue='0'
  1036. style={{ width: 120 }}
  1037. onChange={(value) => {
  1038. setLogType(parseInt(value));
  1039. loadLogs(0, pageSize, parseInt(value));
  1040. }}
  1041. >
  1042. <Select.Option value='0'>{t('全部')}</Select.Option>
  1043. <Select.Option value='1'>{t('充值')}</Select.Option>
  1044. <Select.Option value='2'>{t('消费')}</Select.Option>
  1045. <Select.Option value='3'>{t('管理')}</Select.Option>
  1046. <Select.Option value='4'>{t('系统')}</Select.Option>
  1047. </Select>
  1048. <Button
  1049. theme='light'
  1050. type='tertiary'
  1051. icon={<IconSetting />}
  1052. onClick={() => setShowColumnSelector(true)}
  1053. style={{ marginLeft: 8 }}
  1054. >
  1055. {t('列设置')}
  1056. </Button>
  1057. </div>
  1058. <Table
  1059. style={{ marginTop: 5 }}
  1060. columns={getVisibleColumns()}
  1061. expandedRowRender={expandRowRender}
  1062. expandRowByClick={true}
  1063. dataSource={logs}
  1064. rowKey="key"
  1065. pagination={{
  1066. formatPageText: (page) =>
  1067. t('第 {{start}} - {{end}} 条,共 {{total}} 条', {
  1068. start: page.currentStart,
  1069. end: page.currentEnd,
  1070. total: logCount
  1071. }),
  1072. currentPage: activePage,
  1073. pageSize: pageSize,
  1074. total: logCount,
  1075. pageSizeOpts: [10, 20, 50, 100],
  1076. showSizeChanger: true,
  1077. onPageSizeChange: (size) => {
  1078. handlePageSizeChange(size);
  1079. },
  1080. onPageChange: handlePageChange,
  1081. }}
  1082. />
  1083. </Layout>
  1084. </>
  1085. );
  1086. };
  1087. export default LogsTable;