victor HF Staff commited on
Commit
268c9a8
·
1 Parent(s): c8afc3c

fix: update model fetching to exclude specific model ID and improve data handling

Browse files
Files changed (2) hide show
  1. components/message.tsx +1 -1
  2. 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.slice(0, 5).map((model: any) => model.id);
 
 
 
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) {