Update app.js

app.js

@@ -63,81 +63,102 @@ function getLocation() {
 }
 
 function constructApiUrl(location) {
-  return `https://${location}-aiplatform.googleapis.com/v1/projects/${PROJECT_ID}/locations/${location}/publishers/anthropic/models/${MODEL}:streamRawPredict`;
+  const url = `https://${location}-aiplatform.googleapis.com/v1/projects/${PROJECT_ID}/locations/${location}/publishers/anthropic/models/${MODEL}:streamRawPredict`;
+  console.log('Constructed URL:', url); // 打印URL以验证其正确性
+  return url;
 }
 
 async function handleRequest(request) {
-  if (request.method === 'OPTIONS') {
-    return handleOptions();
-  }
+  try {
+    if (request.method === 'OPTIONS') {
+      return handleOptions();
+    }
 
-  const apiKey = request.headers['x-api-key'];
-  if (apiKey !== API_KEY) {
-    const errorResponse = {
-      status: 403,
-      headers: {
-        'Content-Type': 'application/json',
-        'Access-Control-Allow-Origin': '*',
-        'Access-Control-Allow-Methods': 'POST, GET, OPTIONS, DELETE, HEAD',
-        'Access-Control-Allow-Headers': 'Content-Type, Authorization, x-api-key, anthropic-version, model'
-      },
-      body: JSON.stringify({
+    const apiKey = request.headers['x-api-key'];
+    if (apiKey !== API_KEY) {
+      return handleErrorResponse(403, {
         type: "error",
         error: {
           type: "permission_error",
           message: "Your API key does not have permission to use the specified resource."
         }
-      })
-    };
+      });
+    }
 
-    return errorResponse;
-  }
+    const accessToken = await getAccessToken();
+    const location = getLocation();
+    const apiUrl = constructApiUrl(location);
 
-  const accessToken = await getAccessToken();
-  const location = getLocation();
-  const apiUrl = constructApiUrl(location);
+    let requestBody = JSON.parse(request.body);
+    
+    if (requestBody.anthropic_version) {
+      delete requestBody.anthropic_version;
+    }
+    
+    if (requestBody.model) {
+      delete requestBody.model;
+    }
+    
+    requestBody.anthropic_version = "vertex-2023-10-16";
 
-  let requestBody = JSON.parse(request.body);
-  
-  if (requestBody.anthropic_version) {
-    delete requestBody.anthropic_version;
-  }
-  
-  if (requestBody.model) {
-    delete requestBody.model;
-  }
-  
-  requestBody.anthropic_version = "vertex-2023-10-16";
+    const modifiedHeaders = {
+      ...request.headers,
+      'Authorization': `Bearer ${accessToken}`,
+      'Content-Type': 'application/json; charset=utf-8'
+    };
+    delete modifiedHeaders['anthropic-version'];
 
-  const modifiedHeaders = {
-    ...request.headers,
-    'Authorization': `Bearer ${accessToken}`,
-    'Content-Type': 'application/json; charset=utf-8'
-  };
-  delete modifiedHeaders['anthropic-version'];
+    const modifiedRequest = {
+      headers: modifiedHeaders,
+      method: request.method,
+      body: JSON.stringify(requestBody),
+      redirect: 'manual'
+    };
 
-  const modifiedRequest = {
-    headers: modifiedHeaders,
-    method: request.method,
-    body: JSON.stringify(requestBody),
-    redirect: 'follow'
-  };
+    const response = await fetch(apiUrl, modifiedRequest);
+    if (response.status === 301 || response.status === 302) {
+      const location = response.headers.get('location');
+      console.log('Redirected to:', location);
+      const redirectedResponse = await fetch(location, modifiedRequest);
+      return handleResponse(redirectedResponse);
+    }
+    return handleResponse(response);
+  } catch (error) {
+    console.error('Error handling request:', error);
+    return handleErrorResponse(500, {
+      type: "error",
+      error: {
+        type: "internal_error",
+        message: "Internal server error."
+      }
+    });
+  }
+}
 
-  const response = await fetch(apiUrl, modifiedRequest);
-  const responseBody = await response.text();
-  const modifiedResponse = {
+function handleResponse(response) {
+  return {
     status: response.status,
-    statusText: response.statusText,
     headers: {
       ...response.headers,
       'Access-Control-Allow-Origin': '*',
       'Access-Control-Allow-Methods': 'POST, GET, OPTIONS',
       'Access-Control-Allow-Headers': 'Content-Type, Authorization, x-api-key, anthropic-version, model'
     },
-    body: responseBody
+    body: response.body
+  };
+}
+
+function handleErrorResponse(status, body) {
+  return {
+    status: status,
+    headers: {
+      'Content-Type': 'application/json',
+      'Access-Control-Allow-Origin': '*',
+      'Access-Control-Allow-Methods': 'POST, GET, OPTIONS, DELETE, HEAD',
+      'Access-Control-Allow-Headers': 'Content-Type, Authorization, x-api-key, anthropic-version, model'
+    },
+    body: JSON.stringify(body)
   };
-   
-  return modifiedResponse;
 }
 
 function handleOptions() {
@@ -163,7 +184,7 @@ const server = http.createServer(async (req, res) => {
     };
 
     const response = await handleRequest(request);
-    res.writeHead(response.status, response.statusText, response.headers);
+    res.writeHead(response.status, response.headers);
     res.end(response.body);
   } else {
     res.writeHead(404, { 'Content-Type': 'application/json' });