simplify local

BerriAI · Nov 23, 2024 · 77fe5af · 77fe5af
1 parent 35040f1
commit 77fe5af
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/pass_through_tests/test_local_vertex.js b/tests/pass_through_tests/test_local_vertex.js
@@ -20,14 +20,12 @@ const vertexAI = new VertexAI({
     apiEndpoint: "localhost:4000/vertex-ai"
 });
 
-// Create customHeaders using Headers
-const customHeaders = new Headers({
-    "X-Litellm-Api-Key": "sk-1234"
-});
 
 // Use customHeaders in RequestOptions
 const requestOptions = {
-    customHeaders: customHeaders
+    customHeaders: new Headers({
+        "X-Litellm-Api-Key": "sk-1234"
+    })
 };
 
 const generativeModel = vertexAI.getGenerativeModel(