Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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'
|
| 28 |
},
|
| 29 |
},
|
| 30 |
};
|
|
|
|
| 24 |
},
|
| 25 |
MuiPaper: {
|
| 26 |
styleOverrides: {
|
| 27 |
+
root: { backgroundImage: 'none' },
|
| 28 |
},
|
| 29 |
},
|
| 30 |
};
|