most-client 1.0.23__py3-none-any.whl → 1.0.25__py3-none-any.whl

most/_constrants.py

@@ -0,0 +1,7 @@
+import httpx
+
+
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=600.0, connect=5.0)
+DEFAULT_MAX_RETRIES = 2
+DEFAULT_RETRY_DELAY = 5
+DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=1000, max_keepalive_connections=100)

most/api.py

@@ -2,13 +2,12 @@ import io
 import os
 import uuid
 from pathlib import Path
-from typing import Dict, List, Optional
-
+from typing import Dict, List, Optional, Union
 import json5
-import requests
+import httpx
 from adaptix import Retort
 from pydub import AudioSegment
-
+from most._constrants import DEFAULT_MAX_RETRIES, DEFAULT_RETRY_DELAY
 from most.score_calculation import ScoreCalculation
 from most.types import (
     Audio,
@@ -28,7 +27,13 @@ class MostClient(object):
     def __init__(self,
                  client_id=None,
                  client_secret=None,
-                 model_id=None):
+                 model_id=None,
+
+                 base_url: str | httpx.URL | None = None,
+                 timeout: Union[float, httpx.Timeout] = 1e10,
+                 max_retries: int = DEFAULT_MAX_RETRIES,
+                 # retry_delay: float = DEFAULT_RETRY_DELAY,
+                 http_client: httpx.Client | None = None):
         super(MostClient, self).__init__()
         self.client_id = client_id
         self.client_secret = client_secret
@@ -46,7 +51,19 @@ class MostClient(object):
         else:
             self.save_credentials()
 
-        self.session = requests.Session()
+        if base_url is None:
+            base_url = os.environ.get("MOST_BASE_URL")
+        if base_url is None:
+            base_url = f"https://api.the-most.ai/api/external"
+
+        if http_client is None:
+            http_client = httpx.Client(base_url=base_url,
+                                       timeout=timeout,
+                                       follow_redirects=True,
+                                       transport=httpx.HTTPTransport(retries=max_retries))
+        # self.max_retries = max_retries
+        # self.retry_delay = retry_delay
+        self.session = http_client
         self.access_token = None
         self.model_id = model_id
         self.score_modifier = None
@@ -89,7 +106,7 @@ class MostClient(object):
         return client
 
     def refresh_access_token(self):
-        resp = self.session.post("https://api.the-most.ai/api/external/access_token",
+        resp = self.session.post("/access_token",
                                  json={"client_id": self.client_id,
                                         "client_secret": self.client_secret},
                                  timeout=None)
@@ -114,6 +131,24 @@ class MostClient(object):
             raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
         return resp
 
+    def put(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.put(url,
+                                headers=headers,
+                                timeout=None,
+                                **kwargs)
+        if resp.status_code == 401:
+            self.refresh_access_token()
+            return self.put(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,
@@ -140,18 +175,18 @@ class MostClient(object):
         return resp
 
     def upload_text(self, text: str) -> Text:
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_text",
+        resp = self.post(f"/{self.client_id}/upload_text",
                          files={"text": text})
         return self.retort.load(resp.json(), Text)
 
     def upload_dialog(self, dialog: Dialog) -> DialogResult:
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_dialog",
+        resp = self.post(f"/{self.client_id}/upload_dialog",
                          json={"dialog": dialog})
         return self.retort.load(resp.json(), DialogResult)
 
     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",
+            resp = self.post(f"/{self.client_id}/upload",
                              files={"audio_file": f})
         return self.retort.load(resp.json(), Audio)
 
@@ -162,26 +197,26 @@ class MostClient(object):
         f.seek(0)
         if audio_name is None:
             audio_name = uuid.uuid4().hex + ".mp3"
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+        resp = self.post(f"/{self.client_id}/upload",
                          files={"audio_file": (audio_name, f, 'audio/mp3')})
         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",
+        resp = self.post(f"/{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}")
+        resp = self.get(f"/{self.client_id}/list?offset={offset}&limit={limit}")
         audio_list = resp.json()
         return self.retort.load(audio_list, List[Audio])
 
     def list_texts(self,
                    offset: int = 0,
                    limit: int = 10) -> List[Text]:
-        resp = self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list_texts?offset={offset}&limit={limit}")
+        resp = self.get(f"/{self.client_id}/list_texts?offset={offset}&limit={limit}")
         texts_list = resp.json()
         return self.retort.load(texts_list, List[Text])
 
@@ -189,7 +224,7 @@ class MostClient(object):
         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}/script")
+        resp = self.get(f"/{self.client_id}/model/{self.model_id}/script")
         return self.retort.load(resp.json(), Script)
 
     def get_score_modifier(self):
@@ -197,13 +232,13 @@ class MostClient(object):
             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}/score_mapping")
+            resp = self.get(f"/{self.client_id}/model/{self.model_id}/score_mapping")
             score_mapping = self.retort.load(resp.json(), List[ScriptScoreMapping])
             self.score_modifier = ScoreCalculation(score_mapping)
         return self.score_modifier
 
     def list_models(self):
-        resp = self.get("https://api.the-most.ai/api/external/list_models")
+        resp = self.get("/list_models")
         return [self.with_model(model['model'])
                 for model in resp.json()]
 
@@ -215,7 +250,7 @@ class MostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
+        resp = self.post(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -229,7 +264,7 @@ class MostClient(object):
         if not is_valid_id(text_id):
             raise RuntimeError("Please use valid text_id. [try text.id from list_texts()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
+        resp = self.post(f"/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -243,7 +278,7 @@ class MostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
+        resp = self.post(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -257,7 +292,7 @@ class MostClient(object):
         if not is_valid_id(text_id):
             raise RuntimeError("Please use valid text_id. [try audio.id from list_texts()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
+        resp = self.post(f"/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -270,7 +305,7 @@ class MostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_status")
+        resp = self.post(f"/{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,
@@ -281,7 +316,7 @@ class MostClient(object):
         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}/results")
+        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)
@@ -294,7 +329,7 @@ class MostClient(object):
         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}/text")
+        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:
@@ -304,7 +339,18 @@ class MostClient(object):
         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")
+        resp = self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
+        return self.retort.load(resp.json(), DialogResult)
+
+    def update_dialog(self, audio_id, dialog: Dialog) -> 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.put(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                        json={"dialog": dialog.to_dict()})
         return self.retort.load(resp.json(), DialogResult)
 
     def export(self, audio_ids: List[str],
@@ -316,7 +362,7 @@ class MostClient(object):
         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",
+        resp = self.get(f"/{self.client_id}/model/{self.model_id}/export",
                         params={'audio_ids': ','.join(audio_ids),
                                 'aggregated_by': aggregated_by,
                                 'aggregation_title': aggregation_title})
@@ -328,7 +374,7 @@ class MostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/info",
+        resp = self.post(f"/{self.client_id}/audio/{audio_id}/info",
                          json={
                              "data": data,
                          })
@@ -338,7 +384,7 @@ class MostClient(object):
         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}/info")
+        resp = self.get(f"/{self.client_id}/audio/{audio_id}/info")
         return self.retort.load(resp.json(), StoredAudioData)
 
     def __call__(self, audio_path: Path,
@@ -366,7 +412,7 @@ class MostClient(object):
         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")
+        resp = self.post(f"/{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
@@ -375,7 +421,7 @@ class MostClient(object):
                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",
+        resp = self.post(f"/{self.client_id}/model/{self.model_id}/search",
                          json={
                              "query": query,
                              "filter": filter.to_dict(),

most/async_api.py

@@ -2,13 +2,12 @@ import io
 import os
 import uuid
 from pathlib import Path
-from typing import Dict, List, Optional
-
+from typing import Dict, List, Optional, Union
 import httpx
 import json5
 from adaptix import Retort
 from pydub import AudioSegment
-
+from most._constrants import DEFAULT_MAX_RETRIES
 from most.score_calculation import ScoreCalculation
 from most.types import (
     Audio,
@@ -28,7 +27,13 @@ class AsyncMostClient(object):
     def __init__(self,
                  client_id=None,
                  client_secret=None,
-                 model_id=None):
+                 model_id=None,
+
+                 base_url: str | httpx.URL | None = None,
+                 timeout: Union[float, httpx.Timeout] = 1e10,
+                 max_retries: int = DEFAULT_MAX_RETRIES,
+                 # retry_delay: float = 0,
+                 http_client: httpx.AsyncClient | None = None):
         super(AsyncMostClient, self).__init__()
         self.client_id = client_id
         self.client_secret = client_secret
@@ -46,7 +51,19 @@ class AsyncMostClient(object):
         else:
             self.save_credentials()
 
-        self.session = httpx.AsyncClient()
+        if base_url is None:
+            base_url = os.environ.get("MOST_BASE_URL")
+        if base_url is None:
+            base_url = f"https://api.the-most.ai/api/external"
+
+        if http_client is None:
+            http_client = httpx.AsyncClient(base_url=base_url,
+                                            timeout=timeout,
+                                            follow_redirects=True,
+                                            transport=httpx.AsyncHTTPTransport(retries=max_retries))
+        # self.max_retries = max_retries
+        # self.retry_delay = retry_delay
+        self.session = http_client
         self.access_token = None
         self.model_id = model_id
         self.score_modifier: Optional[ScoreCalculation] = None
@@ -97,7 +114,7 @@ class AsyncMostClient(object):
         return client
 
     async def refresh_access_token(self):
-        resp = await self.session.post("https://api.the-most.ai/api/external/access_token",
+        resp = await self.session.post("/access_token",
                                        json={"client_id": self.client_id,
                                              "client_secret": self.client_secret})
         access_token = resp.json()
@@ -117,9 +134,28 @@ class AsyncMostClient(object):
             return await self.get(url,
                                   headers=headers,
                                   **kwargs)
+
+        if resp.headers.get("Content-Type") == "application/json":
+            raise RuntimeError(resp.json()['message'])
+        resp.raise_for_status()
+        return resp
+
+    async def put(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.put(url,
+                                      headers=headers,
+                                      timeout=None,
+                                      **kwargs)
+        if resp.status_code == 401:
+            await self.refresh_access_token()
+            return await self.put(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.")
+            raise RuntimeError(resp.json()['message'] if resp.headers.get("Content-Type") == "application/json" else "Something went wrong.")
         return resp
 
     async def post(self, url,
@@ -150,7 +186,7 @@ class AsyncMostClient(object):
 
     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",
+            resp = await self.post(f"/{self.client_id}/upload",
                                    files={"audio_file": f})
         return self.retort.load(resp.json(), Audio)
 
@@ -161,36 +197,36 @@ class AsyncMostClient(object):
         f.seek(0)
         if audio_name is None:
             audio_name = uuid.uuid4().hex + ".mp3"
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+        resp = await self.post(f"/{self.client_id}/upload",
                                files={"audio_file": (audio_name, f, 'audio/mp3')})
         return self.retort.load(resp.json(), Audio)
 
     async def upload_text(self, text: str) -> Text:
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_text",
+        resp = await self.post(f"/{self.client_id}/upload_text",
                                json={"text": text})
         return self.retort.load(resp.json(), Text)
 
     async def upload_dialog(self, dialog: Dialog) -> DialogResult:
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_dialog",
+        resp = await self.post(f"/{self.client_id}/upload_dialog",
                                json={"dialog": dialog})
         return self.retort.load(resp.json(), DialogResult)
 
     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",
+        resp = await self.post(f"/{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}")
+        resp = await self.get(f"/{self.client_id}/list?offset={offset}&limit={limit}")
         audio_list = resp.json()
         return self.retort.load(audio_list, List[Audio])
 
     async def list_texts(self,
                          offset: int = 0,
                          limit: int = 10) -> List[Text]:
-        resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list_texts?offset={offset}&limit={limit}")
+        resp = await self.get(f"/{self.client_id}/list_texts?offset={offset}&limit={limit}")
         texts_list = resp.json()
         return self.retort.load(texts_list, List[Text])
 
@@ -198,7 +234,7 @@ class AsyncMostClient(object):
         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}/script")
+        resp = await self.get(f"/{self.client_id}/model/{self.model_id}/script")
         return self.retort.load(resp.json(), Script)
 
     async def get_score_modifier(self):
@@ -206,13 +242,13 @@ class AsyncMostClient(object):
             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}/score_mapping")
+            resp = await self.get(f"/{self.client_id}/model/{self.model_id}/score_mapping")
             score_mapping = self.retort.load(resp.json(), List[ScriptScoreMapping])
             self.score_modifier = ScoreCalculation(score_mapping)
         return self.score_modifier
 
     async def list_models(self):
-        resp = await self.get("https://api.the-most.ai/api/external/list_models")
+        resp = await self.get("/list_models")
         return [self.with_model(model['model'])
                 for model in resp.json()]
 
@@ -224,7 +260,7 @@ class AsyncMostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
+        resp = await self.post(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -238,7 +274,7 @@ class AsyncMostClient(object):
         if not is_valid_id(text_id):
             raise RuntimeError("Please use valid text_id. [try text.id from list_texts()]")
 
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
+        resp = await self.post(f"/{self.client_id}/text/{text_id}/model/{self.model_id}/apply")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -252,7 +288,7 @@ class AsyncMostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
+        resp = await self.post(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_async")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -266,7 +302,7 @@ class AsyncMostClient(object):
         if not is_valid_id(text_id):
             raise RuntimeError("Please use valid text_id. [try audio.id from list_texts()]")
 
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
+        resp = await self.post(f"/{self.client_id}/text/{text_id}/model/{self.model_id}/apply_async")
         result = self.retort.load(resp.json(), Result)
         if modify_scores:
             result = self.score_modifier.modify(result)
@@ -279,7 +315,7 @@ class AsyncMostClient(object):
         if not is_valid_id(audio_id):
             raise RuntimeError("Please use valid audio_id. [try audio.id from list_audios()]")
 
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/model/{self.model_id}/apply_status")
+        resp = await self.post(f"/{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,
@@ -290,7 +326,7 @@ class AsyncMostClient(object):
         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}/results")
+        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)
@@ -303,7 +339,7 @@ class AsyncMostClient(object):
         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}/text")
+        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:
@@ -313,7 +349,18 @@ class AsyncMostClient(object):
         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")
+        resp = await self.get(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog")
+        return self.retort.load(resp.json(), DialogResult)
+
+    async def update_dialog(self, audio_id, dialog: Dialog) -> 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.put(f"/{self.client_id}/audio/{audio_id}/model/{self.model_id}/dialog",
+                              json={"dialog": dialog.to_dict()})
         return self.retort.load(resp.json(), DialogResult)
 
     async def export(self, audio_ids: List[str],
@@ -325,7 +372,7 @@ class AsyncMostClient(object):
         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",
+        resp = await self.get(f"/{self.client_id}/model/{self.model_id}/export",
                               params={'audio_ids': ','.join(audio_ids),
                                       "aggregated_by": aggregated_by,
                                       "aggregation_title": aggregation_title})
@@ -334,7 +381,7 @@ class AsyncMostClient(object):
     async def store_info(self,
                          audio_id: str,
                          data: Dict[str, str]):
-        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/audio/{audio_id}/info",
+        resp = await self.post(f"/{self.client_id}/audio/{audio_id}/info",
                                json={
                                    "data": data,
                                })
@@ -343,7 +390,7 @@ class AsyncMostClient(object):
     async def fetch_info(self, audio_id: str) -> Dict[str, str]:
         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}/info")
+        resp = await self.get(f"/{self.client_id}/audio/{audio_id}/info")
         return self.retort.load(resp.json(), StoredAudioData)
 
     async def __call__(self, audio_path: Path,
@@ -371,7 +418,7 @@ class AsyncMostClient(object):
         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")
+        resp = await self.post(f"/{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
@@ -380,7 +427,7 @@ class AsyncMostClient(object):
                      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",
+        resp = await self.post(f"/{self.client_id}/model/{self.model_id}/search",
                                json={
                                    "query": query,
                                    "filter": filter.to_dict(),

{most_client-1.0.23.dist-info → most_client-1.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: most-client
-Version: 1.0.23
+Version: 1.0.25
 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.25.dist-info/RECORD

@@ -0,0 +1,11 @@
+most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
+most/_constrants.py,sha256=SlHKcBoXwe_sPzk8tdbb7lqhQz-Bfo__FhSoeFWodZE,217
+most/api.py,sha256=t4RqDp-dJUfMayv_0FtBD8yKaGVZpnixCc7D7-UFSKY,17658
+most/async_api.py,sha256=dBZIrzyPfstFMysxVNfrQdd4WDabBIsWMu2ogLsnPb4,18676
+most/score_calculation.py,sha256=1XU1LfIH5LSCwAbAaKkr-EjH5qOTXrJKOUvhCCawka4,1054
+most/types.py,sha256=Qgyv261J8b1cfbmeITz1C9QgkoCMGQQd_L4t4M3dd6M,3603
+most_client-1.0.25.dist-info/METADATA,sha256=RmwkJHqjnOd8hyYX4Za8puXebVgR0Ehq1wo0ODUI2m8,1027
+most_client-1.0.25.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
+most_client-1.0.25.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.25.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.25.dist-info/RECORD,,

{most_client-1.0.23.dist-info → most_client-1.0.25.dist-info}/WHEEL

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

most_client-1.0.23.dist-info/RECORD

@@ -1,10 +0,0 @@
-most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
-most/api.py,sha256=Fp9MeADwnjwf8_BYbleIKB0gFfsK3GNsNZDHRnqgOms,16256
-most/async_api.py,sha256=7kFpsU6yCAeEjOroXL-t_m1RIOwLqmMyPq9tPhoUIjw,17243
-most/score_calculation.py,sha256=1XU1LfIH5LSCwAbAaKkr-EjH5qOTXrJKOUvhCCawka4,1054
-most/types.py,sha256=Qgyv261J8b1cfbmeITz1C9QgkoCMGQQd_L4t4M3dd6M,3603
-most_client-1.0.23.dist-info/METADATA,sha256=pBQkXH2Ogl8Ot9SWv8gLnmB8r7j1t74eY3X6oCoC63M,1027
-most_client-1.0.23.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-most_client-1.0.23.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.23.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.23.dist-info/RECORD,,