most-client 1.0.42__tar.gz → 1.0.43__tar.gz

{most_client-1.0.42/most_client.egg-info → most_client-1.0.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: most-client
-Version: 1.0.42
+Version: 1.0.43
 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.42 → most_client-1.0.43}/most/__init__.py

@@ -4,4 +4,9 @@ from .trainer_api import Trainer
 from .async_trainer_api import AsyncTrainer
 from .searcher import MostSearcher
 from .async_searcher import AsyncMostClient
-from .search_types import SearchParams, IDCondition, ChannelsCondition, DurationCondition, ResultsCondition, StoredInfoCondition, TagsCondition
+from .search_types import SearchParams, IDCondition, ChannelsCondition, DurationCondition, ResultsCondition, StoredInfoCondition, TagsCondition
+from .glossary import Glossary
+from .async_glossary import AsyncGlossary
+from .catalog import Catalog
+from .async_catalog import AsyncCatalog
+from .types import GlossaryNGram, Item

most_client-1.0.43/most/async_catalog.py

@@ -0,0 +1,28 @@
+from typing import List
+
+from .async_api import AsyncMostClient
+from .types import Item
+
+
+class AsyncCatalog(object):
+    def __init__(self, client: AsyncMostClient):
+        super(AsyncCatalog, self).__init__()
+        self.client = client
+
+    async def add_items(self, items: List[Item] | Item) -> List[Item]:
+        if not isinstance(items, list):
+            items = [items]
+        resp = await self.client.post(f"/{self.client.client_id}/upload_items",
+                                      json=[item.to_dict() for item in items])
+        return self.client.retort.load(resp.json(), List[Item])
+
+    async def list_items(self) -> List[Item]:
+        resp = await self.client.get(f"/{self.client.client_id}/items")
+        return self.client.retort.load(resp.json(), List[Item])
+
+    async def delete_items(self, item_ids: List[str]):
+        if not isinstance(item_ids, list):
+            item_ids = [item_ids]
+        await self.client.post(f"/{self.client.client_id}/delete_items",
+                               json=item_ids)
+        return self

most_client-1.0.43/most/async_glossary.py

@@ -0,0 +1,32 @@
+from typing import List
+
+from .async_api import AsyncMostClient
+from .types import GlossaryNGram
+
+
+class AsyncGlossary(object):
+    def __init__(self, client: AsyncMostClient):
+        super(AsyncGlossary, self).__init__()
+        self.client = client
+
+    async def add_ngrams(self, ngrams: List[GlossaryNGram] | GlossaryNGram) -> List[GlossaryNGram]:
+        if not isinstance(ngrams, list):
+            ngrams = [ngrams]
+        resp = await self.client.post(f"/{self.client.client_id}/upload_glossary",
+                                      json=[ngram.to_dict() for ngram in ngrams])
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    async def list_ngrams(self) -> List[GlossaryNGram]:
+        resp = await self.client.get(f"/{self.client.client_id}/glossary")
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    async def del_ngrams(self, ngram_ids: List[str] | str):
+        if not isinstance(ngram_ids, list):
+            ngram_ids = [ngram_ids]
+        await self.client.post(f"/{self.client.client_id}/delete_glossary_ngrams",
+                               json=ngram_ids)
+        return self
+
+    async def drop(self):
+        await self.client.post(f"/{self.client.client_id}/delete_glossary")
+        return self

most_client-1.0.43/most/catalog.py

@@ -0,0 +1,28 @@
+from typing import List
+
+from .api import MostClient
+from .types import Item
+
+
+class Catalog(object):
+    def __init__(self, client: MostClient):
+        super(Catalog, self).__init__()
+        self.client = client
+
+    def add_items(self, items: List[Item] | Item) -> List[Item]:
+        if not isinstance(items, list):
+            items = [items]
+        resp = self.client.post(f"/{self.client.client_id}/upload_items",
+                                json=[item.to_dict() for item in items])
+        return self.client.retort.load(resp.json(), List[Item])
+
+    def list_items(self) -> List[Item]:
+        resp = self.client.get(f"/{self.client.client_id}/items")
+        return self.client.retort.load(resp.json(), List[Item])
+
+    def delete_items(self, item_ids: List[str]):
+        if not isinstance(item_ids, list):
+            item_ids = [item_ids]
+        self.client.post(f"/{self.client.client_id}/delete_items",
+                         json=item_ids)
+        return self

most_client-1.0.43/most/glossary.py

@@ -0,0 +1,32 @@
+from typing import List
+
+from .api import MostClient
+from .types import GlossaryNGram
+
+
+class Glossary(object):
+    def __init__(self, client: MostClient):
+        super(Glossary, self).__init__()
+        self.client = client
+
+    def add_ngrams(self, ngrams: List[GlossaryNGram] | GlossaryNGram) -> List[GlossaryNGram]:
+        if not isinstance(ngrams, list):
+            ngrams = [ngrams]
+        resp = self.client.post(f"/{self.client.client_id}/upload_glossary",
+                                json=[ngram.to_dict() for ngram in ngrams])
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    def list_ngrams(self) -> List[GlossaryNGram]:
+        resp = self.client.get(f"/{self.client.client_id}/glossary")
+        return self.client.retort.load(resp.json(), List[GlossaryNGram])
+
+    def del_ngrams(self, ngram_ids: List[str] | str):
+        if not isinstance(ngram_ids, list):
+            ngram_ids = [ngram_ids]
+        self.client.post(f"/{self.client.client_id}/delete_glossary_ngrams",
+                         json=ngram_ids)
+        return self
+
+    def drop(self):
+        self.client.post(f"/{self.client.client_id}/delete_glossary")
+        return self

{most_client-1.0.42 → most_client-1.0.43}/most/types.py

@@ -131,8 +131,18 @@ class Dialog(DataClassJsonMixin):
 @dataclass_json
 @dataclass
 class GlossaryNGram(DataClassJsonMixin):
-    pronunciation: List[str]
     original: List[str]
+    pronunciation: List[str]
+    weight: float = 2
+    id: Optional[str] = None
+
+
+@dataclass_json
+@dataclass
+class Item(DataClassJsonMixin):
+    title: str
+    pronunciation: str
+    metadata: Dict[str, str]
     id: Optional[str] = None
 
 

{most_client-1.0.42 → most_client-1.0.43/most_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: most-client
-Version: 1.0.42
+Version: 1.0.43
 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.42 → most_client-1.0.43}/most_client.egg-info/SOURCES.txt

@@ -6,9 +6,11 @@ most/__init__.py
 most/_constrants.py
 most/api.py
 most/async_api.py
+most/async_catalog.py
 most/async_glossary.py
 most/async_searcher.py
 most/async_trainer_api.py
+most/catalog.py
 most/glossary.py
 most/score_calculation.py
 most/search_types.py

{most_client-1.0.42 → most_client-1.0.43}/setup.py

@@ -8,7 +8,7 @@ with open('requirements.txt', 'r') as f:
 
 setup(
     name='most-client',
-    version='1.0.42',
+    version='1.0.43',
     python_requires=f'>=3.6',
     description='Most AI API for https://the-most.ai',
     url='https://github.com/the-most-ai/most-client',

most_client-1.0.42/most/async_glossary.py

@@ -1,18 +0,0 @@
-from typing import List
-
-from .async_api import AsyncMostClient
-from .types import GlossaryNGram
-
-
-class AsyncGlossary(object):
-    def __init__(self, client: AsyncMostClient):
-        self.client = client
-
-    async def add_ngram(self, ngram: GlossaryNGram):
-        pass
-
-    async def list_ngrams(self) -> List[GlossaryNGram]:
-        return []
-
-    async def del_ngram(self, ngram_id: str):
-        pass

most_client-1.0.42/most/glossary.py

@@ -1,18 +0,0 @@
-from typing import List
-
-from .api import MostClient
-from .types import GlossaryNGram
-
-
-class Glossary(object):
-    def __init__(self, client: MostClient):
-        self.client = client
-
-    def add_ngram(self, ngram: GlossaryNGram):
-        pass
-
-    def list_ngrams(self) -> List[GlossaryNGram]:
-        return []
-
-    def del_ngram(self, ngram_id: str):
-        pass