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

most/__init__.py

@@ -4,4 +4,9 @@ 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
+from .glossary import Glossary
+from .async_glossary import AsyncGlossary
+from .catalog import Catalog
+from .async_catalog import AsyncCatalog
+from .types import GlossaryNGram, Item

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_catalog.py

@@ -0,0 +1,28 @@
+from typing import List
+
+from .async_api import AsyncMostClient
+from .types import Item
+
+
+class AsyncCatalog(object):
+    def __init__(self, client: AsyncMostClient):
+        super(AsyncCatalog, self).__init__()
+        self.client = client
+
+    async def add_items(self, items: List[Item] | Item) -> List[Item]:
+        if not isinstance(items, list):
+            items = [items]
+        resp = await self.client.post(f"/{self.client.client_id}/upload_items",
+                                      json=[item.to_dict() for item in items])
+        return self.client.retort.load(resp.json(), List[Item])
+
+    async def list_items(self) -> List[Item]:
+        resp = await self.client.get(f"/{self.client.client_id}/items")
+        return self.client.retort.load(resp.json(), List[Item])
+
+    async def delete_items(self, item_ids: List[str]):
+        if not isinstance(item_ids, list):
+            item_ids = [item_ids]
+        await self.client.post(f"/{self.client.client_id}/delete_items",
+                               json=item_ids)
+        return self

most/async_glossary.py

@@ -0,0 +1,32 @@
+from typing import List
+
+from .async_api import AsyncMostClient
+from .types import GlossaryNGram
+
+
+class AsyncGlossary(object):
+    def __init__(self, client: AsyncMostClient):
+        super(AsyncGlossary, self).__init__()
+        self.client = client
+
+    async def add_ngrams(self, ngrams: List[GlossaryNGram] | GlossaryNGram) -> List[GlossaryNGram]:
+        if not isinstance(ngrams, list):
+            ngrams = [ngrams]
+        resp = await self.client.post(f"/{self.client.client_id}/upload_glossary",
+                                      json=[ngram.to_dict() for ngram in ngrams])
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    async def list_ngrams(self) -> List[GlossaryNGram]:
+        resp = await self.client.get(f"/{self.client.client_id}/glossary")
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    async def del_ngrams(self, ngram_ids: List[str] | str):
+        if not isinstance(ngram_ids, list):
+            ngram_ids = [ngram_ids]
+        await self.client.post(f"/{self.client.client_id}/delete_glossary_ngrams",
+                               json=ngram_ids)
+        return self
+
+    async def drop(self):
+        await self.client.post(f"/{self.client.client_id}/delete_glossary")
+        return self

most/catalog.py

@@ -0,0 +1,28 @@
+from typing import List
+
+from .api import MostClient
+from .types import Item
+
+
+class Catalog(object):
+    def __init__(self, client: MostClient):
+        super(Catalog, self).__init__()
+        self.client = client
+
+    def add_items(self, items: List[Item] | Item) -> List[Item]:
+        if not isinstance(items, list):
+            items = [items]
+        resp = self.client.post(f"/{self.client.client_id}/upload_items",
+                                json=[item.to_dict() for item in items])
+        return self.client.retort.load(resp.json(), List[Item])
+
+    def list_items(self) -> List[Item]:
+        resp = self.client.get(f"/{self.client.client_id}/items")
+        return self.client.retort.load(resp.json(), List[Item])
+
+    def delete_items(self, item_ids: List[str]):
+        if not isinstance(item_ids, list):
+            item_ids = [item_ids]
+        self.client.post(f"/{self.client.client_id}/delete_items",
+                         json=item_ids)
+        return self

most/glossary.py

@@ -0,0 +1,32 @@
+from typing import List
+
+from .api import MostClient
+from .types import GlossaryNGram
+
+
+class Glossary(object):
+    def __init__(self, client: MostClient):
+        super(Glossary, self).__init__()
+        self.client = client
+
+    def add_ngrams(self, ngrams: List[GlossaryNGram] | GlossaryNGram) -> List[GlossaryNGram]:
+        if not isinstance(ngrams, list):
+            ngrams = [ngrams]
+        resp = self.client.post(f"/{self.client.client_id}/upload_glossary",
+                                json=[ngram.to_dict() for ngram in ngrams])
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    def list_ngrams(self) -> List[GlossaryNGram]:
+        resp = self.client.get(f"/{self.client.client_id}/glossary")
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    def del_ngrams(self, ngram_ids: List[str] | str):
+        if not isinstance(ngram_ids, list):
+            ngram_ids = [ngram_ids]
+        self.client.post(f"/{self.client.client_id}/delete_glossary_ngrams",
+                         json=ngram_ids)
+        return self
+
+    def drop(self):
+        self.client.post(f"/{self.client.client_id}/delete_glossary")
+        return self

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,37 @@ 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):
+    original: List[str]
+    pronunciation: List[str]
+    weight: float = 2
+    id: Optional[str] = None
+
+
+@dataclass_json
+@dataclass
+class Item(DataClassJsonMixin):
+    title: str
+    pronunciation: str
+    metadata: Dict[str, str]
+    id: Optional[str] = None
+
 
 @dataclass_json
 @dataclass

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

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

@@ -0,0 +1,20 @@
+most/__init__.py,sha256=ZA2n_h824tfGd5tFl3RsD0noUyv_T08fBoBxMEgxmzs,548
+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_catalog.py,sha256=LgQ5zghQtDMnqq-WjmRHqZkqcdixZnHtmoRaVv74VKM,1059
+most/async_glossary.py,sha256=nehgkQvqENxT5vnmMiRLSWgD0jFr2Hzs7yqNZfC9ahs,1281
+most/async_searcher.py,sha256=C0zViW20K7OhKO1BzBZktTbMJYBBvor3uK6LAHZTxz0,2238
+most/async_trainer_api.py,sha256=99rED8RjnOn8VezeEgrTgoVfQrO7DdmOE2Jajumno2g,1052
+most/catalog.py,sha256=XFu-pZdPyxyBcLDl9O3yiYvESXUolXTm0jsQKrQlHxc,985
+most/glossary.py,sha256=64ld6cyvBe8jYW7OR1AVDoEPEAif5PFYu0Hgyng3Gkg,1195
+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=rfIfj_Dylrd_SGQl-GGlTMLUAjM6wI55TIsP3RwAsC4,5088
+most_client-1.0.43.dist-info/METADATA,sha256=jql2yXek-OubXmLj4v8TmaZDoIgZIQKAUNXpGBwZ5U0,1027
+most_client-1.0.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+most_client-1.0.43.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.43.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.43.dist-info/RECORD,,

{most_client-1.0.41.dist-info → most_client-1.0.43.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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,,