most-client 1.0.41__py3-none-any.whl → 1.0.42__py3-none-any.whl

most/__init__.py

@@ -4,4 +4,4 @@ from .trainer_api import Trainer
 from .async_trainer_api import AsyncTrainer
 from .searcher import MostSearcher
 from .async_searcher import AsyncMostClient
-from .search_types import SearchParams, IDCondition, ChannelsCondition, DurationCondition, ResultsCondition, StoredInfoCondition
+from .search_types import SearchParams, IDCondition, ChannelsCondition, DurationCondition, ResultsCondition, StoredInfoCondition, TagsCondition

most/api.py

@@ -3,7 +3,7 @@ import os
 import uuid
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union, Literal
 import json5
 import httpx
 from adaptix import Retort, loader
@@ -361,14 +361,20 @@ class MostClient(object):
         resp = self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/text")
         return self.retort.load(resp.json(), Result)
 
-    def fetch_dialog(self, audio_id) -> DialogResult:
+    def fetch_dialog(self, audio_id,
+                     transcribator_name: Optional[Literal["GroundTruth"]] = None) -> 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.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
+        params = {}
+        if transcribator_name is not None:
+            params["transcribator_name"] = transcribator_name
+
+        resp = self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                        params=params)
         return self.retort.load(resp.json(), DialogResult)
 
     def update_dialog(self, audio_id, dialog: Dialog) -> DialogResult:

most/async_api.py

@@ -3,7 +3,7 @@ import os
 import uuid
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union, Literal
 import httpx
 import json5
 from adaptix import Retort, loader
@@ -374,14 +374,19 @@ class AsyncMostClient(object):
         resp = await self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/text")
         return self.retort.load(resp.json(), Result)
 
-    async def fetch_dialog(self, audio_id) -> DialogResult:
+    async def fetch_dialog(self, audio_id,
+                           transcribator_name: Optional[Literal["GroundTruth"]] = None) -> 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.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
+        params = {}
+        if transcribator_name is not None:
+            params["transcribator_name"] = transcribator_name
+        resp = await self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                              params=params)
         return self.retort.load(resp.json(), DialogResult)
 
     async def update_dialog(self, audio_id, dialog: Dialog) -> DialogResult:

most/async_glossary.py

@@ -0,0 +1,18 @@
+from typing import List
+
+from .async_api import AsyncMostClient
+from .types import GlossaryNGram
+
+
+class AsyncGlossary(object):
+    def __init__(self, client: AsyncMostClient):
+        self.client = client
+
+    async def add_ngram(self, ngram: GlossaryNGram):
+        pass
+
+    async def list_ngrams(self) -> List[GlossaryNGram]:
+        return []
+
+    async def del_ngram(self, ngram_id: str):
+        pass

most/glossary.py

@@ -0,0 +1,18 @@
+from typing import List
+
+from .api import MostClient
+from .types import GlossaryNGram
+
+
+class Glossary(object):
+    def __init__(self, client: MostClient):
+        self.client = client
+
+    def add_ngram(self, ngram: GlossaryNGram):
+        pass
+
+    def list_ngrams(self) -> List[GlossaryNGram]:
+        return []
+
+    def del_ngram(self, ngram_id: str):
+        pass

most/search_types.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass, field
-from typing import List, Optional
+from typing import List, Optional, Literal
 from dataclasses_json import DataClassJsonMixin, dataclass_json
 
 
@@ -10,12 +10,14 @@ class IDCondition(DataClassJsonMixin):
     in_set: Optional[List[str]] = None
     greater_than: Optional[str] = None
     less_than: Optional[str] = None
+    type: Literal["IDCondition"] = "IDCondition"
 
 
 @dataclass_json
 @dataclass
 class ChannelsCondition(DataClassJsonMixin):
     equal: Optional[int] = None
+    type: Literal["ChannelsCondition"] = "ChannelsCondition"
 
 
 @dataclass_json
@@ -23,12 +25,14 @@ class ChannelsCondition(DataClassJsonMixin):
 class DurationCondition(DataClassJsonMixin):
     greater_than: Optional[int] = None
     less_than: Optional[int] = None
+    type: Literal["DurationCondition"] = "DurationCondition"
 
 
 @dataclass_json
 @dataclass
 class TagsCondition(DataClassJsonMixin):
     in_set: Optional[List[str]] = None
+    type: Literal["TagsCondition"] = "TagsCondition"
 
 
 @dataclass_json
@@ -40,6 +44,7 @@ class StoredInfoCondition(DataClassJsonMixin):
     ends_with: Optional[str] = None
     greater_than: Optional[int | str | float] = None
     less_than: Optional[int | str | float] = None
+    type: Literal["StoredInfoCondition"] = "StoredInfoCondition"
 
 
 @dataclass_json
@@ -52,6 +57,7 @@ class ResultsCondition(DataClassJsonMixin):
     score_in_set: Optional[List[int]] = None
     score_greater_than: Optional[int] = None
     score_less_than: Optional[int] = None
+    type: Literal["ResultsCondition"] = "ResultsCondition"
 
     def create_from(self, client,
                     column: str, subcolumn: str,

most/types.py

@@ -114,6 +114,27 @@ class Dialog(DataClassJsonMixin):
         return ''.join([segment.to_text()
                         for segment in self.segments])
 
+    def to_raw_text(self):
+        return ' '.join([segment.text
+                         for segment in self.segments])
+
+    def to_raw_speaker_text(self, speaker):
+        return ' '.join([segment.text
+                         for segment in self.segments
+                         if segment.speaker == speaker])
+
+    def get_speaker_names(self):
+        return list(set([segment.speaker
+                         for segment in self.segments]))
+
+
+@dataclass_json
+@dataclass
+class GlossaryNGram(DataClassJsonMixin):
+    pronunciation: List[str]
+    original: List[str]
+    id: Optional[str] = None
+
 
 @dataclass_json
 @dataclass

{most_client-1.0.41.dist-info → most_client-1.0.42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: most-client
-Version: 1.0.41
+Version: 1.0.42
 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.42.dist-info/RECORD

@@ -0,0 +1,18 @@
+most/__init__.py,sha256=UPx7vJhrRrSNsROGFJQr6-7Lw55qit0z-PIORw_HHFg,366
+most/_constrants.py,sha256=SlHKcBoXwe_sPzk8tdbb7lqhQz-Bfo__FhSoeFWodZE,217
+most/api.py,sha256=-DZK8TZ7_OdgUZTknqLW27i8YigXKM33W0vjr8b92gE,19401
+most/async_api.py,sha256=RJkdQGuNJ9s-PuyHbcuVqGmp_GN5ry8bWbVzi8k4ed0,20696
+most/async_glossary.py,sha256=kYb-6lptBJuBSHL1-dIPIEeT5RKuUkzyCAJWmKoZ7ec,408
+most/async_searcher.py,sha256=C0zViW20K7OhKO1BzBZktTbMJYBBvor3uK6LAHZTxz0,2238
+most/async_trainer_api.py,sha256=99rED8RjnOn8VezeEgrTgoVfQrO7DdmOE2Jajumno2g,1052
+most/glossary.py,sha256=EO6TNJThqimU_gs7zN87JX0QKkSXbLtojPox57SL4XE,369
+most/score_calculation.py,sha256=vLtGqXrR43xZhGjrH5dpQZfWX1q3s74LvTaHn-SKBAg,3254
+most/search_types.py,sha256=GFwXaDJV4qI_gaFZToLMHoYH-JXVfIVwfVTgJ7RDWoY,7211
+most/searcher.py,sha256=9UdiSlScsE6EPc6RpK8xkRLeB5gHNxgPQpXTJ17i3lQ,2135
+most/trainer_api.py,sha256=ZwOv4mhROfY97n6i7IY_ZpafsuNRazOqMBAf2dh708k,992
+most/types.py,sha256=mxE2KbvP88TzI6s6Aw_6JkV4uQXXEGG_hM310QOG-bo,4909
+most_client-1.0.42.dist-info/METADATA,sha256=iF1o_VepUfy-s1rJZw-BcF0S5Rq2QpdK1goJwFRfsdI,1027
+most_client-1.0.42.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+most_client-1.0.42.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.42.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.42.dist-info/RECORD,,

most_client-1.0.41.dist-info/RECORD

@@ -1,16 +0,0 @@
-most/__init__.py,sha256=yoPKMxjZYAzrVqZ7l9rN0Skh0HPuttPX7ub0wlZMby4,351
-most/_constrants.py,sha256=SlHKcBoXwe_sPzk8tdbb7lqhQz-Bfo__FhSoeFWodZE,217
-most/api.py,sha256=HmJSAwLVuy7uaeVAM3y1nd-F-UDZ2kOVww_i2EY0Sjs,19145
-most/async_api.py,sha256=CTUpju0byI_HdE_zw3OPrNfRW-TXpTtoCK0tt7iVkXY,20429
-most/async_searcher.py,sha256=C0zViW20K7OhKO1BzBZktTbMJYBBvor3uK6LAHZTxz0,2238
-most/async_trainer_api.py,sha256=99rED8RjnOn8VezeEgrTgoVfQrO7DdmOE2Jajumno2g,1052
-most/score_calculation.py,sha256=vLtGqXrR43xZhGjrH5dpQZfWX1q3s74LvTaHn-SKBAg,3254
-most/search_types.py,sha256=NI4hK-ALJbyHW6ZhRGlGRVM6H2C2GsLNhQZb9YUMrnU,6854
-most/searcher.py,sha256=9UdiSlScsE6EPc6RpK8xkRLeB5gHNxgPQpXTJ17i3lQ,2135
-most/trainer_api.py,sha256=ZwOv4mhROfY97n6i7IY_ZpafsuNRazOqMBAf2dh708k,992
-most/types.py,sha256=AU74VqYilM9DXfBwptliRbhV5urLAf4BygdIY3wlAN8,4309
-most_client-1.0.41.dist-info/METADATA,sha256=sr64VTdOACoMc6zx4ZiF1G5p9cYQQtth13xpZ5BeXnI,1027
-most_client-1.0.41.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-most_client-1.0.41.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.41.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.41.dist-info/RECORD,,