most-client 1.0.21__tar.gz → 1.0.23__tar.gz

{most_client-1.0.21/most_client.egg-info → most_client-1.0.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.21
+Version: 1.0.23
 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.21 → most_client-1.0.23}/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,
 )
 
 
@@ -144,6 +144,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",
@@ -173,6 +178,13 @@ class MostClient(object):
         audio_list = resp.json()
         return self.retort.load(audio_list, List[Audio])
 
+    def list_texts(self,
+                   offset: int = 0,
+                   limit: int = 10) -> List[Text]:
+        resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list_texts?offset={offset}&limit={limit}")
+        texts_list = resp.json()
+        return self.retort.load(texts_list, List[Text])
+
     def get_model_script(self) -> Script:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")
@@ -209,6 +221,20 @@ class MostClient(object):
             result = self.score_modifier.modify(result)
         return result
 
+    def apply_on_text(self, text_id,
+                      modify_scores: bool = False) -> Result:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(text_id):
+            raise RuntimeError("Please use valid text_id. [try text.id from list_texts()]")
+
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
+        result = self.retort.load(resp.json(), Result)
+        if modify_scores:
+            result = self.score_modifier.modify(result)
+        return result
+
     def apply_later(self, audio_id,
                     modify_scores: bool = False) -> Result:
         if not is_valid_id(self.model_id):
@@ -223,6 +249,20 @@ class MostClient(object):
             result = self.score_modifier.modify(result)
         return result
 
+    def apply_on_text_later(self, text_id,
+                            modify_scores: bool = False) -> Result:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(text_id):
+            raise RuntimeError("Please use valid text_id. [try audio.id from list_texts()]")
+
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
+        result = self.retort.load(resp.json(), Result)
+        if modify_scores:
+            result = self.score_modifier.modify(result)
+        return result
+
     def get_job_status(self, audio_id) -> JobStatus:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")

{most_client-1.0.21 → most_client-1.0.23}/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,
 )
 
 
@@ -170,6 +170,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})
@@ -182,6 +187,13 @@ class AsyncMostClient(object):
         audio_list = resp.json()
         return self.retort.load(audio_list, List[Audio])
 
+    async def list_texts(self,
+                         offset: int = 0,
+                         limit: int = 10) -> List[Text]:
+        resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list_texts?offset={offset}&limit={limit}")
+        texts_list = resp.json()
+        return self.retort.load(texts_list, List[Text])
+
     async def get_model_script(self) -> Script:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")
@@ -218,6 +230,20 @@ class AsyncMostClient(object):
             result = self.score_modifier.modify(result)
         return result
 
+    async def apply_on_text(self, text_id,
+                            modify_scores: bool = False) -> Result:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(text_id):
+            raise RuntimeError("Please use valid text_id. [try text.id from list_texts()]")
+
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
+        result = self.retort.load(resp.json(), Result)
+        if modify_scores:
+            result = self.score_modifier.modify(result)
+        return result
+
     async def apply_later(self, audio_id,
                           modify_scores: bool = False) -> Result:
         if not is_valid_id(self.model_id):
@@ -232,6 +258,20 @@ class AsyncMostClient(object):
             result = self.score_modifier.modify(result)
         return result
 
+    async def apply_on_text_later(self, text_id,
+                                  modify_scores: bool = False) -> Result:
+        if not is_valid_id(self.model_id):
+            raise RuntimeError("Please choose valid model to apply. [try list_models()]")
+
+        if not is_valid_id(text_id):
+            raise RuntimeError("Please use valid text_id. [try audio.id from list_texts()]")
+
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
+        result = self.retort.load(resp.json(), Result)
+        if modify_scores:
+            result = self.score_modifier.modify(result)
+        return result
+
     async def get_job_status(self, audio_id) -> JobStatus:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")

{most_client-1.0.21 → most_client-1.0.23/most_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.21
+Version: 1.0.23
 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.21 → most_client-1.0.23}/setup.py

@@ -8,7 +8,7 @@ with open('requirements.txt', 'r') as f:
 
 setup(
     name='most-client',
-    version='1.0.21',
+    version='1.0.23',
     python_requires=f'>=3.6',
     description='Most AI API for https://the-most.ai',
     url='https://github.com/the-most-ai/most-client',