ausername-12345 commited on
Commit
24129da
·
1 Parent(s): 6a1875d

Add more HF endpoint tests

Browse files
Files changed (1) hide show
  1. server.js +7 -1
server.js CHANGED
@@ -115,15 +115,21 @@ app.get("/api/test-network", async (_, res) => {
115
  results[host] = { dns_error: e.code || e.message };
116
  }
117
  }
 
118
  for (const url of [
119
  "https://google.com",
120
  "https://huggingface.co",
 
121
  "https://huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
122
  "https://router.huggingface.co",
123
  "https://api-inference.huggingface.co",
124
  ]) {
125
  try {
126
- const r = await fetchWithTimeout(url, { redirect: "manual" }, 5000);
 
 
 
 
127
  results[url] = { status: r.status, ok: r.ok };
128
  } catch (e) {
129
  results[url] = { error: e.message || String(e), code: e.cause?.code || e.code };
 
115
  results[host] = { dns_error: e.code || e.message };
116
  }
117
  }
118
+ const token = process.env.HF_TOKEN;
119
  for (const url of [
120
  "https://google.com",
121
  "https://huggingface.co",
122
+ "https://huggingface.co/api/inference",
123
  "https://huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
124
  "https://router.huggingface.co",
125
  "https://api-inference.huggingface.co",
126
  ]) {
127
  try {
128
+ const headers = { "User-Agent": "node" };
129
+ if (token && (url.includes("huggingface.co/api/") || url.includes("api-inference"))) {
130
+ headers.Authorization = `Bearer ${token}`;
131
+ }
132
+ const r = await fetchWithTimeout(url, { redirect: "manual", headers }, 5000);
133
  results[url] = { status: r.status, ok: r.ok };
134
  } catch (e) {
135
  results[url] = { error: e.message || String(e), code: e.cause?.code || e.code };