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

agi_med_common/__init__.py

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

agi_med_common/api.py

@@ -1,6 +1,6 @@
 from typing import List, Tuple
 
-from agi_med_common.models import ChatItem, ReplicaItem, DomainInfo, TrackInfo
+from agi_med_common.models import Chat, ChatMessage, DomainInfo, TrackInfo
 from pydantic import BaseModel
 
 
@@ -16,12 +16,12 @@ class ChatManagerAPI:
     def get_tracks(self, language_code: str, client_id: str) -> List[TrackInfo]:
         raise NotImplementedError
 
-    def get_response(self, chat: ChatItem, request_id: str = "") -> List[ReplicaItem]:
+    def get_response(self, chat: Chat, request_id: str = "") -> List[ChatMessage]:
         raise NotImplementedError
 
 
 class TextGeneratorAPI:
-    def process(self, chat: ChatItem, request_id: str = "") -> str:
+    def process(self, chat: Chat, request_id: str = "") -> str:
         raise NotImplementedError
 
 
@@ -37,7 +37,7 @@ class ContentInterpreterRemoteAPI:
         kind: str,
         query: str,
         resource: bytes,
-        chat: ChatItem | None = None,
+        chat: Chat | None = None,
         request_id: str = "",
     ) -> ContentInterpreterRemoteResponse:
         raise NotImplementedError
@@ -47,22 +47,22 @@ class ClassifierAPI:
     def get_values(self) -> List[Value]:
         raise NotImplementedError
 
-    def evaluate(self, chat: ChatItem, request_id: str = "") -> Value:
+    def evaluate(self, chat: Chat, request_id: str = "") -> Value:
         raise NotImplementedError
 
 
 class CriticAPI:
-    def evaluate(self, text: str, chat: ChatItem | None = None, request_id: str = "") -> float:
+    def evaluate(self, text: str, chat: Chat | None = None, request_id: str = "") -> float:
         raise NotImplementedError
 
 
 class ContentInterpreterAPI:
     def interpret(
-        self, kind: str, query: str, resource_id: str = "", chat: ChatItem | None = None, request_id: str = ""
+        self, kind: str, query: str, resource_id: str = "", chat: Chat | None = None, request_id: str = ""
     ) -> Tuple[Interpretation, ResourceId | None]:
         raise NotImplementedError
 
 
 class TextProcessorAPI:
-    def process(self, text: str, chat: ChatItem | None = None, request_id: str = "") -> str:
+    def process(self, text: str, chat: Chat | None = None, request_id: str = "") -> str:
         raise NotImplementedError

agi_med_common/models/chat.py

@@ -207,9 +207,10 @@ class Chat(_Base):
         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
+        for ii in range(len(self.messages) - 1, -1, -1):
+            message = self.messages[ii]
+            if isinstance(message, AIMessage):
+                return message.state
         return default
 
 

agi_med_common/models/chat_item.py

@@ -133,6 +133,15 @@ class ChatItem(_Base):
     def replace_replicas(self, replicas: List[ReplicaItem]):
         return self.model_copy(update=dict(inner_context=InnerContextItem(replicas=replicas)))
 
+    def get_last_state(self, default: str = "empty") -> str:
+        replicas = self.inner_context.replicas
+        for ii in range(len(replicas) - 1, -1, -1):
+            replica = replicas[ii]
+            if replica.role:
+                return replica.state
+        return default
+
+
     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.26.dist-info → agi_med_common-5.1.0.dist-info}/METADATA

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

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

@@ -1,6 +1,5 @@
-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/__init__.py,sha256=8FdoVXkVoK3EoCnW_WIYfEePasHves8Xw7q1Jj6_8wM,739
+agi_med_common/api.py,sha256=gj6BPEAOvpT6GHWeE7bzmKIR-pvq0yORZ3L0ADvI8ps,1828
 agi_med_common/file_storage.py,sha256=T0Hbs4W-pWO6HdWcmlVqABrQHYdq7lLZH4_Vu-YNVbw,1802
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
@@ -9,14 +8,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=1GVcwOM97UMrsiPg3fAd9xJe2p9cHouidWgIPMWUxc0,10668
-agi_med_common/models/chat_item.py,sha256=eTUBCLlTVPqIS64qnaPB92vCJyiBixIPi8ochrgOPHo,5262
+agi_med_common/models/chat.py,sha256=lj6nxxsk0ng_1bhlakHqR_51ho7JonN2-2kYYhVpSxU,10720
+agi_med_common/models/chat_item.py,sha256=QlJ5Q40eX7lEeH9YBmdTH71ugrEtsU-4thkhD8cJhu0,5548
 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.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,,
+agi_med_common-5.1.0.dist-info/METADATA,sha256=wh-abOnVCcW_TqVKFyNXNnjXR_adZlPcC-LuRcAkujQ,517
+agi_med_common-5.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.1.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.1.0.dist-info/RECORD,,

agi_med_common/api_v2.py

@@ -1,68 +0,0 @@
-from typing import List, Tuple
-
-from agi_med_common.models import Chat, ChatMessage, DomainInfo, TrackInfo
-from pydantic import BaseModel
-
-
-Value = str
-Interpretation = str
-ResourceId = str
-
-
-class ChatManagerAPI:
-    def get_domains(self, language_code: str, client_id: str) -> List[DomainInfo]:
-        raise NotImplementedError
-
-    def get_tracks(self, language_code: str, client_id: str) -> List[TrackInfo]:
-        raise NotImplementedError
-
-    def get_response(self, chat: Chat, request_id: str = "") -> List[ChatMessage]:
-        raise NotImplementedError
-
-
-class TextGeneratorAPI:
-    def process(self, chat: Chat, request_id: str = "") -> str:
-        raise NotImplementedError
-
-
-class ContentInterpreterRemoteResponse(BaseModel):
-    interpretation: str
-    resource_fname: str
-    resource: bytes
-
-
-class ContentInterpreterRemoteAPI:
-    def interpret_remote(
-        self,
-        kind: str,
-        query: str,
-        resource: bytes,
-        chat: Chat | None = None,
-        request_id: str = "",
-    ) -> ContentInterpreterRemoteResponse:
-        raise NotImplementedError
-
-
-class ClassifierAPI:
-    def get_values(self) -> List[Value]:
-        raise NotImplementedError
-
-    def evaluate(self, chat: Chat, request_id: str = "") -> Value:
-        raise NotImplementedError
-
-
-class CriticAPI:
-    def evaluate(self, text: str, chat: Chat | None = None, request_id: str = "") -> float:
-        raise NotImplementedError
-
-
-class ContentInterpreterAPI:
-    def interpret(
-        self, kind: str, query: str, resource_id: str = "", chat: Chat | None = None, request_id: str = ""
-    ) -> Tuple[Interpretation, ResourceId | None]:
-        raise NotImplementedError
-
-
-class TextProcessorAPI:
-    def process(self, text: str, chat: Chat | None = None, request_id: str = "") -> str:
-        raise NotImplementedError