From 2ac751fb7931adb7f6655f1b9186038c5c815026 Mon Sep 17 00:00:00 2001
From: jingfelix <jingfelix@outlook.com>
Date: Fri, 3 May 2024 16:29:03 +0800
Subject: [PATCH] fix: annotation in AsyncClient.embedding

Signed-off-by: jingfelix <jingfelix@outlook.com>
---
 ollama/_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ollama/_client.py b/ollama/_client.py
index 88fca50..3c5a767 100644
--- a/ollama/_client.py
+++ b/ollama/_client.py
@@ -474,7 +474,7 @@ async def embeddings(
     prompt: str = '',
     options: Optional[Options] = None,
     keep_alive: Optional[Union[float, str]] = None,
-  ) -> Sequence[float]:
+  ) -> Mapping[str, Sequence[float]]:
     response = await self._request(
       'POST',
       '/api/embeddings',