tfrere HF Staff Cursor commited on
Commit
b160a21
·
1 Parent(s): d745ea8

Fix: model selector dropdown bg, get_max_tokens None fallback

Browse files
agent/core/session.py CHANGED
@@ -36,7 +36,11 @@ def _get_max_tokens_safe(model_name: str) -> int:
36
  # Fallback: try litellm but with a short timeout via threading
37
  try:
38
  from litellm import get_max_tokens
39
- return get_max_tokens(model_name)
 
 
 
 
40
  except Exception as e:
41
  logger.warning(f"get_max_tokens failed for {model_name}, using default: {e}")
42
  return _DEFAULT_MAX_TOKENS
 
36
  # Fallback: try litellm but with a short timeout via threading
37
  try:
38
  from litellm import get_max_tokens
39
+ result = get_max_tokens(model_name)
40
+ if result and isinstance(result, int):
41
+ return result
42
+ logger.warning(f"get_max_tokens returned {result} for {model_name}, using default")
43
+ return _DEFAULT_MAX_TOKENS
44
  except Exception as e:
45
  logger.warning(f"get_max_tokens failed for {model_name}, using default: {e}")
46
  return _DEFAULT_MAX_TOKENS
frontend/src/components/Layout/AppLayout.tsx CHANGED
@@ -335,15 +335,20 @@ export default function AppLayout() {
335
 
336
  <Box sx={{ display: 'flex', alignItems: 'center', gap: 0.5 }}>
337
  {/* Model selector */}
338
- {availableModels.length > 0 && (
339
  <Select
340
  value={currentModel}
341
  onChange={(e) => handleModelChange(e.target.value)}
342
  size="small"
343
  variant="outlined"
 
 
 
 
344
  sx={{
345
  fontSize: '0.72rem',
346
  height: 30,
 
347
  color: 'var(--muted-text)',
348
  '& .MuiOutlinedInput-notchedOutline': {
349
  borderColor: 'var(--border)',
 
335
 
336
  <Box sx={{ display: 'flex', alignItems: 'center', gap: 0.5 }}>
337
  {/* Model selector */}
338
+ {availableModels.length > 0 && currentModel && (
339
  <Select
340
  value={currentModel}
341
  onChange={(e) => handleModelChange(e.target.value)}
342
  size="small"
343
  variant="outlined"
344
+ renderValue={(val) => {
345
+ const m = availableModels.find((x) => x.id === val);
346
+ return m?.label || val;
347
+ }}
348
  sx={{
349
  fontSize: '0.72rem',
350
  height: 30,
351
+ minWidth: 120,
352
  color: 'var(--muted-text)',
353
  '& .MuiOutlinedInput-notchedOutline': {
354
  borderColor: 'var(--border)',
frontend/src/theme.ts CHANGED
@@ -24,7 +24,7 @@ const sharedComponents: ThemeOptions['components'] = {
24
  },
25
  MuiPaper: {
26
  styleOverrides: {
27
- root: { backgroundImage: 'none', backgroundColor: 'transparent' },
28
  },
29
  },
30
  };
 
24
  },
25
  MuiPaper: {
26
  styleOverrides: {
27
+ root: { backgroundImage: 'none' },
28
  },
29
  },
30
  };