most-client 1.0.37__py3-none-any.whl → 1.0.38__py3-none-any.whl

most/api.py

@@ -24,8 +24,8 @@ from most.types import (
 
 class MostClient(object):
     retort = Retort(recipe=[
-        loader(int, lambda x: int(x)),
-        loader(float, lambda x: float(x)),
+        loader(int, lambda x: x if isinstance(x, int) else int(x)),
+        loader(float, lambda x: x if isinstance(x, float) else float(x)),
         loader(datetime, lambda x: datetime.fromtimestamp(x).astimezone(tz=timezone.utc) if isinstance(x, (int, float)) else datetime.fromisoformat(x)),
     ],)
 
@@ -264,7 +264,7 @@ class MostClient(object):
                          params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            result = self.get_score_modifier().modify(result)
         return result
 
     def apply_on_text(self, text_id,
@@ -280,7 +280,7 @@ class MostClient(object):
                          params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            result = self.get_score_modifier().modify(result)
         return result
 
     def transcribe_later(self, audio_id,
@@ -308,7 +308,7 @@ class MostClient(object):
                          params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            result = self.get_score_modifier().modify(result)
         return result
 
     def apply_on_text_later(self, text_id,
@@ -324,7 +324,7 @@ class MostClient(object):
                          params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            result = self.get_score_modifier().modify(result)
         return result
 
     def get_job_status(self, audio_id) -> JobStatus:
@@ -348,7 +348,7 @@ class MostClient(object):
         resp = self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/results")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            result = self.get_score_modifier().modify(result)
         return result
 
     def fetch_text(self, audio_id: str) -> Result:
@@ -401,7 +401,7 @@ class MostClient(object):
 
     def store_info(self,
                    audio_id: str,
-                   data: Dict[str, str]) -> StoredAudioData:
+                   data: Dict[str, Union[str, int, float]]) -> StoredAudioData:
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
@@ -409,14 +409,14 @@ class MostClient(object):
                          json={
                              "data": data,
                          })
-        return self.retort.load(resp.json(), StoredAudioData)
+        return StoredAudioData.from_dict(resp.json())
 
     def fetch_info(self, audio_id: str) -> StoredAudioData:
         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}/info")
-        return self.retort.load(resp.json(), StoredAudioData)
+        return StoredAudioData.from_dict(resp.json())
 
     def __call__(self, audio_path: Path,
                  modify_scores: bool = False) -> Result:

most/async_api.py

@@ -24,8 +24,8 @@ from most.types import (
 
 class AsyncMostClient(object):
     retort = Retort(recipe=[
-        loader(int, lambda x: int(x)),
-        loader(float, lambda x: float(x)),
+        loader(int, lambda x: x if isinstance(x, int) else int(x)),
+        loader(float, lambda x: x if isinstance(x, float) else float(x)),
         loader(datetime, lambda x: datetime.fromtimestamp(x).astimezone(tz=timezone.utc) if isinstance(x, (int, float)) else datetime.fromisoformat(x)),
     ])
 
@@ -274,7 +274,8 @@ class AsyncMostClient(object):
                                params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            score_modifier = await self.get_score_modifier()
+            result = score_modifier.modify(result)
         return result
 
     async def apply_on_text(self, text_id,
@@ -290,7 +291,8 @@ class AsyncMostClient(object):
                                params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            score_modifier = await self.get_score_modifier()
+            result = score_modifier.modify(result)
         return result
 
     async def transcribe_later(self, audio_id,
@@ -318,7 +320,8 @@ class AsyncMostClient(object):
                                params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            score_modifier = await self.get_score_modifier()
+            result = score_modifier.modify(result)
         return result
 
     async def apply_on_text_later(self, text_id,
@@ -334,7 +337,8 @@ class AsyncMostClient(object):
                                params={"overwrite": overwrite})
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            score_modifier = await self.get_score_modifier()
+            result = score_modifier.modify(result)
         return result
 
     async def get_job_status(self, audio_id) -> JobStatus:
@@ -358,7 +362,8 @@ class AsyncMostClient(object):
         resp = await self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/results")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
-            result = self.score_modifier.modify(result)
+            score_modifier = await self.get_score_modifier()
+            result = score_modifier.modify(result)
         return result
 
     async def fetch_text(self, audio_id) -> Result:
@@ -411,18 +416,18 @@ class AsyncMostClient(object):
 
     async def store_info(self,
                          audio_id: str,
-                         data: Dict[str, str]) -> StoredAudioData:
+                         data: Dict[str, Union[str, int, float]]) -> StoredAudioData:
         resp = await self.post(f"/{self.client_id}/audio/{audio_id}/info",
                                json={
                                    "data": data,
                                })
-        return self.retort.load(resp.json(), StoredAudioData)
+        return StoredAudioData.from_dict(resp.json())
 
     async def fetch_info(self, audio_id: str) -> StoredAudioData:
         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}/info")
-        return self.retort.load(resp.json(), StoredAudioData)
+        return StoredAudioData.from_dict(resp.json())
 
     async def __call__(self, audio_path: Path,
                        modify_scores: bool = False) -> Result:

most/score_calculation.py

@@ -1,4 +1,4 @@
-from typing import Dict, Tuple, List, Optional
+from typing import Dict, Tuple, List, Optional, Literal
 from dataclasses_json import dataclass_json, DataClassJsonMixin
 from dataclasses import dataclass, replace
 from .types import Result, ScriptScoreMapping
@@ -25,3 +25,55 @@ class ScoreCalculation(DataClassJsonMixin):
                                                            subcolumn_result.score)
 
         return result
+
+    def unmodify(self, result: Optional[Result]):
+        score_mapping = {
+            (sm.column, sm.subcolumn, sm.to_score): sm.from_score
+            for sm in self.score_mapping
+        }
+        if result is None:
+            return None
+        result = replace(result)
+        for column_result in result.results:
+            for subcolumn_result in column_result.subcolumns:
+                subcolumn_result.score = score_mapping.get((column_result.name,
+                                                            subcolumn_result.name,
+                                                            subcolumn_result.score),
+                                                           subcolumn_result.score)
+
+        return result
+
+    def modify_single(self,
+                      column: str, subcolumn: str,
+                      from_score: int):
+        for sm in self.score_mapping:
+            if sm.column == column and sm.subcolumn == subcolumn and sm.from_score == from_score:
+                return sm.to_score
+
+    def unmodify_single(self,
+                        column: str, subcolumn: str,
+                        to_score: int,
+                        bound: Literal["strict", "upper", "lower"] = "strict"):
+        upper_from_score = None
+        lower_from_score = None
+
+        for sm in self.score_mapping:
+            if sm.column == column and sm.subcolumn == subcolumn:
+                if sm.to_score == to_score:
+                    return sm.from_score
+
+                if sm.to_score > to_score:
+                    if upper_from_score is None or sm.to_score < upper_from_score[1]:
+                        upper_from_score = (sm.from_score, sm.to_score)
+                elif sm.to_score < to_score:
+                    if lower_from_score is None or sm.to_score > lower_from_score[1]:
+                        lower_from_score = (sm.from_score, sm.to_score)
+
+        if bound == "strict":
+            return None
+        elif bound == "upper" and upper_from_score is not None:
+            return upper_from_score[0]
+        elif bound == "lower" and lower_from_score is not None:
+            return lower_from_score[0]
+        else:
+            return None

most/search_types.py

@@ -1,7 +1,5 @@
-import re
 from dataclasses import dataclass, field
-from datetime import datetime
-from typing import Dict, List, Literal, Optional, Union
+from typing import List, Optional
 
 from bson import ObjectId
 from dataclasses_json import DataClassJsonMixin, dataclass_json
@@ -33,9 +31,11 @@ class DurationCondition(DataClassJsonMixin):
 @dataclass
 class StoredInfoCondition(DataClassJsonMixin):
     key: str
-    match: Optional[str] = None
+    match: Optional[int | str | float] = None
     starts_with: Optional[str] = None
     ends_with: Optional[str] = None
+    greater_than: Optional[int | str | float] = None
+    less_than: Optional[int | str | float] = None
 
 
 @dataclass_json
@@ -49,6 +49,90 @@ class ResultsCondition(DataClassJsonMixin):
     score_greater_than: Optional[int] = None
     score_less_than: Optional[int] = None
 
+    def create_from(self, client,
+                    column: str, subcolumn: str,
+                    score_equal: Optional[int] = None,
+                    score_in_set: Optional[List[int]] = None,
+                    score_greater_than: Optional[int] = None,
+                    score_less_than: Optional[int] = None,
+                    modified_scores: bool = False) -> 'ResultsCondition':
+        from .api import MostClient
+        client: MostClient
+        script = client.get_model_script()
+        column_idx = [column.name for column in script.columns].index(column)
+        subcolumn_idx = script.columns[column_idx].subcolumns.index(subcolumn)
+
+        if modified_scores:
+            score_modifier = client.get_score_modifier()
+            if score_equal is not None:
+                score_equal = score_modifier.unmodify_single(column, subcolumn,
+                                                             score_equal,
+                                                             bound="strict")
+            if score_in_set is not None:
+                score_in_set = [score_modifier.unmodify_single(column, subcolumn,
+                                                               score,
+                                                               bound="strict")
+                                for score in score_in_set]
+            if score_greater_than is not None:
+                score_greater_than = score_modifier.unmodify_single(column, subcolumn,
+                                                                    score_greater_than,
+                                                                    bound="upper")
+
+            if score_less_than is not None:
+                score_less_than = score_modifier.unmodify_single(column, subcolumn,
+                                                                 score_less_than,
+                                                                 bound="lower")
+
+        return ResultsCondition(model_id=client.model_id,
+                                column_idx=column_idx,
+                                subcolumn_idx=subcolumn_idx,
+                                score_equal=score_equal,
+                                score_in_set=score_in_set,
+                                score_greater_than=score_greater_than,
+                                score_less_than=score_less_than)
+
+    async def acreate_from(self, client,
+                           column: str, subcolumn: str,
+                           score_equal: Optional[int] = None,
+                           score_in_set: Optional[List[int]] = None,
+                           score_greater_than: Optional[int] = None,
+                           score_less_than: Optional[int] = None,
+                           modified_scores: bool = False) -> 'ResultsCondition':
+        from .async_api import AsyncMostClient
+        client: AsyncMostClient
+        script = await client.get_model_script()
+        column_idx = [column.name for column in script.columns].index(column)
+        subcolumn_idx = script.columns[column_idx].subcolumns.index(subcolumn)
+
+        if modified_scores:
+            score_modifier = await client.get_score_modifier()
+            if score_equal is not None:
+                score_equal = score_modifier.unmodify_single(column, subcolumn,
+                                                             score_equal,
+                                                             bound="strict")
+            if score_in_set is not None:
+                score_in_set = [score_modifier.unmodify_single(column, subcolumn,
+                                                               score,
+                                                               bound="strict")
+                                for score in score_in_set]
+            if score_greater_than is not None:
+                score_greater_than = score_modifier.unmodify_single(column, subcolumn,
+                                                                    score_greater_than,
+                                                                    bound="upper")
+
+            if score_less_than is not None:
+                score_less_than = score_modifier.unmodify_single(column, subcolumn,
+                                                                 score_less_than,
+                                                                 bound="lower")
+
+        return ResultsCondition(model_id=client.model_id,
+                                column_idx=column_idx,
+                                subcolumn_idx=subcolumn_idx,
+                                score_equal=score_equal,
+                                score_in_set=score_in_set,
+                                score_greater_than=score_greater_than,
+                                score_less_than=score_less_than)
+
 
 @dataclass_json
 @dataclass

most/types.py

@@ -1,8 +1,7 @@
 import re
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Dict, List, Literal, Optional
-from .search_types import ResultsCondition
+from typing import Dict, List, Literal, Optional, Union
 from dataclasses_json import DataClassJsonMixin, dataclass_json
 
 
@@ -10,7 +9,7 @@ from dataclasses_json import DataClassJsonMixin, dataclass_json
 @dataclass
 class StoredAudioData(DataClassJsonMixin):
     id: str
-    data: Dict[str, str]
+    data: Dict[str, Union[str, int, float]]
 
 
 @dataclass_json
@@ -53,22 +52,6 @@ class Column(DataClassJsonMixin):
 class Script(DataClassJsonMixin):
     columns: List[Column]
 
-    def create_results_condition(self,
-                                 model_id: str,
-                                 column: str, subcolumn: str,
-                                 score_equal: Optional[int] = None,
-                                 score_greater_than: Optional[int] = None,
-                                 score_less_than: Optional[int] = None) -> ResultsCondition:
-
-        column_idx = [column.name for column in self.columns].index(column)
-        subcolumn_idx = self.columns[column_idx].subcolumns.index(subcolumn)
-        return ResultsCondition(model_id=model_id,
-                                column_idx=column_idx,
-                                subcolumn_idx=subcolumn_idx,
-                                score_equal=score_equal,
-                                score_greater_than=score_greater_than,
-                                score_less_than=score_less_than)
-
 
 @dataclass_json
 @dataclass

{most_client-1.0.37.dist-info → most_client-1.0.38.dist-info}/METADATA

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

@@ -0,0 +1,16 @@
+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=Xyo1sGN4QstqIZh_4YDT6FStaXR0HRy6fbtbqc7uqK8,20562
+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=63kBZvvkqlN2UOHXcxTraI9vEq-ANoqXmBnHi_QABDE,6729
+most/searcher.py,sha256=9UdiSlScsE6EPc6RpK8xkRLeB5gHNxgPQpXTJ17i3lQ,2135
+most/trainer_api.py,sha256=ZwOv4mhROfY97n6i7IY_ZpafsuNRazOqMBAf2dh708k,992
+most/types.py,sha256=AU74VqYilM9DXfBwptliRbhV5urLAf4BygdIY3wlAN8,4309
+most_client-1.0.38.dist-info/METADATA,sha256=17PN1v9-yRjWGMSVCEIlIKdkK7z6QQlFIaP9ZtvtQFU,1047
+most_client-1.0.38.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+most_client-1.0.38.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.38.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.38.dist-info/RECORD,,

most_client-1.0.37.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=fwVmqyuCGhALQFOsgRW7IyX7ELY4sePPA5uHFRprCKQ,19052
-most/async_api.py,sha256=Kgy4sHZMLr4FD3dOop1BwfrNHeg5n7Ny0v7ZiYv1_ig,20219
-most/async_searcher.py,sha256=C0zViW20K7OhKO1BzBZktTbMJYBBvor3uK6LAHZTxz0,2238
-most/async_trainer_api.py,sha256=99rED8RjnOn8VezeEgrTgoVfQrO7DdmOE2Jajumno2g,1052
-most/score_calculation.py,sha256=1XU1LfIH5LSCwAbAaKkr-EjH5qOTXrJKOUvhCCawka4,1054
-most/search_types.py,sha256=fuuIkWEuYPTqljl-EMcoNgL5SOiHOVx9j5iH46ecVp8,1857
-most/searcher.py,sha256=9UdiSlScsE6EPc6RpK8xkRLeB5gHNxgPQpXTJ17i3lQ,2135
-most/trainer_api.py,sha256=ZwOv4mhROfY97n6i7IY_ZpafsuNRazOqMBAf2dh708k,992
-most/types.py,sha256=og32qmGnUGTvC0qTH2wzLnhYKeFTIcEmkpIDnclmhYM,5226
-most_client-1.0.37.dist-info/METADATA,sha256=VdEUqqsSRBSfiCdpwYh3bGdKcundkUeHReMTjCbHd2s,1047
-most_client-1.0.37.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-most_client-1.0.37.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.37.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.37.dist-info/RECORD,,