mmar-mapi 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

mmar_mapi/__init__.py

@@ -1,12 +1,47 @@
-__version__ = "5.2.4"
-
 from .file_storage import FileStorage, ResourceId
 from .models.base import Base
-from .models.base_config_models import GigaChatConfig
-from .models.chat import Chat, Context, ChatMessage, AIMessage, HumanMessage, MiscMessage, make_content, Content
+from .models.chat import (
+    Chat,
+    Context,
+    ChatMessage,
+    AIMessage,
+    HumanMessage,
+    MiscMessage,
+    make_content,
+    Content,
+    BaseMessage,
+)
 from .models.chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
 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 .xml_parser import XMLParser
+
+__all__ = [
+    "AIMessage",
+    "Base",
+    "BaseMessage",
+    "Chat",
+    "ChatItem",
+    "ChatMessage",
+    "Content",
+    "Context",
+    "DiagnosticsXMLTagEnum",
+    "DoctorChoiceXMLTagEnum",
+    "DomainInfo",
+    "FileStorage",
+    "HumanMessage",
+    "InnerContextItem",
+    "MTRSLabelEnum",
+    "MTRSXMLTagEnum",
+    "MiscMessage",
+    "OuterContextItem",
+    "ReplicaItem",
+    "ResourceId",
+    "TrackInfo",
+    "Widget",
+    "XMLParser",
+    "make_content",
+    "make_session_id",
+]

mmar_mapi/models/chat.py

@@ -1,13 +1,13 @@
 import warnings
+from collections.abc import Callable
 from copy import deepcopy
 from datetime import datetime
 from typing import Any, Literal, TypeVar
-from collections.abc import Callable
 
 from mmar_mapi.models.chat_item import ChatItem, ReplicaItem, OuterContextItem
 from mmar_mapi.models.widget import Widget
 from mmar_mapi.type_union import TypeUnion
-from pydantic import Field, ValidationError, Extra
+from pydantic import Field, ValidationError
 
 from .base import Base
 
@@ -146,10 +146,6 @@ class BaseMessage(Base):
     def widget(self) -> Widget | None:
         return _get_widget(self.content)
 
-    @staticmethod
-    def DATETIME_FORMAT() -> str:
-        return _DT_FORMAT
-
     def with_now_datetime(self):
         return self.model_copy(update=dict(date_time=now_pretty()))
 
@@ -161,6 +157,21 @@ class BaseMessage(Base):
     def is_human(self):
         return self.type == "human"
 
+    @staticmethod
+    def DATETIME_FORMAT() -> str:
+        return _DT_FORMAT
+
+    @staticmethod
+    def find_resource_id(msg: "BaseMessage", ext: str | None = None, type: str=None) -> str | None:
+        resource_id = msg.resource_id
+        if type and type != msg.type:
+            return None
+        if not resource_id:
+            return None
+        if ext and not resource_id.endswith(ext):
+            return None
+        return resource_id
+
 
 class HumanMessage(BaseMessage):
     type: Literal["human"] = "human"
@@ -182,15 +193,18 @@ class MiscMessage(BaseMessage):
 ChatMessage = TypeUnion[HumanMessage, AIMessage, MiscMessage]
 
 
+def find_in_messages(messages: list[ChatMessage], func: Callable[[ChatMessage], T | None]) -> T | None:
+    return next(filter(None, map(func, messages)), None)
+
+
 class Chat(Base):
     context: Context = Field(default_factory=Context)
     messages: list[ChatMessage] = Field(default_factory=list)
 
-    class Config:
-        extra = Extra.ignore
+    model_config = {"extra": "ignore"}
 
     def __init__(self, **data):
-        extra_fields = set(data.keys()) - set(self.__fields__.keys())
+        extra_fields = set(data.keys()) - set(type(self).model_fields.keys())
         if extra_fields:
             warnings.warn(f"Chat initialization: extra fields will be ignored: {extra_fields}")
         super().__init__(**data)
@@ -222,6 +236,12 @@ class Chat(Base):
                 return message.state
         return default
 
+    def find_in_messages(self, func: Callable[[ChatMessage], T | None]) -> T | None:
+        return find_in_messages(self.messages, func)
+
+    def rfind_in_messages(self, func: Callable[[ChatMessage], T | None]) -> T | None:
+        return find_in_messages(self.messages[::-1], func)
+
 
 def make_content(
     text: str | None = None,

mmar_mapi/models/widget.py

@@ -1,5 +1,6 @@
 from typing import Self, Literal
 
+from more_itertools import chunked
 from pydantic import BaseModel, model_validator
 
 
@@ -20,3 +21,14 @@ class Widget(BaseModel):
                     continue
                 raise ValueError(f"Expected buttons like `<callback>:<caption>`, found: {btn}")
         return self
+
+    @staticmethod
+    def make_inline_buttons(ibuttons: dict[str, str], by=1) -> "Widget":
+        return _make_inline_buttons(ibuttons=ibuttons, by=by)
+
+
+def _make_inline_buttons(ibuttons: dict[str, str], by=1) -> "Widget":
+    ibs0 = [f"{key}:{val}" for key, val in ibuttons.items()]
+    ibs = list(chunked(ibs0, n=by))
+    res = Widget(ibuttons=ibs)
+    return res

{mmar_mapi-1.0.6.dist-info → mmar_mapi-1.0.7.dist-info}/METADATA

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

{mmar_mapi-1.0.6.dist-info → mmar_mapi-1.0.7.dist-info}/RECORD

@@ -1,19 +1,17 @@
-mmar_mapi/__init__.py,sha256=cj3McZDG8zU1nLSMjwtVHG0u99xN7LaNLW90d4tK7Io,613
+mmar_mapi/__init__.py,sha256=hhLEK5u9NL1Du3X1M0SzmMPYYq5FIUcbiVNL2_jCZYc,1084
 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/base_config_models/__init__.py,sha256=KjS_bSCka8BOMsigwcIML-e6eNB2ouMU6gxlhRmzeuY,44
-mmar_mapi/models/base_config_models/gigachat_config.py,sha256=QvKTnp0VioXzd3_PUgNBeYx5RDTZT-45j-Ll-wXEI_o,421
-mmar_mapi/models/chat.py,sha256=sAYuudA--i7qVfEERy5K5wmAkIFAjn6vUrKRT_BmfIE,12480
+mmar_mapi/models/chat.py,sha256=IsIYoHY3Taxnfn-Audeml-WLGIbJgKr_-Arg-ZA8FIE,13293
 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=KXSP3d4yQNeificMfzRxROdByw9IkxwCjkDD0kc7tcQ,704
+mmar_mapi/models/widget.py,sha256=pQHiOukNLzsrz5lr5ptMeARPxSzJhJnijA0rVpVSIhk,1108
 mmar_mapi/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
 mmar_mapi/utils.py,sha256=hcKJVslvTBLw2vjZ9zcKZxh_tqk48obHcVs_i3Rxn3M,112
 mmar_mapi/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
-mmar_mapi-1.0.6.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
-mmar_mapi-1.0.6.dist-info/WHEEL,sha256=Jb20R3Ili4n9P1fcwuLup21eQ5r9WXhs4_qy7VTrgPI,79
-mmar_mapi-1.0.6.dist-info/METADATA,sha256=dxEk5xdH21b8jW053i5ojhUkguzCkgJoZMRc_JIRyAo,914
-mmar_mapi-1.0.6.dist-info/RECORD,,
+mmar_mapi-1.0.7.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
+mmar_mapi-1.0.7.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
+mmar_mapi-1.0.7.dist-info/METADATA,sha256=GnTR9S4PdQJak3j-3tIJ61xrA7bU1AjubCe-vRWkL3I,914
+mmar_mapi-1.0.7.dist-info/RECORD,,

{mmar_mapi-1.0.6.dist-info → mmar_mapi-1.0.7.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.8.15
+Generator: uv 0.8.17
 Root-Is-Purelib: true
 Tag: py3-none-any

mmar_mapi/models/base_config_models/__init__.py

@@ -1 +0,0 @@
-from .gigachat_config import GigaChatConfig

mmar_mapi/models/base_config_models/gigachat_config.py

@@ -1,14 +0,0 @@
-from typing import Self
-
-from pydantic import BaseModel, model_validator, SecretStr
-
-
-class GigaChatConfig(BaseModel):
-    client_id: SecretStr = SecretStr("")
-    client_secret: SecretStr = SecretStr("")
-
-    @model_validator(mode="after")
-    def empty_validator(self) -> Self:
-        if not (self.client_id and self.client_secret):
-            raise ValueError("creds for gigachat is not filled")
-        return self