mmar-mapi 1.0.8__py3-none-any.whl → 1.0.10__py3-none-any.whl

mmar_mapi/__init__.py

@@ -15,9 +15,10 @@ from .models.chat_item import ChatItem, OuterContextItem, InnerContextItem, Repl
 from .models.enums import MTRSLabelEnum, DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 from .models.tracks import TrackInfo, DomainInfo
 from .models.widget import Widget
-from .utils import make_session_id
+from .utils import make_session_id, chunked
 from .xml_parser import XMLParser
 
+
 __all__ = [
     "AIMessage",
     "Base",
@@ -42,6 +43,7 @@ __all__ = [
     "TrackInfo",
     "Widget",
     "XMLParser",
+    "chunked",
     "make_content",
     "make_session_id",
 ]

mmar_mapi/models/chat.py

@@ -249,6 +249,9 @@ class Chat(Base):
         message = messages[-1]
         return message if isinstance(message, HumanMessage) else None
 
+    def count_messages(self, func: Callable[[ChatMessage], bool]) -> int:
+        return sum(map(func, self.messages))
+
 
 def make_content(
     text: str | None = None,

mmar_mapi/models/widget.py

@@ -1,6 +1,6 @@
 from typing import Self, Literal
 
-from more_itertools import chunked
+from mmar_mapi.utils import chunked
 from pydantic import BaseModel, model_validator
 
 

mmar_mapi/utils.py

@@ -1,5 +1,16 @@
+from itertools import islice
+from collections.abc import Iterable
 from datetime import datetime
 
 
 def make_session_id() -> str:
     return f"{datetime.now():%Y-%m-%d--%H-%M-%S}"
+
+
+def chunked(items: Iterable, n) -> list:
+    "behavior like in more_itertools.chunked"
+    iterator = iter(items)
+    res = []
+    while chunk := list(islice(iterator, n)):
+        res.append(chunk)
+    return res

{mmar_mapi-1.0.8.dist-info → mmar_mapi-1.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mmar-mapi
-Version: 1.0.8
+Version: 1.0.10
 Summary: Common pure/IO utilities for multi-modal architectures team
 Keywords: 
 Author: Eugene Tagin

{mmar_mapi-1.0.8.dist-info → mmar_mapi-1.0.10.dist-info}/RECORD

@@ -1,17 +1,17 @@
-mmar_mapi/__init__.py,sha256=hhLEK5u9NL1Du3X1M0SzmMPYYq5FIUcbiVNL2_jCZYc,1084
+mmar_mapi/__init__.py,sha256=NxCsA3aMKjxIZ2f4pQTJ1LPpg7t2yz7PsiDeb_HrAJE,1109
 mmar_mapi/api.py,sha256=C9Sr8dISvf51xfEznPjccI_odaG4coQE3HI_0jVpjMQ,1677
 mmar_mapi/file_storage.py,sha256=GbahBabBdAKjlAnv1MszERUxxZyA9HGMiR9tz2a9dgY,4409
 mmar_mapi/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mmar_mapi/models/base.py,sha256=mKtXV2x51XVj7W-et9tjGcPMDUUUMelW-BywMgFc2p0,411
-mmar_mapi/models/chat.py,sha256=DtA_YoRSi6TkhuUfKRvYpc3XMWzw3fWtCZiIdVwDfxo,13535
+mmar_mapi/models/chat.py,sha256=HC-zl32wu5xMyam3iZXZKc4vSfxtc4f-vMb9FgB8m1Q,13655
 mmar_mapi/models/chat_item.py,sha256=ZfCKvTqr7gpuJSAuHVxWRnlTefRwki_IVNA2N_CXGdg,5557
 mmar_mapi/models/enums.py,sha256=J-GNpql9MCnKnWiV9aJRQGI-pAybvV86923RZs99grA,1006
 mmar_mapi/models/tracks.py,sha256=HKDp-BX1p7AlDfSEKfOKCu0TRSK9cD4Dmq1vJt8oRjw,307
-mmar_mapi/models/widget.py,sha256=ObQlY24nJ1buQ501eFoVFSHe5HcvnuQd_x2hrDevDpo,1354
+mmar_mapi/models/widget.py,sha256=ue5o4AkN8SG09aA8eQLOOxbwah-mkZir0ZGL6OA8S9Q,1355
 mmar_mapi/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
-mmar_mapi/utils.py,sha256=hcKJVslvTBLw2vjZ9zcKZxh_tqk48obHcVs_i3Rxn3M,112
+mmar_mapi/utils.py,sha256=oVrEAHGcfFBprwdTEzSbFCpdLtkDRVzWtOnYpS439os,394
 mmar_mapi/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
-mmar_mapi-1.0.8.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
-mmar_mapi-1.0.8.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
-mmar_mapi-1.0.8.dist-info/METADATA,sha256=KJxwmuH9z4h1obMo6xKup42cKxKdahUMuMTT70COKnQ,914
-mmar_mapi-1.0.8.dist-info/RECORD,,
+mmar_mapi-1.0.10.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
+mmar_mapi-1.0.10.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
+mmar_mapi-1.0.10.dist-info/METADATA,sha256=cL99l0W1GOtgnJOUNDyflzdkbMNHNfGwNpdiLnhFIz0,915
+mmar_mapi-1.0.10.dist-info/RECORD,,