diff --git a/samples/README.md b/samples/README.md
index 0c37e0efa..1411f86e9 100644
--- a/samples/README.md
+++ b/samples/README.md
@@ -49,3 +49,4 @@ to comment out cases you do not want to run.
 | [search_grounding.js](./search_grounding.js)             | Generate with google search grounding |
 | [log_prob.js](./log_prob.js)                             | Generate with log probability for each token |
 | [text_generation.js](./text_generation.js)               | Generating text |
+| [api_version.js](./api_version.js)                       | Setting api version|
diff --git a/samples/api_version.js b/samples/api_version.js
new file mode 100644
index 000000000..dc15c16b4
--- /dev/null
+++ b/samples/api_version.js
@@ -0,0 +1,37 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {
+  GoogleGenerativeAI,
+} from "@google/generative-ai";
+
+async function apiVersion() {
+  // [START api_version]
+  // Make sure to include these imports:
+  // import { GoogleGenerativeAI } from "@google/generative-ai";
+  const genAI = new GoogleGenerativeAI(process.env.API_KEY);
+  const model = genAI.getGenerativeModel(
+    {model: "gemini-1.5-flash"},
+    {apiVersion: "v1alpha"},
+  );
+
+  const result = await model.generateContent("Why is the sky blue?");
+  console.log(result.response.text());
+  // [END api_version]
+}
+
+apiVersion();
diff --git a/samples/search_grounding.js b/samples/search_grounding.js
index c587e35a1..cc1c7af13 100644
--- a/samples/search_grounding.js
+++ b/samples/search_grounding.js
@@ -43,7 +43,6 @@ async function searchGrounding() {
         },
       ],
     },
-    { apiVersion: "v1beta" },
   );
 
   const prompt = "What is the price of Google stock today?";