Change text -> texts for embeddings

examples/example_embeddings.py

@@ -6,5 +6,5 @@
     credentials=...,
     verify_ssl_certs=False,
 ) as giga:
-    response = giga.embeddings("Hello world!")
+    response = giga.embeddings(["Hello world!"])
     print(response)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "gigachat"
-version = "0.1.12"
+version = "0.1.12.1"
 description = "GigaChat. Python-library for GigaChain and LangChain"
 authors = ["Konstantin Krestnikov <[email protected]>", "Sergey Malyshev <[email protected]>"]
 license = "MIT"

src/gigachat/api/post_embeddings.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 
 import httpx
 
@@ -16,7 +16,7 @@
 
 def _get_kwargs(
     *,
-    input_: str,
+    input_: List[str],
     model: str,
     access_token: Optional[str] = None,
 ) -> Dict[str, Any]:
@@ -62,7 +62,7 @@ def _build_response(response: httpx.Response) -> Embeddings:
 def sync(
     client: httpx.Client,
     *,
-    input_: str,
+    input_: List[str],
     model: str,
     access_token: Optional[str] = None,
 ) -> Embeddings:
@@ -74,7 +74,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     *,
-    input_: str,
+    input_: List[str],
     model: str,
     access_token: Optional[str] = None,
 ) -> Embeddings:

src/gigachat/client.py

@@ -206,10 +206,10 @@ def tokens_count(self, input_: List[str], model: Optional[str] = None) -> List[T
             lambda: post_tokens_count.sync(self._client, input_=input_, model=model, access_token=self.token)
         )
 
-    def embeddings(self, text: str, model: str = "Embeddings") -> Embeddings:
+    def embeddings(self, texts: List[str], model: str = "Embeddings") -> Embeddings:
         """Возвращает эмбеддинги"""
         return self._decorator(
-            lambda: post_embeddings.sync(self._client, access_token=self.token, input_=text, model=model)
+            lambda: post_embeddings.sync(self._client, access_token=self.token, input_=texts, model=model)
         )
 
     def get_models(self) -> Models:
@@ -301,11 +301,11 @@ async def _acall() -> List[TokensCount]:
 
         return await self._adecorator(_acall)
 
-    async def aembeddings(self, text: str, model: str = "Embeddings") -> Embeddings:
+    async def aembeddings(self, texts: List[str], model: str = "Embeddings") -> Embeddings:
         """Возвращает эмбеддинги"""
 
         async def _acall() -> Embeddings:
-            return await post_embeddings.asyncio(self._aclient, access_token=self.token, input_=text, model=model)
+            return await post_embeddings.asyncio(self._aclient, access_token=self.token, input_=texts, model=model)
 
         return await self._adecorator(_acall)
 

tests/unit_tests/gigachat/test_client.py

@@ -253,7 +253,7 @@ def test_embeddings(httpx_mock: HTTPXMock) -> None:
     httpx_mock.add_response(url=EMBEDDINGS_URL, json=EMBEDDINGS)
 
     with GigaChatSyncClient(base_url=BASE_URL) as client:
-        response = client.embeddings(text="text", model="model")
+        response = client.embeddings(texts=["text"], model="model")
     assert isinstance(response, Embeddings)
     for row in response.data:
         assert isinstance(row, Embedding)
@@ -456,7 +456,7 @@ async def test_aembeddings(httpx_mock: HTTPXMock) -> None:
     httpx_mock.add_response(url=EMBEDDINGS_URL, json=EMBEDDINGS)
 
     async with GigaChatAsyncClient(base_url=BASE_URL) as client:
-        response = await client.aembeddings(text="text", model="model")
+        response = await client.aembeddings(texts=["text"], model="model")
     assert isinstance(response, Embeddings)
     for row in response.data:
         assert isinstance(row, Embedding)