agi-med-common 5.0.23__py3-none-any.whl → 5.0.25__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.23"
+__version__ = "5.0.25"
 
 from .models import (
     MTRSLabelEnum,

agi_med_common/models/chat.py

@@ -10,6 +10,7 @@ from agi_med_common.utils import first_nonnull
 from pydantic import Field, ValidationError
 
 from ._base import _Base
+from typing import Callable
 
 
 _DT_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
@@ -17,7 +18,7 @@ _EXAMPLE_DT: str = datetime(year=1970, month=1, day=1).strftime(_DT_FORMAT)
 StrDict = Dict[str, Any]
 ContentBase = str | Widget | StrDict
 Content = ContentBase | List[ContentBase]
-T = TypeVar('T')
+T = TypeVar("T")
 
 
 def now_pretty() -> str:
@@ -76,6 +77,18 @@ def _get_text(obj: Content) -> str:
     return ""
 
 
+def _modify_text(obj: Content, callback: Callable[[str], str | None]) -> str:
+    if isinstance(obj, str):
+        return callback(obj)
+    if isinstance(obj, list):
+        return [_modify_text(el, callback) for el in obj]
+    if isinstance(obj, dict) and obj.get("type") == "text":
+        text = _get_field(obj, "text", str) or ""
+        text_upd = callback(text)
+        return {"type": "text", "text": text_upd}
+    return obj
+
+
 def _get_resource_id(obj: Content) -> str | None:
     if isinstance(obj, list):
         return first_nonnull(map(_get_resource_id, obj))
@@ -113,6 +126,10 @@ class BaseMessage(_Base):
     def text(self) -> str:
         return _get_text(self.content)
 
+    def modify_text(self, callback: Callable[[str], str]) -> "BaseMessage":
+        content_upd = _modify_text(self.content, callback)
+        return self.model_copy(update=dict(content=content_upd))
+
     @property
     def body(self) -> str:
         # legacy: eliminate after migration
@@ -168,6 +185,13 @@ class Chat(_Base):
     def to_chat_item(self) -> ChatItem:
         return convert_chat_to_chat_item(self)
 
+    def add_message(self, message: ChatMessage):
+        self.messages.append(message)
+
+    def add_messages(self, messages: List[ChatMessage]):
+        for message in messages:
+            self.messages.append(message)
+
 
 def convert_replica_item_to_message(replica: ReplicaItem) -> ChatMessage:
     # legacy: eliminate after migration
@@ -253,7 +277,7 @@ def convert_message_to_replica_item(message: ChatMessage) -> ReplicaItem | None:
     else:
         return None
 
-    extra = deepcopy(message.extra) if message.extra or {}
+    extra = deepcopy(message.extra) if message.extra else {}
     action = extra.pop("action", "")
     moderation = extra.pop("moderation", "OK")
 

{agi_med_common-5.0.23.dist-info → agi_med_common-5.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 5.0.23
+Version: 5.0.25
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-5.0.23.dist-info → agi_med_common-5.0.25.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=ygwhdPR5f4EnhXs8JSK6yncgDCSnScoDWxivgcxp_TE,740
+agi_med_common/__init__.py,sha256=_cVr0Xpyf_UeT_iBf-t76dvcwpgFjzri5yywDp08rd8,740
 agi_med_common/api.py,sha256=kGLKiebbF91s_7V2S55ETS7R6tQgZNkoFMoawdjeMYw,1860
 agi_med_common/api_v2.py,sha256=gj6BPEAOvpT6GHWeE7bzmKIR-pvq0yORZ3L0ADvI8ps,1828
 agi_med_common/file_storage.py,sha256=T0Hbs4W-pWO6HdWcmlVqABrQHYdq7lLZH4_Vu-YNVbw,1802
@@ -9,14 +9,14 @@ agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
 agi_med_common/models/__init__.py,sha256=dqr2kP-RuxFfAZhCr103PQzTVZFKIcdxyzTYiHhdTsE,375
 agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-agi_med_common/models/chat.py,sha256=ag0Q3YaOf_7YMJS_pWLdVxHvTey16v-P51Jq3AjkmLU,9228
+agi_med_common/models/chat.py,sha256=Tc-qPDvmLOkKalHEsx9OX2qQ6o4RCcdod0ciHune4Q8,10115
 agi_med_common/models/chat_item.py,sha256=bg_l6B4RAqiSO84vF50nQN7kYCkLB6JHAzItwm1-RzM,4791
 agi_med_common/models/enums.py,sha256=J-GNpql9MCnKnWiV9aJRQGI-pAybvV86923RZs99grA,1006
 agi_med_common/models/tracks.py,sha256=UP-jeWqDiCK6dyoMDfs7hemgl_xsJKee_DApjBf-XYc,311
 agi_med_common/models/widget.py,sha256=aJZ2vWx_PTFN02Wz16eokz9IIVrxqNuZYVDqLG36toE,710
 agi_med_common/models/base_config_models/__init__.py,sha256=KjS_bSCka8BOMsigwcIML-e6eNB2ouMU6gxlhRmzeuY,44
 agi_med_common/models/base_config_models/gigachat_config.py,sha256=WNSCTO8Fjpxc1v2LRUHfKqo9aeMDpXltTHYBFgTD2N0,422
-agi_med_common-5.0.23.dist-info/METADATA,sha256=kq3tcdnb2NwO5mHGtAmPTFa2ulZTMnnaqK_tVYni4k4,518
-agi_med_common-5.0.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-5.0.23.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-5.0.23.dist-info/RECORD,,
+agi_med_common-5.0.25.dist-info/METADATA,sha256=ESt2Yy4UvU-c9T35RdeX4kX4APEusfOY2C0QpJXEsow,518
+agi_med_common-5.0.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.0.25.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.0.25.dist-info/RECORD,,