most-client 1.0.8__tar.gz

most_client-1.0.8/MANIFEST.in

@@ -0,0 +1,3 @@
+recursive-include most *
+include README.md
+include requirements.txt

most_client-1.0.8/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.1
+Name: most-client
+Version: 1.0.8
+Summary: Most AI API for https://the-most.ai
+Home-page: https://github.com/the-most-ai/most-client
+Author: George Kasparyants
+Author-email: george@the-most.ai
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.6
+Requires-Python: >=3.6
+Requires-Dist: requests
+Requires-Dist: wheel
+Requires-Dist: adaptix
+Requires-Dist: json5
+Requires-Dist: dataclasses_json
+Requires-Dist: black
+Requires-Dist: coverage
+Requires-Dist: flake8
+Requires-Dist: mypy
+Requires-Dist: pylint
+Requires-Dist: pytest
+Requires-Dist: tox
+Requires-Dist: twine
+Requires-Dist: httpx

most_client-1.0.8/README.md

@@ -0,0 +1,30 @@
+# MOST AI Python Client  
+
+
+```python
+from most import MostClient
+
+most = MostClient(model_id="most-***")
+
+audio_path = "example.mp3"
+results = most(audio_path)
+print(results)
+
+{
+    "id": "AudioID",
+    "url": "Audio CDN Url",
+    "text": "Transcribed text from audio",
+    "results": [
+        {
+            "name": "Приветствие",
+            "subcolumns": [
+                {
+                    "name": "Назвал свое имя",
+                    "description": "Менеджер назвал свое имя - Алина",
+                    "score": 2,
+                }
+            ]
+        }
+    ]
+}
+```

most_client-1.0.8/most/__init__.py

@@ -0,0 +1,2 @@
+from .api import MostClient
+from .async_api import AsyncMostClient

most_client-1.0.8/most/api.py

@@ -0,0 +1,197 @@
+from typing import List
+import json5
+import requests
+from adaptix import Retort
+from most.types import Audio, Result, Script, JobStatus
+from pathlib import Path
+
+
+class MostClient(object):
+    retort = Retort()
+
+    def __init__(self,
+                 client_id=None,
+                 client_secret=None,
+                 model_id=None):
+        super(MostClient, self).__init__()
+        self.client_id = client_id
+        self.client_secret = client_secret
+
+        if self.client_id is None:
+            credentials = self.load_credentials()
+            self.client_id = credentials.get('client_id')
+            self.client_secret = credentials.get('client_secret')
+
+            if self.client_id is None:
+                print("Visit: https://app.the-most.ai/integrations and get clientId, clientSecret")
+                self.client_id = input("Please enter your client ID: ")
+                self.client_secret = input("Please enter your client secret: ")
+                self.save_credentials()
+        else:
+            self.save_credentials()
+
+        self.session = requests.Session()
+        self.access_token = None
+        self.model_id = model_id
+
+        self.refresh_access_token()
+
+    @property
+    def cache_path(self):
+        path = Path.home() / ".most"
+        path.mkdir(parents=True, exist_ok=True)
+        return path
+
+    def load_credentials(self):
+        path = self.cache_path / "credentials.json"
+        if not path.exists():
+            return {}
+        else:
+            return json5.loads(path.read_text())
+
+    def save_credentials(self):
+        path = self.cache_path / "credentials.json"
+        path.write_text(json5.dumps({
+            "client_id": self.client_id,
+            "client_secret": self.client_secret,
+        }))
+
+    def clone(self):
+        client = MostClient(client_id=self.client_id,
+                            client_secret=self.client_secret,
+                            model_id=self.model_id)
+        client.access_token = self.access_token
+        client.session = self.session
+        return client
+
+    def with_model(self, model_id):
+        client = self.clone()
+        client.model_id = model_id
+        return client
+
+    def refresh_access_token(self):
+        resp = self.session.post("https://api.the-most.ai/api/external/access_token",
+                                 json={"client_id": self.client_id,
+                                        "client_secret": self.client_secret},
+                                 timeout=None)
+        access_token = resp.json()
+        self.access_token = access_token
+
+    def get(self, url, **kwargs):
+        if self.access_token is None:
+            self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = self.session.get(url,
+                                headers=headers,
+                                timeout=None,
+                                **kwargs)
+        if resp.status_code == 401:
+            self.refresh_access_token()
+            return self.get(url,
+                            headers=headers,
+                            **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
+    def post(self, url,
+             data=None,
+             json=None,
+             **kwargs):
+        if self.access_token is None:
+            self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = self.session.post(url,
+                                 data=data,
+                                 json=json,
+                                 headers=headers,
+                                 timeout=None,
+                                 **kwargs)
+        if resp.status_code == 401:
+            self.refresh_access_token()
+            return self.post(url,
+                             data=data,
+                             json=json,
+                             headers=headers,
+                             **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
+    def upload_audio(self, audio_path) -> Audio:
+        with open(audio_path, 'rb') as f:
+            resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+                             files={"audio_file": f})
+        return self.retort.load(resp.json(), Audio)
+
+    def upload_audio_url(self, audio_url) -> Audio:
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_url",
+                         json={"audio_url": audio_url})
+        return self.retort.load(resp.json(), Audio)
+
+    def list_audios(self,
+                    offset: int = 0,
+                    limit: int = 10) -> List[Audio]:
+        resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list?offset={offset}&limit={limit}")
+        audio_list = resp.json()
+        return self.retort.load(audio_list, List[Audio])
+
+    def get_model_script(self) -> Script:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/model/{self.model_id}/script")
+        return self.retort.load(resp.json(), Script)
+
+    def list_models(self):
+        resp = self.get("https://api.the-most.ai/api/external/list_models")
+        return [self.with_model(model['model'])
+                for model in resp.json()]
+
+    def apply(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
+        return self.retort.load(resp.json(), Result)
+
+    def apply_later(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
+        return self.retort.load(resp.json(), Result)
+
+    def get_job_status(self, audio_id) -> JobStatus:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_status")
+        return self.retort.load(resp.json(), JobStatus)
+
+    def fetch_results(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+
+        resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/results")
+        return self.retort.load(resp.json(), Result)
+
+    def fetch_text(self, audio_id: str) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+
+        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 export(self, audio_ids: List[str]) -> str:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a 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)})
+        return resp.url
+
+    def __call__(self, audio_path: Path):
+        audio = self.upload_audio(audio_path)
+        return self.apply(audio.id)
+
+    def __repr__(self):
+        return "<MostClient(model_id='%s')>" % (self.model_id, )

most_client-1.0.8/most/async_api.py

@@ -0,0 +1,204 @@
+from typing import List
+import json5
+from adaptix import Retort
+from most.types import Audio, Result, Script, JobStatus
+from pathlib import Path
+import httpx
+
+
+class AsyncMostClient(object):
+    retort = Retort()
+
+    def __init__(self,
+                 client_id=None,
+                 client_secret=None,
+                 model_id=None):
+        super(AsyncMostClient, self).__init__()
+        self.client_id = client_id
+        self.client_secret = client_secret
+
+        if self.client_id is None:
+            credentials = self.load_credentials()
+            self.client_id = credentials.get('client_id')
+            self.client_secret = credentials.get('client_secret')
+
+            if self.client_id is None:
+                print("Visit: https://app.the-most.ai/integrations and get clientId, clientSecret")
+                self.client_id = input("Please enter your client ID: ")
+                self.client_secret = input("Please enter your client secret: ")
+                self.save_credentials()
+        else:
+            self.save_credentials()
+
+        self.session = httpx.AsyncClient()
+        self.access_token = None
+        self.model_id = model_id
+
+    async def __aenter__(self):
+        await self.session.__aenter__()
+        await self.refresh_access_token()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.session.__aexit__(exc_type, exc_val, exc_tb)
+
+    @property
+    def cache_path(self):
+        path = Path.home() / ".most"
+        path.mkdir(parents=True, exist_ok=True)
+        return path
+
+    def load_credentials(self):
+        path = self.cache_path / "credentials.json"
+        if not path.exists():
+            return {}
+        else:
+            return json5.loads(path.read_text())
+
+    def save_credentials(self):
+        path = self.cache_path / "credentials.json"
+        path.write_text(json5.dumps({
+            "client_id": self.client_id,
+            "client_secret": self.client_secret,
+        }))
+
+    def clone(self):
+        client = AsyncMostClient(client_id=self.client_id,
+                                 client_secret=self.client_secret,
+                                 model_id=self.model_id)
+        client.access_token = self.access_token
+        client.session = self.session
+        return client
+
+    def with_model(self, model_id):
+        client = self.clone()
+        client.model_id = model_id
+        return client
+
+    async def refresh_access_token(self):
+        resp = await self.session.post("https://api.the-most.ai/api/external/access_token",
+                                       json={"client_id": self.client_id,
+                                             "client_secret": self.client_secret})
+        access_token = resp.json()
+        self.access_token = access_token
+
+    async def get(self, url, **kwargs):
+        if self.access_token is None:
+            await self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = await self.session.get(url,
+                                      headers=headers,
+                                      timeout=None,
+                                      **kwargs)
+        if resp.status_code == 401:
+            await self.refresh_access_token()
+            return await self.get(url,
+                                  headers=headers,
+                                  **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get(
+                "Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
+    async def post(self, url,
+                   data=None,
+                   json=None,
+                   **kwargs):
+        if self.access_token is None:
+            await self.refresh_access_token()
+        headers = kwargs.pop("headers", {})
+        headers.update({"Authorization": "Bearer %s" % self.access_token})
+        resp = await self.session.post(url,
+                                       data=data,
+                                       json=json,
+                                       headers=headers,
+                                       timeout=None,
+                                       **kwargs)
+        if resp.status_code == 401:
+            await self.refresh_access_token()
+            return await self.post(url,
+                                   data=data,
+                                   json=json,
+                                   headers=headers,
+                                   **kwargs)
+        if resp.status_code >= 400:
+            raise RuntimeError(resp.json()['message'] if resp.headers.get(
+                "Content-Type") == "application/json" else "Something went wrong.")
+        return resp
+
+    async def upload_audio(self, audio_path) -> Audio:
+        with open(audio_path, mode='rb') as f:
+            resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+                                   files={"audio_file": f})
+        return self.retort.load(resp.json(), Audio)
+
+    async def upload_audio_url(self, audio_url) -> Audio:
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_url",
+                               json={"audio_url": audio_url})
+        return self.retort.load(resp.json(), Audio)
+
+    async def list_audios(self,
+                    offset: int = 0,
+                    limit: int = 10) -> List[Audio]:
+        resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list?offset={offset}&limit={limit}")
+        audio_list = resp.json()
+        return self.retort.load(audio_list, List[Audio])
+
+    async def get_model_script(self) -> Script:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a 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}/script")
+        return self.retort.load(resp.json(), Script)
+
+    async def list_models(self):
+        resp = await self.get("https://api.the-most.ai/api/external/list_models")
+        return [self.with_model(model['model'])
+                for model in resp.json()]
+
+    async def apply(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
+        return self.retort.load(resp.json(), Result)
+
+    async def apply_later(self, audio_id):
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
+        return self.retort.load(resp.json(), Result)
+
+    async def get_job_status(self, audio_id) -> JobStatus:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_status")
+        return self.retort.load(resp.json(), JobStatus)
+
+    async def fetch_results(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+
+        resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/results")
+        return self.retort.load(resp.json(), Result)
+
+    async def fetch_text(self, audio_id) -> Result:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a model to apply. [try list_models()]")
+
+        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 export(self, audio_ids: List[str]) -> str:
+        if self.model_id is None:
+            raise RuntimeError("Please choose a 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)})
+        return resp.next_request.url
+
+    async def __call__(self, audio_path: Path):
+        audio = await self.upload_audio(audio_path)
+        return await self.apply(audio.id)
+
+    def __repr__(self):
+        return "<AsyncMostClient(model_id='%s')>" % (self.model_id, )

most_client-1.0.8/most/types.py

@@ -0,0 +1,62 @@
+from dataclasses import dataclass
+from dataclasses_json import dataclass_json, DataClassJsonMixin
+from typing import Optional, List, Literal
+
+
+@dataclass_json
+@dataclass
+class Audio(DataClassJsonMixin):
+    id: str
+    url: str
+
+
+@dataclass_json
+@dataclass
+class SubcolumnResult(DataClassJsonMixin):
+    name: str
+    score: Optional[int]
+    description: str
+
+
+@dataclass_json
+@dataclass
+class ColumnResult(DataClassJsonMixin):
+    name: str
+    subcolumns: List[SubcolumnResult]
+
+
+@dataclass_json
+@dataclass
+class Column(DataClassJsonMixin):
+    name: str
+    subcolumns: List[str]
+
+
+@dataclass_json
+@dataclass
+class Script(DataClassJsonMixin):
+    columns: List[Column]
+
+
+@dataclass_json
+@dataclass
+class JobStatus(DataClassJsonMixin):
+    status: Literal["not_found", "pending", "completed", "error"]
+
+
+@dataclass_json
+@dataclass
+class Result(DataClassJsonMixin):
+    id: str
+    text: Optional[str]
+    url: Optional[str]
+    results: Optional[List[ColumnResult]]
+
+    def get_script(self) -> Optional[Script]:
+        if self.results is None:
+            return None
+
+        return Script(columns=[Column(name=column_result.name,
+                                      subcolumns=[subcolumn_result.name
+                                                  for subcolumn_result in column_result.subcolumns])
+                               for column_result in self.results])

most_client-1.0.8/most_client.egg-info/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.1
+Name: most-client
+Version: 1.0.8
+Summary: Most AI API for https://the-most.ai
+Home-page: https://github.com/the-most-ai/most-client
+Author: George Kasparyants
+Author-email: george@the-most.ai
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.6
+Requires-Python: >=3.6
+Requires-Dist: requests
+Requires-Dist: wheel
+Requires-Dist: adaptix
+Requires-Dist: json5
+Requires-Dist: dataclasses_json
+Requires-Dist: black
+Requires-Dist: coverage
+Requires-Dist: flake8
+Requires-Dist: mypy
+Requires-Dist: pylint
+Requires-Dist: pytest
+Requires-Dist: tox
+Requires-Dist: twine
+Requires-Dist: httpx

most_client-1.0.8/most_client.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+MANIFEST.in
+README.md
+requirements.txt
+setup.py
+most/__init__.py
+most/api.py
+most/async_api.py
+most/types.py
+most_client.egg-info/PKG-INFO
+most_client.egg-info/SOURCES.txt
+most_client.egg-info/dependency_links.txt
+most_client.egg-info/requires.txt
+most_client.egg-info/top_level.txt
+most_client.egg-info/zip-safe

most_client-1.0.8/most_client.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

most_client-1.0.8/most_client.egg-info/requires.txt

@@ -0,0 +1,14 @@
+requests
+wheel
+adaptix
+json5
+dataclasses_json
+black
+coverage
+flake8
+mypy
+pylint
+pytest
+tox
+twine
+httpx

most_client-1.0.8/most_client.egg-info/top_level.txt

@@ -0,0 +1 @@
+most

most_client-1.0.8/most_client.egg-info/zip-safe

@@ -0,0 +1 @@
+

most_client-1.0.8/requirements.txt

@@ -0,0 +1,14 @@
+requests
+wheel
+adaptix
+json5
+dataclasses_json
+black
+coverage
+flake8
+mypy
+pylint
+pytest
+tox
+twine
+httpx

most_client-1.0.8/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

most_client-1.0.8/setup.py

@@ -0,0 +1,31 @@
+from setuptools import setup
+from setuptools import find_packages
+
+
+with open('requirements.txt', 'r') as f:
+    requirements = f.read()
+
+
+setup(
+    name='most-client',
+    version='1.0.8',
+    python_requires=f'>=3.6',
+    description='Most AI API for https://the-most.ai',
+    url='https://github.com/the-most-ai/most-client',
+    author='George Kasparyants',
+    author_email='george@the-most.ai',
+    license='',
+    packages=find_packages(include=['most', 'most.*']),
+    install_requires=requirements,
+    zip_safe=True,
+    include_package_data=True,
+    exclude_package_data={'': ['notebooks']},
+    classifiers=[
+        "Intended Audience :: Developers",
+        "Operating System :: OS Independent",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+        "Programming Language :: Python",
+        "Programming Language :: Python :: 3 :: Only",
+        "Programming Language :: Python :: 3.6"
+    ],
+)