Add missing types for text-embedding-*

dqbd · Feb 2, 2025 · 335b291 · 335b291
1 parent 5982def
commit 335b291
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/wasm/src/lib.rs b/wasm/src/lib.rs
@@ -385,6 +385,8 @@ export type TiktokenModel =
     | "text-davinci-edit-001"
     | "code-davinci-edit-001"
     | "text-embedding-ada-002"
+    | "text-embedding-3-small"
+    | "text-embedding-3-large"
     | "text-similarity-davinci-001"
     | "text-similarity-curie-001"
     | "text-similarity-babbage-001"