most-client 1.0.17__py3-none-any.whl → 1.0.19__py3-none-any.whl

most/api.py

@@ -17,7 +17,7 @@ from most.types import (
     Script,
     StoredAudioData,
     Text,
-    is_valid_id,
+    is_valid_id, SearchParams,
 )
 
 
@@ -241,12 +241,19 @@ 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 export(self, audio_ids: List[str]) -> str:
+    def export(self, audio_ids: List[str],
+               aggregated_by: Optional[str] = None,
+               aggregation_title: Optional[str] = None) -> str:
+        if aggregation_title is None:
+            aggregation_title = aggregated_by
+
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")
 
         resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/model/{self.model_id}/export",
-                        params={'audio_ids': ','.join(audio_ids)})
+                        params={'audio_ids': ','.join(audio_ids),
+                                'aggregated_by': aggregated_by,
+                                'aggregation_title': aggregation_title})
         return resp.url
 
     def store_info(self,
@@ -289,3 +296,24 @@ class MostClient(object):
         audio = AudioSegment.from_file(io.BytesIO(resp.content),
                                        format=format)
         return audio
+
+    def index_audio(self, audio_id: str) -> None:
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/indexing")
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio can't be indexed")
+        return None
+
+    def search(self,
+               query: str,
+               filter: SearchParams,
+               limit: int = 10) -> List[Audio]:
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/model/{self.model_id}/search",
+                         json={
+                             "query": query,
+                             "filter": filter.to_dict(),
+                             "limit": limit,
+                         })
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio can't be indexed")
+        audio_list = resp.json()
+        return self.retort.load(audio_list, List[Audio])

most/async_api.py

@@ -17,7 +17,7 @@ from most.types import (
     Script,
     StoredAudioData,
     Text,
-    is_valid_id,
+    is_valid_id, SearchParams,
 )
 
 
@@ -250,12 +250,19 @@ 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 export(self, audio_ids: List[str]) -> str:
+    async def export(self, audio_ids: List[str],
+                     aggregated_by: Optional[str] = None,
+                     aggregation_title: Optional[str] = None) -> str:
+        if aggregation_title is None:
+            aggregation_title = aggregated_by
+
         if not is_valid_id(self.model_id):
             raise RuntimeError("Please choose valid model to apply. [try list_models()]")
 
         resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/model/{self.model_id}/export",
-                              params={'audio_ids': ','.join(audio_ids)})
+                              params={'audio_ids': ','.join(audio_ids),
+                                      "aggregated_by": aggregated_by,
+                                      "aggregation_title": aggregation_title})
         return resp.next_request.url
 
     async def store_info(self,
@@ -294,3 +301,24 @@ class AsyncMostClient(object):
         audio = AudioSegment.from_file(io.BytesIO(resp.content),
                                        format=format)
         return audio
+
+    async def index_audio(self, audio_id: str) -> None:
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/indexing")
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio can't be indexed")
+        return None
+
+    async def search(self,
+                     query: str,
+                     filter: SearchParams,
+                     limit: int = 10) -> List[Audio]:
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/model/{self.model_id}/search",
+                               json={
+                                   "query": query,
+                                   "filter": filter.to_dict(),
+                                   "limit": limit,
+                               })
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio can't be indexed")
+        audio_list = resp.json()
+        return self.retort.load(audio_list, List[Audio])

most/types.py

@@ -113,6 +113,32 @@ class DialogResult(DataClassJsonMixin):
     results: Optional[List[ColumnResult]]
 
 
+@dataclass_json
+@dataclass
+class StoredInfoCondition(DataClassJsonMixin):
+    key: str
+    match: Optional[str] = None
+    starts_with: Optional[str] = None
+    ends_with: Optional[str] = None
+
+
+@dataclass_json
+@dataclass
+class ResultsCondition(DataClassJsonMixin):
+    column: str
+    subcolumn: str
+    score_greater_than: Optional[int] = None
+    score_less_than: Optional[int] = None
+
+
+@dataclass_json
+@dataclass
+class SearchParams(DataClassJsonMixin):
+    must: List[StoredInfoCondition | ResultsCondition]
+    should: List[StoredInfoCondition | ResultsCondition]
+    must_not: List[StoredInfoCondition | ResultsCondition]
+
+
 def is_valid_objectid(oid: str) -> bool:
     """
     Check if a given string is a valid MongoDB ObjectId (24-character hex).

{most_client-1.0.17.dist-info → most_client-1.0.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.17
+Version: 1.0.19
 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.19.dist-info/RECORD

@@ -0,0 +1,9 @@
+most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
+most/api.py,sha256=fn5Husx-uayQ2MR_I7S1YA9QiCHc7fSFLTIzWqyAJc0,13046
+most/async_api.py,sha256=D4sM9zo7EN5vfoMuI2-FsdNwpht0nLuRej0o4DY5smc,13875
+most/types.py,sha256=M6KZiy0OurFCHSnJnnqrkwmsCjTyrq7nsICIfAhLo-A,3455
+most_client-1.0.19.dist-info/METADATA,sha256=jUYmnk-Qx5LM5vGUiKOLZY8ZA57ZSf4pxVZeJH0ous4,1027
+most_client-1.0.19.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+most_client-1.0.19.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.19.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.19.dist-info/RECORD,,

most_client-1.0.17.dist-info/RECORD

@@ -1,9 +0,0 @@
-most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
-most/api.py,sha256=yRGL7lZ5qBWSdaclz-t7fZAJ_k1zu0YbHptUX9uaVdM,11772
-most/async_api.py,sha256=xRZPsb64i1d_U0PWIJKC_7fPMjZRPfamvvtSMyFlA5E,12505
-most/types.py,sha256=apUz6FHFAHWJNVvRh57wVQk5DFw2XKnLLhkAyWgtvfw,2825
-most_client-1.0.17.dist-info/METADATA,sha256=51ED0zyrdWP5OqTlY31mCcXRpBmk39y1BXNWJYx2BMY,1027
-most_client-1.0.17.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-most_client-1.0.17.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.17.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.17.dist-info/RECORD,,