Merge pull request #144 from rjmacarthy/chore/gemini-2.0

Add gemini-2.0 models

Author: RPate97

.changeset/strange-shoes-begin.md

@@ -0,0 +1,5 @@
+---
+'token.js': patch
+---
+
+Add gemini-2.0 models

docs/providers/gemini.md

@@ -38,12 +38,14 @@ main()
 <!-- compatibility -->
 ## Supported Models
 
-| Model               | Chat Completion | Streaming | JSON Output | Image Input | Function Calling | N > 1 |
-| ------------------- | --------------- | --------- | ----------- | ----------- | ---------------- | ----- |
-| gemini-1.5-pro      | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
-| gemini-1.5-flash    | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
-| gemini-1.5-flash-8b | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
-| gemini-1.0-pro      | ✅               | ✅         | ➖           | ➖           | ✅                | ✅     |
+| Model                               | Chat Completion | Streaming | JSON Output | Image Input | Function Calling | N > 1 |
+| ----------------------------------- | --------------- | --------- | ----------- | ----------- | ---------------- | ----- |
+| gemini-2.0-flash-001                | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
+| gemini-2.0-flash-lite-preview-02-05 | ✅               | ✅         | ✅           | ✅           | ➖                | ✅     |
+| gemini-1.5-pro                      | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
+| gemini-1.5-flash                    | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
+| gemini-1.5-flash-8b                 | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
+| gemini-1.0-pro                      | ✅               | ✅         | ➖           | ➖           | ✅                | ✅     |
 
 ### Legend
 | Symbol             | Description                           |

src/models.ts

@@ -167,29 +167,38 @@ export const models = {
   },
   gemini: {
     models: [
+      'gemini-2.0-flash-001',
+      'gemini-2.0-flash-lite-preview-02-05',
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
       'gemini-1.0-pro',
     ] as const,
     supportsCompletion: true,
     supportsStreaming: [
+      'gemini-2.0-flash-001',
+      'gemini-2.0-flash-lite-preview-02-05',
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
       'gemini-1.0-pro',
     ] as const,
     supportsJSON: [
+      'gemini-2.0-flash-001',
+      'gemini-2.0-flash-lite-preview-02-05',
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
     ] as const,
     supportsImages: [
+      'gemini-2.0-flash-001',
+      'gemini-2.0-flash-lite-preview-02-05',
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
     ] as const,
     supportsToolCalls: [
+      'gemini-2.0-flash-001',
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',