feat: add ability to config the api version

.github/README.md

@@ -423,3 +423,17 @@ const gemini = new Gemini(API_KEY, {
 
 <h2 align="center" id="contributors">Contributors</h2>
 <p align="center">A special shoutout to developers of and contributors to the <a href="https://github.com/EvanZhouDev/bard-ai"><code>bard-ai</code></a> and <a href="https://github.com/EvanZhouDev/palm-api"><code>palm-api</code></a> libraries. Gemini AI's interface is heavily based on what we have developed on these two projects.</p>
+
+### I Want more control over the API version
+
+By default the project uses the `v1beta` API version to enable the most features and backwards compatibility. You can read more about it [here](https://ai.google.dev/docs/api_versions).
+
+If you feel the need to use a different version than the default you can pass it as an option on gemini initialization:
+
+```javascript
+import Gemini from "gemini-ai";
+
+const gemini = new Gemini(API_KEY, {
+	apiVersion: "v1",
+});
+```

index.js

@@ -47,6 +47,7 @@ const answerPairToParameter = (message) => {
 export default class Gemini {
 	#fetch;
 	#dispatcher;
+	#apiVersion;
 
 	static JSON = "json";
 	static TEXT = "markdown";
@@ -61,6 +62,7 @@ export default class Gemini {
 		const config = this.#parseConfig(rawConfig, {
 			fetch: defaultFetch,
 			dispatcher: undefined,
+			apiVersion: 'v1beta'
 		});
 
 		if (!config.fetch)
@@ -71,6 +73,7 @@ export default class Gemini {
 		this.#fetch = config.fetch;
 		this.key = key;
 		this.#dispatcher = config.dispatcher;
+		this.#apiVersion = config.apiVersion;
 	}
 
 	#parseConfig(raw = {}, defaults = {}) {
@@ -110,7 +113,7 @@ export default class Gemini {
 		};
 
 		const response = await this.#fetch(
-			`https://generativelanguage.googleapis.com/v1beta/models/${model}:${command}?key=${this.key}`,
+			`https://generativelanguage.googleapis.com/${this.#apiVersion}/models/${model}:${command}?key=${this.key}`,
 			opts,
 		);
 

test/fullCoverage.test.js

@@ -228,3 +228,12 @@ test("Gemini.createChat() with JSON", async () => {
 		}),
 	).toBe(generateContentResponse);
 });
+
+test("Gemini Create with correct version", async () => {
+	const fetchSpy = vi.spyOn(global, 'fetch')
+	fetch.mockReturnValueOnce(createFetchResponse(generateContentResponse));
+
+	const gemini = new Gemini(API_KEY, { apiVersion: "v1" });
+	expect(await gemini.ask("Hello!")).toBe("Hi!");
+	expect(fetchSpy).toHaveBeenCalledWith('https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent?key=demo-key', expect.any(Object));
+})