From c76b0ba76f933a9621432440f3ac47e0934e0002 Mon Sep 17 00:00:00 2001
From: Michael Yang <git@mxy.ng>
Date: Wed, 10 Sep 2025 14:42:12 -0700
Subject: [PATCH] feat: add dimensions to embed request

---
 ollama/_client.py | 4 ++++
 ollama/_types.py  | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/ollama/_client.py b/ollama/_client.py
index 0a85a74a..4bcc1b15 100644
--- a/ollama/_client.py
+++ b/ollama/_client.py
@@ -363,6 +363,7 @@ def embed(
     truncate: Optional[bool] = None,
     options: Optional[Union[Mapping[str, Any], Options]] = None,
     keep_alive: Optional[Union[float, str]] = None,
+    dimensions: Optional[int] = None,
   ) -> EmbedResponse:
     return self._request(
       EmbedResponse,
@@ -374,6 +375,7 @@ def embed(
         truncate=truncate,
         options=options,
         keep_alive=keep_alive,
+        dimensions=dimensions,
       ).model_dump(exclude_none=True),
     )
 
@@ -875,6 +877,7 @@ async def embed(
     truncate: Optional[bool] = None,
     options: Optional[Union[Mapping[str, Any], Options]] = None,
     keep_alive: Optional[Union[float, str]] = None,
+    dimensions: Optional[int] = None,
   ) -> EmbedResponse:
     return await self._request(
       EmbedResponse,
@@ -886,6 +889,7 @@ async def embed(
         truncate=truncate,
         options=options,
         keep_alive=keep_alive,
+        dimensions=dimensions,
       ).model_dump(exclude_none=True),
     )
 
diff --git a/ollama/_types.py b/ollama/_types.py
index 04822875..8d3226e1 100644
--- a/ollama/_types.py
+++ b/ollama/_types.py
@@ -382,6 +382,9 @@ class EmbedRequest(BaseRequest):
 
   keep_alive: Optional[Union[float, str]] = None
 
+  dimensions: Optional[int] = None
+  'Dimensions truncates the output embedding to the specified dimension.'
+
 
 class EmbedResponse(BaseGenerateResponse):
   """