most-client 1.0.13__py3-none-any.whl → 1.0.15__py3-none-any.whl

most/api.py

@@ -2,7 +2,7 @@ from typing import List, Dict
 import json5
 import requests
 from adaptix import Retort
-from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id
+from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id, DialogResult
 from pathlib import Path
 
 
@@ -205,6 +205,16 @@ class MostClient(object):
         resp = self.get(f"https://api.the-most.ai/api/external/{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:
+        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"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 export(self, audio_ids: List[str]) -> str:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")

most/async_api.py

@@ -1,7 +1,7 @@
 from typing import List, Dict
 import json5
 from adaptix import Retort
-from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id
+from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id, DialogResult
 from pathlib import Path
 import httpx
 
@@ -214,6 +214,16 @@ 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}/text")
         return self.retort.load(resp.json(), Result)
 
+    async def fetch_dialog(self, audio_id) -> 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"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 export(self, audio_ids: List[str]) -> str:
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")

most/types.py

@@ -1,6 +1,5 @@
+import re
 from dataclasses import dataclass
-
-from bson import ObjectId
 from dataclasses_json import dataclass_json, DataClassJsonMixin
 from typing import Optional, List, Literal, Dict
 
@@ -77,6 +76,49 @@ class Result(DataClassJsonMixin):
                                for column_result in self.results])
 
 
+@dataclass_json
+@dataclass
+class DialogSegment(DataClassJsonMixin):
+    start_time_ms: int
+    end_time_ms: int
+    text: str
+    speaker: str
+    emotion: Optional[str] = None
+    intensity: Optional[float] = None
+
+    def to_text(self):
+        if self.emotion is None:
+            return f'{self.speaker}: {self.text}\n'
+        else:
+            return f'{self.speaker}: <emotion name="{self.emotion}" intensity="{self.intensity}">{self.text}</emotion>\n'
+
+
+@dataclass_json
+@dataclass
+class Dialog(DataClassJsonMixin):
+    segments: List[DialogSegment]
+
+    def to_text(self):
+        return ''.join([segment.to_text()
+                        for segment in self.segments])
+
+
+@dataclass_json
+@dataclass
+class DialogResult(DataClassJsonMixin):
+    id: str
+    dialog: Optional[Dialog]
+    url: Optional[str]
+    results: Optional[List[ColumnResult]]
+
+
+def is_valid_objectid(oid: str) -> bool:
+    """
+    Check if a given string is a valid MongoDB ObjectId (24-character hex).
+    """
+    return bool(re.fullmatch(r"^[0-9a-fA-F]{24}$", oid))
+
+
 def is_valid_id(smth_id: Optional[str]) -> bool:
     if smth_id is None:
         return False
@@ -84,8 +126,4 @@ def is_valid_id(smth_id: Optional[str]) -> bool:
     if smth_id.startswith("most-"):
         smth_id = smth_id[5:]
 
-    try:
-        ObjectId(smth_id)
-        return True
-    except:
-        return False
+    return is_valid_objectid(smth_id)

{most_client-1.0.13.dist-info → most_client-1.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.13
+Version: 1.0.15
 Summary: Most AI API for https://the-most.ai
 Home-page: https://github.com/the-most-ai/most-client
 Author: George Kasparyants
@@ -26,7 +26,6 @@ Requires-Dist: pytest
 Requires-Dist: tox
 Requires-Dist: twine
 Requires-Dist: httpx
-Requires-Dist: bson==0.5.10
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

most_client-1.0.15.dist-info/RECORD

@@ -0,0 +1,9 @@
+most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
+most/api.py,sha256=xFuvjL45gNHDJyPd9_QG3u7uv75CMi3EWX5m47odd7E,10598
+most/async_api.py,sha256=eB7bF5a_4hDBCn64VXqXIcu0GIlzYdEIkqihh287qEo,11271
+most/types.py,sha256=tZFuRKx6hxnOvpLw5pPe7p3H1mW_cxFkTc2fJTKxHiI,2824
+most_client-1.0.15.dist-info/METADATA,sha256=-PTbL9VVxEatBuS1VB_icOQjvEZjIjg1XAWxSo8leFw,1006
+most_client-1.0.15.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
+most_client-1.0.15.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.15.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.15.dist-info/RECORD,,

most_client-1.0.13.dist-info/RECORD

@@ -1,9 +0,0 @@
-most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
-most/api.py,sha256=sGChrPUuwBjSdkkJK3BDOh3YYaxe9xAuGS82GcEdQk4,10074
-most/async_api.py,sha256=ZiUXdqRBG8yTEBHfuglN1eaBMr29S9ibWWfFM5EP61w,10735
-most/types.py,sha256=nSYmVm27y9py_HG6a7oKSObDRHCymmTYo-x5bYhKqug,1844
-most_client-1.0.13.dist-info/METADATA,sha256=KsD6Q75Q2h5mt2wWENoNB0W4QYdbtAVi5q5M0hVdVjY,1034
-most_client-1.0.13.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
-most_client-1.0.13.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.13.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.13.dist-info/RECORD,,