Spaces:
Running
Running
fix: update model fetching to exclude specific model ID and improve data handling
Browse files- components/message.tsx +1 -1
- lib/models.ts +4 -1
components/message.tsx
CHANGED
@@ -244,7 +244,7 @@ const PurePreviewMessage = ({
|
|
244 |
return (
|
245 |
<div
|
246 |
className={cn(
|
247 |
-
"w-full mx-auto px-4 group/message",
|
248 |
message.role === "assistant" ? "mb-8" : "mb-6"
|
249 |
)}
|
250 |
data-role={message.role}
|
|
|
244 |
return (
|
245 |
<div
|
246 |
className={cn(
|
247 |
+
"w-full mx-auto sm:px-4 group/message",
|
248 |
message.role === "assistant" ? "mb-8" : "mb-6"
|
249 |
)}
|
250 |
data-role={message.role}
|
lib/models.ts
CHANGED
@@ -13,7 +13,10 @@ export async function getModels(): Promise<string[]> {
|
|
13 |
try {
|
14 |
const response = await fetch("https://router.huggingface.co/v1/models");
|
15 |
const data = await response.json();
|
16 |
-
const modelIds = data.data
|
|
|
|
|
|
|
17 |
modelsCache = modelIds;
|
18 |
return modelIds;
|
19 |
} catch (e) {
|
|
|
13 |
try {
|
14 |
const response = await fetch("https://router.huggingface.co/v1/models");
|
15 |
const data = await response.json();
|
16 |
+
const modelIds = data.data
|
17 |
+
.filter((model: any) => model.id !== "THUDM/GLM-4.1V-9B-Thinking")
|
18 |
+
.slice(0, 5)
|
19 |
+
.map((model: any) => model.id);
|
20 |
modelsCache = modelIds;
|
21 |
return modelIds;
|
22 |
} catch (e) {
|