diff --git a/package.json b/package.json
index c0e6d43..1c66167 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "vlmrun",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "The official TypeScript library for the Vlm API",
   "author": "Vlm <support@vlm.run>",
   "types": "dist/index.d.ts",
diff --git a/src/resources/document.ts b/src/resources/document.ts
index 6248bee..4aff6e1 100644
--- a/src/resources/document.ts
+++ b/src/resources/document.ts
@@ -75,7 +75,7 @@ export interface DocumentGenerateParams {
   /**
    * The model to use for generating the response.
    */
-  model?: 'vlm-1';
+  model?: 'vlm-1' | 'vlm-1-embeddings';
 
   /**
    * The URL of the file (provide either `file_id` or `url`).
diff --git a/src/resources/image.ts b/src/resources/image.ts
index d80c783..7327961 100644
--- a/src/resources/image.ts
+++ b/src/resources/image.ts
@@ -68,7 +68,7 @@ export interface ImageGenerateParams {
   /**
    * The model to use for generating the response.
    */
-  model?: 'vlm-1';
+  model?: 'vlm-1' | 'vlm-1-embeddings';
 }
 
 export namespace ImageGenerateParams {
diff --git a/src/version.ts b/src/version.ts
index 6a0e493..6554624 100644
--- a/src/version.ts
+++ b/src/version.ts
@@ -1 +1 @@
-export const VERSION = '0.1.1';
+export const VERSION = '0.1.5';