From 42f9f70baea5138c685e301d85e0bd20006a3501 Mon Sep 17 00:00:00 2001
From: Niko <n.vlahovic@proton.me>
Date: Fri, 5 Apr 2024 11:56:16 +0200
Subject: [PATCH] modified embeddings return annotation

---
 ollama/_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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',