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

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.25"
+__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
@@ -10,7 +10,6 @@ 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"
@@ -154,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"
@@ -163,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"
@@ -192,6 +203,15 @@ class Chat(_Base):
         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.25.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.25
+Version: 5.0.26
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

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

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=_cVr0Xpyf_UeT_iBf-t76dvcwpgFjzri5yywDp08rd8,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=Tc-qPDvmLOkKalHEsx9OX2qQ6o4RCcdod0ciHune4Q8,10115
-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.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,,
+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,,