feat(auth-files): add codex websocket toggle in auth file editor

src/features/authFiles/components/AuthFileCard.tsx

@@ -2,14 +2,20 @@ import { useTranslation } from 'react-i18next';
 import { Button } from '@/components/ui/Button';
 import { LoadingSpinner } from '@/components/ui/LoadingSpinner';
 import { ToggleSwitch } from '@/components/ui/ToggleSwitch';
-import { IconBot, IconCheck, IconCode, IconDownload, IconInfo, IconTrash2 } from '@/components/ui/icons';
+import {
+  IconBot,
+  IconCheck,
+  IconCode,
+  IconDownload,
+  IconInfo,
+  IconTrash2,
+} from '@/components/ui/icons';
 import { ProviderStatusBar } from '@/components/providers/ProviderStatusBar';
 import type { AuthFileItem } from '@/types';
 import { resolveAuthProvider } from '@/utils/quota';
 import { calculateStatusBarData, normalizeAuthIndex, type KeyStats } from '@/utils/usage';
 import { formatFileSize } from '@/utils/format';
 import {
-  AUTH_FILE_REFRESH_WARNING_MS,
   QUOTA_PROVIDER_TYPES,
   formatModified,
   getTypeColor,
@@ -17,26 +23,13 @@ import {
   isRuntimeOnlyAuthFile,
   resolveAuthFileStats,
   type QuotaProviderType,
-  type ResolvedTheme
+  type ResolvedTheme,
 } from '@/features/authFiles/constants';
 import type { AuthFileStatusBarData } from '@/features/authFiles/hooks/useAuthFilesStatusBarCache';
 import { AuthFileQuotaSection } from '@/features/authFiles/components/AuthFileQuotaSection';
 import styles from '@/pages/AuthFilesPage.module.scss';
 
-type AuthFileHealthStatus = 'healthy' | 'warning' | 'disabled' | 'unknown';
-
 const HEALTHY_STATUS_MESSAGES = new Set(['ok', 'healthy', 'ready', 'success', 'available']);
-const GOOD_STATUS_VALUES = new Set(['', 'ok', 'ready', 'healthy', 'available']);
-
-const parseDateFromUnknown = (value: unknown): Date | null => {
-  if (value === null || value === undefined || value === '') return null;
-  const asNumber = Number(value);
-  const date =
-    Number.isFinite(asNumber) && !Number.isNaN(asNumber)
-      ? new Date(Math.abs(asNumber) < 1e12 ? asNumber * 1000 : asNumber)
-      : new Date(String(value));
-  return Number.isNaN(date.getTime()) ? null : date;
-};
 
 export type AuthFileCardProps = {
   file: AuthFileItem;
@@ -48,11 +41,10 @@ export type AuthFileCardProps = {
   quotaFilterType: QuotaProviderType | null;
   keyStats: KeyStats;
   statusBarCache: Map<string, AuthFileStatusBarData>;
-  nowMs: number;
   onShowModels: (file: AuthFileItem) => void;
   onShowDetails: (file: AuthFileItem) => void;
   onDownload: (name: string) => void;
-  onOpenPrefixProxyEditor: (name: string) => void;
+  onOpenPrefixProxyEditor: (file: AuthFileItem) => void;
   onDelete: (name: string) => void;
   onToggleStatus: (file: AuthFileItem, enabled: boolean) => void;
   onToggleSelect: (name: string) => void;
@@ -65,7 +57,7 @@ const resolveQuotaType = (file: AuthFileItem): QuotaProviderType | null => {
 };
 
 export function AuthFileCard(props: AuthFileCardProps) {
-  const { t, i18n } = useTranslation();
+  const { t } = useTranslation();
   const {
     file,
     selected,
@@ -76,14 +68,13 @@ export function AuthFileCard(props: AuthFileCardProps) {
     quotaFilterType,
     keyStats,
     statusBarCache,
-    nowMs,
     onShowModels,
     onShowDetails,
     onDownload,
     onOpenPrefixProxyEditor,
     onDelete,
     onToggleStatus,
-    onToggleSelect
+    onToggleSelect,
   } = props;
 
   const fileStats = resolveAuthFileStats(file, keyStats);
@@ -110,63 +101,9 @@ export function AuthFileCard(props: AuthFileCardProps) {
   const authIndexKey = normalizeAuthIndex(rawAuthIndex);
   const statusData =
     (authIndexKey && statusBarCache.get(authIndexKey)) || calculateStatusBarData([]);
-  const rawStatus = String(file.status ?? file['status'] ?? '')
-    .trim()
-    .toLowerCase();
   const rawStatusMessage = String(file['status_message'] ?? file.statusMessage ?? '').trim();
-  const normalizedStatusMessage = rawStatusMessage.toLowerCase();
-  const isFileDisabled = file.disabled === true || rawStatus === 'disabled';
-  const isUnavailable = file.unavailable === true || rawStatus === 'unavailable';
-  const lastRefreshDate = parseDateFromUnknown(file['last_refresh'] ?? file.lastRefresh);
-  const isRefreshStale = lastRefreshDate
-    ? nowMs - lastRefreshDate.getTime() > AUTH_FILE_REFRESH_WARNING_MS
-    : false;
   const hasStatusWarning =
-    Boolean(rawStatusMessage) && !HEALTHY_STATUS_MESSAGES.has(normalizedStatusMessage);
-  const hasStatusFailure = rawStatus === 'error' || rawStatus === 'failed' || rawStatus === 'warning';
-  const healthStatus: AuthFileHealthStatus = isFileDisabled
-    ? 'disabled'
-    : hasStatusWarning || hasStatusFailure || isUnavailable || isRefreshStale
-      ? 'warning'
-      : lastRefreshDate && !isRefreshStale && GOOD_STATUS_VALUES.has(rawStatus)
-        ? 'healthy'
-        : 'unknown';
-  const healthStatusClass =
-    healthStatus === 'healthy'
-      ? styles.healthStatusHealthy
-      : healthStatus === 'warning'
-        ? styles.healthStatusWarning
-        : healthStatus === 'disabled'
-          ? styles.healthStatusDisabled
-          : styles.healthStatusUnknown;
-  const healthStatusLabel = t(`auth_files.health_status_${healthStatus}`);
-  const lastRefreshText = (() => {
-    if (!lastRefreshDate) return t('auth_files.refresh_not_available');
-
-    const diffMs = lastRefreshDate.getTime() - nowMs;
-    const absMs = Math.abs(diffMs);
-    if (absMs < 30 * 1000) {
-      return t('auth_files.refresh_just_now');
-    }
-
-    const units: ReadonlyArray<{ unit: Intl.RelativeTimeFormatUnit; ms: number }> = [
-      { unit: 'day', ms: 24 * 60 * 60 * 1000 },
-      { unit: 'hour', ms: 60 * 60 * 1000 },
-      { unit: 'minute', ms: 60 * 1000 },
-      { unit: 'second', ms: 1000 }
-    ];
-    const matched = units.find(({ ms }) => absMs >= ms) || units[units.length - 1];
-    const value = Math.round(diffMs / matched.ms);
-    if (typeof Intl === 'undefined' || typeof Intl.RelativeTimeFormat !== 'function') {
-      return lastRefreshDate.toLocaleString(i18n.language);
-    }
-    const formatter = new Intl.RelativeTimeFormat(i18n.language, { numeric: 'auto' });
-    return formatter.format(value, matched.unit);
-  })();
-  const lastRefreshTitle = lastRefreshDate
-    ? lastRefreshDate.toLocaleString(i18n.language)
-    : t('auth_files.refresh_not_available');
-  const healthStatusTitle = rawStatusMessage || t('auth_files.health_status_no_message');
+    Boolean(rawStatusMessage) && !HEALTHY_STATUS_MESSAGES.has(rawStatusMessage.toLowerCase());
 
   return (
     <div
@@ -180,7 +117,9 @@ export function AuthFileCard(props: AuthFileCardProps) {
                 type="button"
                 className={`${styles.selectionToggle} ${selected ? styles.selectionToggleActive : ''}`}
                 onClick={() => onToggleSelect(file.name)}
-                aria-label={selected ? t('auth_files.batch_deselect') : t('auth_files.batch_select_all')}
+                aria-label={
+                  selected ? t('auth_files.batch_deselect') : t('auth_files.batch_select_all')
+                }
                 aria-pressed={selected}
                 title={selected ? t('auth_files.batch_deselect') : t('auth_files.batch_select_all')}
               >
@@ -192,7 +131,7 @@ export function AuthFileCard(props: AuthFileCardProps) {
               style={{
                 backgroundColor: typeColor.bg,
                 color: typeColor.text,
-                ...(typeColor.border ? { border: typeColor.border } : {})
+                ...(typeColor.border ? { border: typeColor.border } : {}),
               }}
             >
               {getTypeLabel(t, file.type || 'unknown')}
@@ -209,17 +148,6 @@ export function AuthFileCard(props: AuthFileCardProps) {
             </span>
           </div>
 
-          <div className={styles.cardHealthRow}>
-            <span className={`${styles.healthStatusBadge} ${healthStatusClass}`} title={healthStatusTitle}>
-              {t('auth_files.health_status_label')}: {healthStatusLabel}
-            </span>
-            <span
-              className={`${styles.lastRefreshText} ${isRefreshStale ? styles.lastRefreshStale : ''}`}
-              title={lastRefreshTitle}
-            >
-              {t('auth_files.last_refresh_label')}: {lastRefreshText}
-            </span>
-          </div>
           {rawStatusMessage && hasStatusWarning && (
             <div className={styles.healthStatusMessage} title={rawStatusMessage}>
               {rawStatusMessage}
@@ -238,7 +166,11 @@ export function AuthFileCard(props: AuthFileCardProps) {
           <ProviderStatusBar statusData={statusData} styles={styles} />
 
           {showQuotaLayout && quotaType && (
-            <AuthFileQuotaSection file={file} quotaType={quotaType} disableControls={disableControls} />
+            <AuthFileQuotaSection
+              file={file}
+              quotaType={quotaType}
+              disableControls={disableControls}
+            />
           )}
 
           <div className={styles.cardActions}>
@@ -279,7 +211,7 @@ export function AuthFileCard(props: AuthFileCardProps) {
                 <Button
                   variant="secondary"
                   size="sm"
-                  onClick={() => onOpenPrefixProxyEditor(file.name)}
+                  onClick={() => onOpenPrefixProxyEditor(file)}
                   className={styles.iconButton}
                   title={t('auth_files.prefix_proxy_button')}
                   disabled={disableControls}
@@ -313,7 +245,9 @@ export function AuthFileCard(props: AuthFileCardProps) {
               </div>
             )}
             {isRuntimeOnly && (
-              <div className={styles.virtualBadge}>{t('auth_files.type_virtual') || '虚拟认证文件'}</div>
+              <div className={styles.virtualBadge}>
+                {t('auth_files.type_virtual') || '虚拟认证文件'}
+              </div>
             )}
           </div>
         </div>

src/features/authFiles/components/AuthFilesPrefixProxyEditorModal.tsx

@@ -3,9 +3,11 @@ import { Modal } from '@/components/ui/Modal';
 import { Button } from '@/components/ui/Button';
 import { LoadingSpinner } from '@/components/ui/LoadingSpinner';
 import { Input } from '@/components/ui/Input';
+import { ToggleSwitch } from '@/components/ui/ToggleSwitch';
 import type {
   PrefixProxyEditorField,
-  PrefixProxyEditorState
+  PrefixProxyEditorFieldValue,
+  PrefixProxyEditorState,
 } from '@/features/authFiles/hooks/useAuthFilesPrefixProxyEditor';
 import styles from '@/pages/AuthFilesPage.module.scss';
 
@@ -16,7 +18,7 @@ export type AuthFilesPrefixProxyEditorModalProps = {
   dirty: boolean;
   onClose: () => void;
   onSave: () => void;
-  onChange: (field: PrefixProxyEditorField, value: string) => void;
+  onChange: (field: PrefixProxyEditorField, value: PrefixProxyEditorFieldValue) => void;
 };
 
 export function AuthFilesPrefixProxyEditorModal(props: AuthFilesPrefixProxyEditorModalProps) {
@@ -42,9 +44,7 @@ export function AuthFilesPrefixProxyEditorModal(props: AuthFilesPrefixProxyEdito
           <Button
             onClick={onSave}
             loading={editor?.saving === true}
-            disabled={
-              disableControls || editor?.saving === true || !dirty || !editor?.json
-            }
+            disabled={disableControls || editor?.saving === true || !dirty || !editor?.json}
           >
             {t('common.save')}
           </Button>
@@ -114,6 +114,18 @@ export function AuthFilesPrefixProxyEditorModal(props: AuthFilesPrefixProxyEdito
                   disabled={disableControls || editor.saving || !editor.json}
                   onChange={(e) => onChange('disableCooling', e.target.value)}
                 />
+                {editor.isCodexFile && (
+                  <div className="form-group">
+                    <label>{t('ai_providers.codex_websockets_label')}</label>
+                    <ToggleSwitch
+                      checked={Boolean(editor.websocket)}
+                      disabled={disableControls || editor.saving || !editor.json}
+                      ariaLabel={t('ai_providers.codex_websockets_label')}
+                      onChange={(value) => onChange('websocket', value)}
+                    />
+                    <div className="hint">{t('ai_providers.codex_websockets_hint')}</div>
+                  </div>
+                )}
               </div>
             </>
           )}
@@ -122,4 +134,3 @@ export function AuthFilesPrefixProxyEditorModal(props: AuthFilesPrefixProxyEdito
     </Modal>
   );
 }
-

src/features/authFiles/hooks/useAuthFilesPrefixProxyEditor.ts

@@ -1,6 +1,7 @@
 import { useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { authFilesApi } from '@/services/api';
+import type { AuthFileItem } from '@/types';
 import { useNotificationStore } from '@/stores';
 import { formatFileSize } from '@/utils/format';
 import { MAX_AUTH_FILE_SIZE } from '@/utils/constants';
@@ -8,7 +9,7 @@ import {
   normalizeExcludedModels,
   parseDisableCoolingValue,
   parseExcludedModelsText,
-  parsePriorityValue
+  parsePriorityValue,
 } from '@/features/authFiles/constants';
 
 export type PrefixProxyEditorField =
@@ -16,10 +17,14 @@ export type PrefixProxyEditorField =
   | 'proxyUrl'
   | 'priority'
   | 'excludedModelsText'
-  | 'disableCooling';
+  | 'disableCooling'
+  | 'websocket';
+
+export type PrefixProxyEditorFieldValue = string | boolean;
 
 export type PrefixProxyEditorState = {
   fileName: string;
+  isCodexFile: boolean;
   loading: boolean;
   saving: boolean;
   error: string | null;
@@ -31,6 +36,7 @@ export type PrefixProxyEditorState = {
   priority: string;
   excludedModelsText: string;
   disableCooling: string;
+  websocket: boolean;
 };
 
 export type UseAuthFilesPrefixProxyEditorOptions = {
@@ -43,9 +49,12 @@ export type UseAuthFilesPrefixProxyEditorResult = {
   prefixProxyEditor: PrefixProxyEditorState | null;
   prefixProxyUpdatedText: string;
   prefixProxyDirty: boolean;
-  openPrefixProxyEditor: (name: string) => Promise<void>;
+  openPrefixProxyEditor: (file: Pick<AuthFileItem, 'name' | 'type' | 'provider'>) => Promise<void>;
   closePrefixProxyEditor: () => void;
-  handlePrefixProxyChange: (field: PrefixProxyEditorField, value: string) => void;
+  handlePrefixProxyChange: (
+    field: PrefixProxyEditorField,
+    value: PrefixProxyEditorFieldValue
+  ) => void;
   handlePrefixProxySave: () => Promise<void>;
 };
 
@@ -80,6 +89,10 @@ const buildPrefixProxyUpdatedText = (editor: PrefixProxyEditorState | null): str
     delete next.disable_cooling;
   }
 
+  if (editor.isCodexFile) {
+    next.websocket = editor.websocket;
+  }
+
   return JSON.stringify(next);
 };
 
@@ -102,7 +115,16 @@ export function useAuthFilesPrefixProxyEditor(
     setPrefixProxyEditor(null);
   };
 
-  const openPrefixProxyEditor = async (name: string) => {
+  const openPrefixProxyEditor = async (file: Pick<AuthFileItem, 'name' | 'type' | 'provider'>) => {
+    const name = file.name;
+    const normalizedType = String(file.type ?? '')
+      .trim()
+      .toLowerCase();
+    const normalizedProvider = String(file.provider ?? '')
+      .trim()
+      .toLowerCase();
+    const isCodexFile = normalizedType === 'codex' || normalizedProvider === 'codex';
+
     if (disableControls) return;
     if (prefixProxyEditor?.fileName === name) {
       setPrefixProxyEditor(null);
@@ -111,6 +133,7 @@ export function useAuthFilesPrefixProxyEditor(
 
     setPrefixProxyEditor({
       fileName: name,
+      isCodexFile,
       loading: true,
       saving: false,
       error: null,
@@ -121,7 +144,8 @@ export function useAuthFilesPrefixProxyEditor(
       proxyUrl: '',
       priority: '',
       excludedModelsText: '',
-      disableCooling: ''
+      disableCooling: '',
+      websocket: false,
     });
 
     try {
@@ -139,7 +163,7 @@ export function useAuthFilesPrefixProxyEditor(
             loading: false,
             error: t('auth_files.prefix_proxy_invalid_json'),
             rawText: trimmed,
-            originalText: trimmed
+            originalText: trimmed,
           };
         });
         return;
@@ -153,19 +177,24 @@ export function useAuthFilesPrefixProxyEditor(
             loading: false,
             error: t('auth_files.prefix_proxy_invalid_json'),
             rawText: trimmed,
-            originalText: trimmed
+            originalText: trimmed,
           };
         });
         return;
       }
 
-      const json = parsed as Record<string, unknown>;
+      const json = { ...(parsed as Record<string, unknown>) };
+      if (isCodexFile) {
+        const websocketValue = parseDisableCoolingValue(json.websocket);
+        json.websocket = websocketValue ?? false;
+      }
       const originalText = JSON.stringify(json);
       const prefix = typeof json.prefix === 'string' ? json.prefix : '';
       const proxyUrl = typeof json.proxy_url === 'string' ? json.proxy_url : '';
       const priority = parsePriorityValue(json.priority);
       const excludedModels = normalizeExcludedModels(json.excluded_models);
       const disableCoolingValue = parseDisableCoolingValue(json.disable_cooling);
+      const websocketValue = parseDisableCoolingValue(json.websocket);
 
       setPrefixProxyEditor((prev) => {
         if (!prev || prev.fileName !== name) return prev;
@@ -181,7 +210,8 @@ export function useAuthFilesPrefixProxyEditor(
           excludedModelsText: excludedModels.join('\n'),
           disableCooling:
             disableCoolingValue === undefined ? '' : disableCoolingValue ? 'true' : 'false',
-          error: null
+          websocket: websocketValue ?? false,
+          error: null,
         };
       });
     } catch (err: unknown) {
@@ -194,14 +224,18 @@ export function useAuthFilesPrefixProxyEditor(
     }
   };
 
-  const handlePrefixProxyChange = (field: PrefixProxyEditorField, value: string) => {
+  const handlePrefixProxyChange = (
+    field: PrefixProxyEditorField,
+    value: PrefixProxyEditorFieldValue
+  ) => {
     setPrefixProxyEditor((prev) => {
       if (!prev) return prev;
-      if (field === 'prefix') return { ...prev, prefix: value };
-      if (field === 'proxyUrl') return { ...prev, proxyUrl: value };
-      if (field === 'priority') return { ...prev, priority: value };
-      if (field === 'excludedModelsText') return { ...prev, excludedModelsText: value };
-      return { ...prev, disableCooling: value };
+      if (field === 'prefix') return { ...prev, prefix: String(value) };
+      if (field === 'proxyUrl') return { ...prev, proxyUrl: String(value) };
+      if (field === 'priority') return { ...prev, priority: String(value) };
+      if (field === 'excludedModelsText') return { ...prev, excludedModelsText: String(value) };
+      if (field === 'disableCooling') return { ...prev, disableCooling: String(value) };
+      return { ...prev, websocket: Boolean(value) };
     });
   };
 
@@ -249,6 +283,6 @@ export function useAuthFilesPrefixProxyEditor(
     openPrefixProxyEditor,
     closePrefixProxyEditor,
     handlePrefixProxyChange,
-    handlePrefixProxySave
+    handlePrefixProxySave,
   };
 }

src/i18n/locales/en.json

@@ -1023,6 +1023,7 @@
     "trace_confidence_low": "Low",
     "trace_score": "Score {{score}}",
     "trace_delta_seconds": "Δt {{seconds}}s",
+    "trace_model_matched": "Model Matched",
     "trace_request_id": "Request ID",
     "trace_method": "Method",
     "trace_path": "Path",

src/i18n/locales/ru.json

@@ -1026,6 +1026,7 @@
     "trace_confidence_low": "Низкая",
     "trace_score": "Оценка {{score}}",
     "trace_delta_seconds": "Δt {{seconds}}с",
+    "trace_model_matched": "Модель совпала",
     "trace_request_id": "Request ID",
     "trace_method": "Метод",
     "trace_path": "Путь",

src/i18n/locales/zh-CN.json

@@ -1023,6 +1023,7 @@
     "trace_confidence_low": "低",
     "trace_score": "分数 {{score}}",
     "trace_delta_seconds": "时间差 {{seconds}} 秒",
+    "trace_model_matched": "模型匹配",
     "trace_request_id": "请求 ID",
     "trace_method": "请求方法",
     "trace_path": "路径",

src/pages/AuthFilesPage.module.scss

@@ -605,59 +605,6 @@
   border-bottom: 1px solid var(--border-color);
 }
 
-.cardHealthRow {
-  display: flex;
-  align-items: center;
-  justify-content: space-between;
-  gap: $spacing-sm;
-  flex-wrap: wrap;
-}
-
-.healthStatusBadge {
-  display: inline-flex;
-  align-items: center;
-  gap: 6px;
-  padding: 5px 10px;
-  border-radius: $radius-full;
-  border: 1px solid var(--border-color);
-  font-size: 12px;
-  font-weight: 600;
-  line-height: 1.2;
-}
-
-.healthStatusHealthy {
-  color: var(--success-badge-text, #065f46);
-  background-color: var(--success-badge-bg, #d1fae5);
-  border-color: var(--success-badge-border, #6ee7b7);
-}
-
-.healthStatusWarning {
-  color: var(--warning-text);
-  background-color: var(--warning-bg);
-  border-color: var(--warning-border);
-}
-
-.healthStatusDisabled {
-  color: var(--text-secondary);
-  background-color: var(--bg-tertiary);
-  border-color: var(--border-color);
-}
-
-.healthStatusUnknown {
-  color: var(--text-secondary);
-  background-color: var(--bg-secondary);
-  border-style: dashed;
-}
-
-.lastRefreshText {
-  font-size: 12px;
-  color: var(--text-secondary);
-}
-
-.lastRefreshStale {
-  color: var(--warning-text);
-}
-
 .healthStatusMessage {
   font-size: 12px;
   color: var(--warning-text);

src/pages/AuthFilesPage.tsx

@@ -58,7 +58,6 @@ export function AuthFilesPage() {
   const [selectedFile, setSelectedFile] = useState<AuthFileItem | null>(null);
   const [viewMode, setViewMode] = useState<'diagram' | 'list'>('list');
   const [batchActionBarVisible, setBatchActionBarVisible] = useState(false);
-  const [nowMs, setNowMs] = useState(() => Date.now());
   const floatingBatchActionsRef = useRef<HTMLDivElement>(null);
   const previousSelectionCountRef = useRef(0);
   const selectionCountRef = useRef(0);
@@ -223,7 +222,6 @@ export function AuthFilesPage() {
     },
     isCurrentLayer ? 240_000 : null
   );
-  useInterval(() => setNowMs(Date.now()), isCurrentLayer ? 60_000 : null);
 
   const existingTypes = useMemo(() => {
     const types = new Set<string>(['all']);
@@ -519,7 +517,6 @@ export function AuthFilesPage() {
                 quotaFilterType={quotaFilterType}
                 keyStats={keyStats}
                 statusBarCache={statusBarCache}
-                nowMs={nowMs}
                 onShowModels={showModels}
                 onShowDetails={showDetails}
                 onDownload={handleDownload}

src/pages/LogsPage.module.scss

@@ -693,34 +693,18 @@
   flex-wrap: wrap;
 }
 
-.traceConfidenceBadge {
+.traceModelBadge {
   display: inline-flex;
   align-items: center;
   padding: 2px 8px;
   border-radius: $radius-full;
-  border: 1px solid var(--border-color);
+  border: 1px solid var(--success-badge-border, #6ee7b7);
   font-size: 11px;
   font-weight: 700;
-}
-
-.traceConfidenceHigh {
   color: var(--success-badge-text, #065f46);
   background: var(--success-badge-bg, #d1fae5);
-  border-color: var(--success-badge-border, #6ee7b7);
 }
 
-.traceConfidenceMedium {
-  color: var(--warning-text);
-  background: var(--warning-bg);
-  border-color: var(--warning-border);
-}
-
-.traceConfidenceLow {
-  color: var(--text-secondary);
-  background: var(--bg-primary);
-}
-
-.traceScore,
 .traceDelta {
   font-size: 11px;
   color: var(--text-secondary);

src/pages/LogsPage.tsx

@@ -970,12 +970,6 @@ export function LogsPage() {
             ) : (
               <div className={styles.traceCandidates}>
                 {trace.traceCandidates.map((candidate) => {
-                  const confidenceClass =
-                    candidate.confidence === 'high'
-                      ? styles.traceConfidenceHigh
-                      : candidate.confidence === 'medium'
-                        ? styles.traceConfidenceMedium
-                        : styles.traceConfidenceLow;
                   const sourceInfo = trace.resolveTraceSourceInfo(
                     String(candidate.detail.source ?? ''),
                     candidate.detail.auth_index
@@ -986,12 +980,11 @@ export function LogsPage() {
                       className={styles.traceCandidate}
                     >
                       <div className={styles.traceCandidateHeader}>
-                        <span className={`${styles.traceConfidenceBadge} ${confidenceClass}`}>
-                          {t(`logs.trace_confidence_${candidate.confidence}`)}
-                        </span>
-                        <span className={styles.traceScore}>
-                          {t('logs.trace_score', { score: candidate.score })}
-                        </span>
+                        {candidate.modelMatched && (
+                          <span className={styles.traceModelBadge}>
+                            {t('logs.trace_model_matched')}
+                          </span>
+                        )}
                         {candidate.timeDeltaMs !== null && (
                           <span className={styles.traceDelta}>
                             {t('logs.trace_delta_seconds', {

src/pages/hooks/useTraceResolver.ts

@@ -12,19 +12,14 @@ import {
 } from '@/utils/usage';
 import type { ParsedLogLine } from './logTypes';
 
-type TraceConfidence = 'high' | 'medium' | 'low';
-
 export type TraceCandidate = {
   detail: UsageDetailWithEndpoint;
-  score: number;
-  confidence: TraceConfidence;
+  modelMatched: boolean;
   timeDeltaMs: number | null;
 };
 
 const TRACE_AUTH_CACHE_MS = 60 * 1000;
-const TRACE_MATCH_STRONG_WINDOW_MS = 3 * 1000;
-const TRACE_MATCH_WINDOW_MS = 10 * 1000;
-const TRACE_MATCH_MAX_WINDOW_MS = 30 * 1000;
+const TRACE_MAX_CANDIDATES = 5;
 
 const TRACEABLE_EXACT_PATHS = new Set(['/v1/chat/completions', '/v1/messages', '/v1/responses']);
 const TRACEABLE_PREFIX_PATHS = ['/v1beta/models'];
@@ -48,70 +43,17 @@ export const isTraceableRequestPath = (value?: string): boolean => {
   return TRACEABLE_PREFIX_PATHS.some((prefix) => normalizedPath.startsWith(prefix));
 };
 
-const scoreTraceCandidate = (
-  line: ParsedLogLine,
-  detail: UsageDetailWithEndpoint
-): TraceCandidate | null => {
-  let score = 0;
-  let timeDeltaMs: number | null = null;
+const MODEL_EXTRACT_REGEX = /\bmodel[=:]\s*"?([a-zA-Z0-9._\-/]+)"?/i;
 
-  const logTimestampMs = line.timestamp ? Date.parse(line.timestamp) : Number.NaN;
-  const detailTimestampMs = detail.__timestampMs;
-  if (!Number.isNaN(logTimestampMs) && detailTimestampMs > 0) {
-    timeDeltaMs = Math.abs(logTimestampMs - detailTimestampMs);
-    if (timeDeltaMs <= TRACE_MATCH_STRONG_WINDOW_MS) {
-      score += 42;
-    } else if (timeDeltaMs <= TRACE_MATCH_WINDOW_MS) {
-      score += 30;
-    } else if (timeDeltaMs <= TRACE_MATCH_MAX_WINDOW_MS) {
-      score += 12;
-    } else {
-      score -= 12;
-    }
-  }
+const extractModelFromMessage = (message?: string): string | undefined => {
+  if (!message) return undefined;
+  const match = message.match(MODEL_EXTRACT_REGEX);
+  return match?.[1] || undefined;
+};
 
-  let methodMatched = false;
-  if (line.method && detail.__endpointMethod) {
-    if (line.method.toUpperCase() === detail.__endpointMethod.toUpperCase()) {
-      score += 18;
-      methodMatched = true;
-    } else {
-      score -= 8;
-    }
-  }
-
-  const logPath = normalizeTracePath(line.path);
-  const detailPath = normalizeTracePath(detail.__endpointPath);
-  let pathMatched = false;
-  if (logPath && detailPath) {
-    if (logPath === detailPath) {
-      score += 24;
-      pathMatched = true;
-    } else if (logPath.startsWith(detailPath) || detailPath.startsWith(logPath)) {
-      score += 12;
-      pathMatched = true;
-    } else {
-      score -= 8;
-    }
-  }
-
-  if (typeof line.statusCode === 'number') {
-    const logFailed = line.statusCode >= 400;
-    score += logFailed === detail.failed ? 10 : -6;
-  }
-
-  if (
-    timeDeltaMs !== null &&
-    timeDeltaMs > TRACE_MATCH_MAX_WINDOW_MS &&
-    !methodMatched &&
-    !pathMatched
-  ) {
-    return null;
-  }
-
-  if (score <= 0) return null;
-  const confidence: TraceConfidence = score >= 70 ? 'high' : score >= 45 ? 'medium' : 'low';
-  return { detail, score, confidence, timeDeltaMs };
+const isPathMatch = (logPath: string, detailPath: string): boolean => {
+  if (!logPath || !detailPath) return false;
+  return logPath === detailPath || logPath.startsWith(detailPath) || detailPath.startsWith(logPath);
 };
 
 const getErrorMessage = (err: unknown): string => {
@@ -236,16 +178,42 @@ export function useTraceResolver(options: UseTraceResolverOptions): UseTraceReso
 
   const traceCandidates = useMemo(() => {
     if (!traceLogLine) return [];
-    const scored = traceUsageDetails
-      .map((detail) => scoreTraceCandidate(traceLogLine, detail))
-      .filter((item): item is TraceCandidate => item !== null)
-      .sort((a, b) => {
-        if (b.score !== a.score) return b.score - a.score;
-        const aDelta = a.timeDeltaMs ?? Number.MAX_SAFE_INTEGER;
-        const bDelta = b.timeDeltaMs ?? Number.MAX_SAFE_INTEGER;
-        return aDelta - bDelta;
-      });
-    return scored.slice(0, 8);
+
+    const logPath = normalizeTracePath(traceLogLine.path);
+    if (!logPath) return [];
+
+    const logTimestampMs = traceLogLine.timestamp
+      ? Date.parse(traceLogLine.timestamp)
+      : Number.NaN;
+
+    // Step 1: filter by path match
+    const pathMatched = traceUsageDetails.filter((detail) =>
+      isPathMatch(logPath, normalizeTracePath(detail.__endpointPath))
+    );
+    if (pathMatched.length === 0) return [];
+
+    // Step 2: try to extract model from log message, then filter by model
+    const logModel = extractModelFromMessage(traceLogLine.message);
+    const modelMatched = logModel
+      ? pathMatched.filter(
+          (d) => d.__modelName?.toLowerCase() === logModel.toLowerCase()
+        )
+      : [];
+
+    // Step 3: prefer model-matched set; fall back to path-matched
+    const useModelSet = modelMatched.length > 0;
+    const source = useModelSet ? modelMatched : pathMatched;
+
+    return source
+      .map((detail) => {
+        const timeDeltaMs =
+          !Number.isNaN(logTimestampMs) && detail.__timestampMs > 0
+            ? Math.abs(logTimestampMs - detail.__timestampMs)
+            : null;
+        return { detail, modelMatched: useModelSet, timeDeltaMs } satisfies TraceCandidate;
+      })
+      .sort((a, b) => (b.detail.__timestampMs || 0) - (a.detail.__timestampMs || 0))
+      .slice(0, TRACE_MAX_CANDIDATES);
   }, [traceLogLine, traceUsageDetails]);
 
   const resolveTraceSourceInfo = useCallback(