agi-med-common 5.0.24__py3-none-any.whl → 5.0.26__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.24"
+__version__ = "5.0.26"
 
 from .models import (
     MTRSLabelEnum,

agi_med_common/models/chat.py

@@ -1,7 +1,7 @@
 import warnings
 from copy import deepcopy
 from datetime import datetime
-from typing import Any, List, Dict, Literal, TypeVar
+from typing import Any, List, Dict, Literal, TypeVar, Callable
 
 from agi_med_common.models.chat_item import ChatItem, ReplicaItem, OuterContextItem
 from agi_med_common.models.widget import Widget
@@ -17,7 +17,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 +76,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 +125,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
@@ -137,6 +153,14 @@ class BaseMessage(_Base):
     def with_now_datetime(self):
         return self.model_copy(update=dict(date_time=now_pretty()))
 
+    @property
+    def is_ai(self):
+        return self.type == "ai"
+
+    @property
+    def is_human(self):
+        return self.type == "human"
+
 
 class HumanMessage(BaseMessage):
     type: Literal["human"] = "human"
@@ -146,6 +170,10 @@ class AIMessage(BaseMessage):
     type: Literal["ai"] = "ai"
     state: str = Field("", examples=["COLLECTION"])
 
+    @property
+    def action(self) -> str:
+        return self.extra.get("action", "")
+
 
 class MiscMessage(BaseMessage):
     type: Literal["misc"] = "misc"
@@ -168,6 +196,22 @@ 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 replace_messages(self, messages: List[ChatMessage]):
+        return self.model_copy(update=dict(messages=messages))
+
+    def get_last_state(self, default: str = "empty") -> str:
+        for msg in range(self.messages - 1, -1, -1):
+            if isinstance(msg, AIMessage):
+                return msg.state
+        return default
+
 
 def convert_replica_item_to_message(replica: ReplicaItem) -> ChatMessage:
     # legacy: eliminate after migration

agi_med_common/models/chat_item.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Annotated, Any, List
+from typing import Annotated, Any, List, Callable
 
 from agi_med_common.models.widget import Widget
 from pydantic import Field, ConfigDict, BeforeValidator, AfterValidator
@@ -92,6 +92,18 @@ class ReplicaItem(_Base):
     def with_now_datetime(self):
         return self.model_copy(update=dict(date_time=now_pretty()))
 
+    @property
+    def is_ai(self):
+        return self.role
+
+    @property
+    def is_human(self):
+        return not self.role
+
+    def modify_text(self, callback: Callable[[str], str]) -> "ReplicaItem":
+        body_upd = callback(self.body)
+        return self.model_copy(update=dict(body=body_upd))
+
 
 class InnerContextItem(_Base):
     replicas: list[ReplicaItem] = Field(alias="Replicas")
@@ -118,6 +130,9 @@ class ChatItem(_Base):
         for replica in replicas:
             self.inner_context.replicas.append(replica)
 
+    def replace_replicas(self, replicas: List[ReplicaItem]):
+        return self.model_copy(update=dict(inner_context=InnerContextItem(replicas=replicas)))
+
     def zip_history(self, field: str) -> list[Any]:
         return [replica.to_dict().get(field, None) for replica in self.inner_context.replicas]
 

{agi_med_common-5.0.24.dist-info → agi_med_common-5.0.26.dist-info}/METADATA

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

{agi_med_common-5.0.24.dist-info → agi_med_common-5.0.26.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=MKI-hB8aFY-Yktyicq9m-8w-pX37IyFztVaKV5Q3Ipk,740
+agi_med_common/__init__.py,sha256=3iqTNDjcukCgm6cnLY8vQaom03mBn2KH-MPxaVOM4y0,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=0VXu-SUPlOeGRO8SMCFPfGoAiVoPIm_v3hKMG-nGhNI,9230
-agi_med_common/models/chat_item.py,sha256=bg_l6B4RAqiSO84vF50nQN7kYCkLB6JHAzItwm1-RzM,4791
+agi_med_common/models/chat.py,sha256=1GVcwOM97UMrsiPg3fAd9xJe2p9cHouidWgIPMWUxc0,10668
+agi_med_common/models/chat_item.py,sha256=eTUBCLlTVPqIS64qnaPB92vCJyiBixIPi8ochrgOPHo,5262
 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.24.dist-info/METADATA,sha256=2stOgKcSCPgY9VKJ4XJLKmAFnsyCXchTasY9WTaGgks,518
-agi_med_common-5.0.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-5.0.24.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-5.0.24.dist-info/RECORD,,
+agi_med_common-5.0.26.dist-info/METADATA,sha256=wlJpGlj3NNyi2UvSjBx9z8474Nysj20mMfbOmVyhNO0,518
+agi_med_common-5.0.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.0.26.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.0.26.dist-info/RECORD,,