feat: Support o1-preview, o1-mini, gemini-1.5-flash-8b

RPate97 · RPate97 · commit b5d8fa60fccc · 2024-11-16T12:10:18.000-08:00
diff --git a/.changeset/healthy-bikes-switch.md b/.changeset/healthy-bikes-switch.md
@@ -0,0 +1,5 @@
+---
+'token.js': patch
+---
+
+Support o1-preview and o1-mini
diff --git a/.changeset/heavy-pets-carry.md b/.changeset/heavy-pets-carry.md
@@ -0,0 +1,5 @@
+---
+'token.js': patch
+---
+
+Support gemini-1.5-flash-8b
diff --git a/docs/providers/gemini.md b/docs/providers/gemini.md
@@ -38,11 +38,12 @@ 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.0-pro   | ✅               | ✅         | ➖           | ➖           | ✅                | ✅     |
+| 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      | ✅               | ✅         | ➖           | ➖           | ✅                | ✅     |
 
 ### Legend
 | Symbol             | Description                           |
diff --git a/docs/providers/openai.md b/docs/providers/openai.md
@@ -63,6 +63,10 @@ main()
 | gpt-3.5-turbo-1106     | ✅               | ✅         | ✅           | ➖           | ✅                | ✅     |
 | gpt-3.5-turbo-0125     | ✅               | ✅         | ✅           | ➖           | ✅                | ✅     |
 | gpt-3.5-turbo-16k-0613 | ✅               | ✅         | ➖           | ➖           | ➖                | ✅     |
+| o1-mini                | ✅               | ➖         | ➖           | ➖           | ➖                | ✅     |
+| o1-mini-2024-09-12     | ✅               | ➖         | ➖           | ➖           | ➖                | ✅     |
+| o1-preview             | ✅               | ➖         | ➖           | ➖           | ➖                | ✅     |
+| o1-preview-2024-09-12  | ✅               | ➖         | ➖           | ➖           | ➖                | ✅     |
 
 ### Legend
 | Symbol             | Description                           |
diff --git a/package.json b/package.json
@@ -36,7 +36,7 @@
     "cohere-ai": "^7.10.6",
     "mime-types": "^2.1.35",
     "nanoid": "^5.0.7",
-    "openai": "^4.52.2"
+    "openai": "^4.72.0"
   },
   "devDependencies": {
     "@babel/eslint-parser": "^7.18.2",
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/models.ts b/src/models.ts

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +'token.js': patch
 +---
++
 +Support o1-preview and o1-mini