agi-med-common 5.0.24__tar.gz → 5.0.26__tar.gz

{agi_med_common-5.0.24 → agi_med_common-5.0.26}/PKG-INFO

@@ -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 → agi_med_common-5.0.26}/src/agi_med_common/__init__.py

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

{agi_med_common-5.0.24 → agi_med_common-5.0.26}/src/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-5.0.24 → agi_med_common-5.0.26}/src/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 → agi_med_common-5.0.26}/src/agi_med_common.egg-info/PKG-INFO

@@ -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