diff --git a/ollama/_client.py b/ollama/_client.py
index 607f12c..66104dd 100644
--- a/ollama/_client.py
+++ b/ollama/_client.py
@@ -194,7 +194,7 @@ class Client(BaseClient):
     prompt: str = '',
     options: Optional[Options] = None,
     keep_alive: Optional[Union[float, str]] = None,
-  ) -> Sequence[float]:
+  ) -> Mapping[str, Sequence[float]]:
     return self._request(
       'POST',
       '/api/embeddings',