most-client 1.0.22__py3-none-any.whl → 1.0.24__py3-none-any.whl

most/api.py

@@ -18,7 +18,7 @@ from most.types import (
     Script,
     StoredAudioData,
     Text,
-    is_valid_id, SearchParams, ScriptScoreMapping,
+    is_valid_id, SearchParams, ScriptScoreMapping, Dialog,
 )
 
 
@@ -114,6 +114,24 @@ class MostClient(object):
             raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
         return resp
 
+    def put(self, url, **kwargs):
+        if self.access_token is None:
+            self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = self.session.put(url,
+                                headers=headers,
+                                timeout=None,
+                                **kwargs)
+        if resp.status_code == 401:
+            self.refresh_access_token()
+            return self.put(url,
+                            headers=headers,
+                            **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
     def post(self, url,
              data=None,
              json=None,
@@ -144,6 +162,11 @@ class MostClient(object):
                          files={"text": text})
         return self.retort.load(resp.json(), Text)
 
+    def upload_dialog(self, dialog: Dialog) -> DialogResult:
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_dialog",
+                         json={"dialog": dialog})
+        return self.retort.load(resp.json(), DialogResult)
+
     def upload_audio(self, audio_path) -> Audio:
         with open(audio_path, 'rb') as f:
             resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
@@ -302,6 +325,17 @@ class MostClient(object):
         resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
         return self.retort.load(resp.json(), DialogResult)
 
+    def update_dialog(self, audio_id, dialog: Dialog) -> DialogResult:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(audio_id):
+            raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
+
+        resp = self.put(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                        json={"dialog": dialog.to_dict()})
+        return self.retort.load(resp.json(), DialogResult)
+
     def export(self, audio_ids: List[str],
                aggregated_by: Optional[str] = None,
                aggregation_title: Optional[str] = None) -> str:

most/async_api.py

@@ -18,7 +18,7 @@ from most.types import (
     Script,
     StoredAudioData,
     Text,
-    is_valid_id, SearchParams, ScriptScoreMapping,
+    is_valid_id, SearchParams, ScriptScoreMapping, Dialog,
 )
 
 
@@ -122,6 +122,24 @@ class AsyncMostClient(object):
                 "Content-Type") == "application/json" else "Something went wrong.")
         return resp
 
+    async def put(self, url, **kwargs):
+        if self.access_token is None:
+            await self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = await self.session.put(url,
+                                      headers=headers,
+                                      timeout=None,
+                                      **kwargs)
+        if resp.status_code == 401:
+            await self.refresh_access_token()
+            return await self.put(url,
+                                  headers=headers,
+                                  **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
     async def post(self, url,
                    data=None,
                    json=None,
@@ -170,6 +188,11 @@ class AsyncMostClient(object):
                                json={"text": text})
         return self.retort.load(resp.json(), Text)
 
+    async def upload_dialog(self, dialog: Dialog) -> DialogResult:
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_dialog",
+                               json={"dialog": dialog})
+        return self.retort.load(resp.json(), DialogResult)
+
     async def upload_audio_url(self, audio_url) -> Audio:
         resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_url",
                                json={"audio_url": audio_url})
@@ -311,6 +334,17 @@ class AsyncMostClient(object):
         resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
         return self.retort.load(resp.json(), DialogResult)
 
+    async def update_dialog(self, audio_id, dialog: Dialog) -> DialogResult:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(audio_id):
+            raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
+
+        resp = await self.put(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                              json={"dialog": dialog.to_dict()})
+        return self.retort.load(resp.json(), DialogResult)
+
     async def export(self, audio_ids: List[str],
                      aggregated_by: Optional[str] = None,
                      aggregation_title: Optional[str] = None) -> str:

{most_client-1.0.22.dist-info → most_client-1.0.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.22
+Version: 1.0.24
 Summary: Most AI API for https://the-most.ai
 Home-page: https://github.com/the-most-ai/most-client
 Author: George Kasparyants

most_client-1.0.24.dist-info/RECORD

@@ -0,0 +1,10 @@
+most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
+most/api.py,sha256=HWSwJX1QFRCvhf5OkH3Fx8lvGjLt_6gM9H6D5aKZ1og,17638
+most/async_api.py,sha256=w56Ll0r97I__PSqFxvCGO6GOCeno1qvOdAwgK5FhtlM,18703
+most/score_calculation.py,sha256=1XU1LfIH5LSCwAbAaKkr-EjH5qOTXrJKOUvhCCawka4,1054
+most/types.py,sha256=Qgyv261J8b1cfbmeITz1C9QgkoCMGQQd_L4t4M3dd6M,3603
+most_client-1.0.24.dist-info/METADATA,sha256=63Geupu8DOnw5svjeWeOSBW1S5qPSZZ7ZbZLynNXs9k,1027
+most_client-1.0.24.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+most_client-1.0.24.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.24.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.24.dist-info/RECORD,,

most_client-1.0.22.dist-info/RECORD

@@ -1,10 +0,0 @@
-most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
-most/api.py,sha256=ffgsS-i9pxNOgjMBd7-G-PqA69-VFCVn47vHSJW8XeM,15980
-most/async_api.py,sha256=UF7DXTmIGfPl93noMR742FkMiRtSO5zLV-c4qZVDMNs,16949
-most/score_calculation.py,sha256=1XU1LfIH5LSCwAbAaKkr-EjH5qOTXrJKOUvhCCawka4,1054
-most/types.py,sha256=Qgyv261J8b1cfbmeITz1C9QgkoCMGQQd_L4t4M3dd6M,3603
-most_client-1.0.22.dist-info/METADATA,sha256=ske3Ee2ObuBu-q6omKkTPdwBgwfAKqVDTmf-jn0vlDs,1027
-most_client-1.0.22.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-most_client-1.0.22.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.22.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.22.dist-info/RECORD,,