agi-med-common 4.2.5__py3-none-any.whl → 4.2.6__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,8 +1,7 @@
-__version__ = "4.2.5"
+__version__ = "4.2.6"
 
-from .logger import LogLevelEnum, logger_init
+from .logger import LogLevelEnum, logger_init, log_llm_error
 from .models import (
-    TrackIdEnum,
     StateEnum,
     MTRSLabelEnum,
     ActionEnum,
@@ -13,6 +12,7 @@ from .models import (
     ReplicaItem,
     ReplicaItemPair,
 )
+from .models.widget import Widget
 from .file_storage import FileStorage
 from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 from .utils import make_session_id, read_json

agi_med_common/models/__init__.py

@@ -1,6 +1,6 @@
 from ._base import _Base
 
-from .enums import TrackIdEnum, StateEnum, MTRSLabelEnum, ActionEnum, ModerationLabelEnum
+from .enums import StateEnum, MTRSLabelEnum, ActionEnum, ModerationLabelEnum
 from .enums import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 
 from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem, ReplicaItemPair

agi_med_common/models/chat_item/outer_context_item.py

@@ -2,7 +2,6 @@ from pydantic import Field, ConfigDict
 from typing import Any
 
 from .. import _Base
-from ..enums import TrackIdEnum
 
 
 class OuterContextItem(_Base):
@@ -14,7 +13,7 @@ class OuterContextItem(_Base):
     user_id: str = Field("", alias="UserId", examples=["123456789"])
     session_id: str = Field("", alias="SessionId", examples=["987654321"])
     client_id: str = Field("", alias="ClientId", examples=["543216789"])
-    track_id: TrackIdEnum | str = Field(default=TrackIdEnum.CONSULTATION, alias="TrackId")
+    track_id: str = Field(default="Consultation", alias="TrackId")
     entrypoint_key: str = Field("", alias="EntrypointKey", examples=["giga"])
 
     def create_id(self, short: bool = False, clean: bool = False) -> str:

agi_med_common/models/chat_item/replica_item.py

@@ -1,10 +1,11 @@
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field, ConfigDict
 
 from .. import _Base
 from ..enums import StateEnum, ActionEnum, ModerationLabelEnum
+from agi_med_common.models.widget import Widget
 
 
 _DATETIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
@@ -13,7 +14,8 @@ _EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_DATETIME_FORMAT
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
-    resource_id: Optional[str] = Field(None, alias="ResourceId", examples=["<link-id>"])
+    resource_id: str | None = Field(None, alias="ResourceId", examples=["<link-id>"])
+    widget: Widget | None = Field(None, alias="Widget", exampels=[None])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
     date_time: str = Field(
         _EXAMPLE_DATETIME,
@@ -37,6 +39,10 @@ class ReplicaItem(_Base):
     def DATETIME_FORMAT() -> str:
         return _DATETIME_FORMAT
 
+    def with_now_datetime(self):
+        dt = datetime.now().strftime(ReplicaItem.DATETIME_FORMAT())
+        return self.model_copy(update=dict(date_time=dt))
+
 
 class ReplicaItemPair(_Base):
     # remove annoying warning for protected `model_` namespace

agi_med_common/models/enums.py

@@ -0,0 +1,58 @@
+from enum import IntEnum, StrEnum, auto
+
+
+StateEnum, ActionEnum = str, str
+
+
+class ModerationLabelEnum(IntEnum):
+    @staticmethod
+    def _generate_next_value_(name, start, count, last_values):
+        return count
+
+    OK = auto()
+    NON_MED = auto()
+    CHILD = auto()
+    ABSURD = auto()
+    GREETING = auto()
+    RECEIPT = auto()
+
+
+class DoctorChoiceXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"{name.lower()}"
+
+    DIAGNOSTICS = auto()
+    SUMMARIZATION = auto()
+    MTRS = auto()
+
+
+class MTRSXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"mtrs_{name.lower()}"
+
+    NAME = auto()
+    LABEL = auto()
+    DESC = auto()
+
+
+class MTRSLabelEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return name.upper()
+
+    LABORATORY = auto()
+    INSTRUMENTAL = auto()
+
+
+class DiagnosticsXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"diag_{name.lower()}"
+
+    DIAG = auto()
+    DOC = auto()
+    DESC = auto()
+
+

agi_med_common/models/widget.py

@@ -0,0 +1,20 @@
+from typing import List, Self
+from pydantic import BaseModel, model_validator
+
+
+class Widget(BaseModel):
+    buttons: List[List[str]] | None = None
+    ibuttons: List[List[str]] | None = None
+
+    @model_validator(mode="after")
+    def check(self) -> Self:
+        if not self.buttons and not self.ibuttons:
+            raise ValueError("Empty widget is not allowed!")
+        if not self.ibuttons:
+            return self
+        for row in self.ibuttons:
+            for btn in row:
+                if ":" in btn:
+                    continue
+                raise ValueError(f"Expected buttons like `<callback>:<caption>`, found: {btn}")
+        return self

{agi_med_common-4.2.5.dist-info → agi_med_common-4.2.6.dist-info}/METADATA

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

agi_med_common-4.2.6.dist-info/RECORD

@@ -0,0 +1,24 @@
+agi_med_common/__init__.py,sha256=cTWBAd5qlfMBDiCyXXt9kSdgZouwSAc54KqIUcli19Q,665
+agi_med_common/file_storage.py,sha256=4GW0CU0YtVzWmxV6vGjZg0DfyEgjlrgLRsZ7cEP-6Qk,1659
+agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
+agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
+agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
+agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
+agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
+agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
+agi_med_common/logger/logger.py,sha256=-VU69ULbw2-pNpOcxKiMpOeRAPe6H-EsGV9WRTv1SUo,839
+agi_med_common/models/__init__.py,sha256=KXnXxD2c5v2TVXrc-n7gJipTAyvtMasbocxThINIojs,330
+agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
+agi_med_common/models/enums.py,sha256=RxPPbkW1NcHAGO-gHuzwIvOjv0_ZBsr9b1aq_wk4WFM,1313
+agi_med_common/models/widget.py,sha256=g7Bobyk_gANghGEhjDjvi4i-YRoHo5Bm1Pr4eF-zf0U,661
+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/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTgdV43TZOcfHasHg,185
+agi_med_common/models/chat_item/chat_item.py,sha256=xP_uP14qK-u8z0ZUs2HFhhea-IP230UgOmjeqeC_N6Q,1180
+agi_med_common/models/chat_item/inner_context_item.py,sha256=2wIwDtZwicwhSyXeuOqH8-xONcskeEiw6QhjSnEEwZs,329
+agi_med_common/models/chat_item/outer_context_item.py,sha256=P2ulGbKnEXBJ50Ou-_p_qKCdO1lB4SUvdSu1ZG1yRdo,1213
+agi_med_common/models/chat_item/replica_item.py,sha256=aUwgB2McvMUoeFUqeFVqLg-1aJzpjznBxOL6ngji3h4,1940
+agi_med_common-4.2.6.dist-info/METADATA,sha256=b0sMgl96lXlO1VTImkmPb-ovcIygFoa8Zlapwob1t-s,546
+agi_med_common-4.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-4.2.6.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.2.6.dist-info/RECORD,,

agi_med_common/models/enums/__init__.py

@@ -1,9 +0,0 @@
-from .track_id_enum import TrackIdEnum
-from .mtrs_label_enum import MTRSLabelEnum
-from .moderation_label_enum import ModerationLabelEnum
-
-from .diagnostics_xml_tag_enum import DiagnosticsXMLTagEnum
-from .mtrs_xml_tag_enum import MTRSXMLTagEnum
-from .doctor_choice_xml_tag_enum import DoctorChoiceXMLTagEnum
-
-StateEnum, ActionEnum = str, str

agi_med_common/models/enums/diagnostics_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class DiagnosticsXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"diag_{name.lower()}"
-
-    DIAG = auto()
-    DOC = auto()
-    DESC = auto()

agi_med_common/models/enums/doctor_choice_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class DoctorChoiceXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"{name.lower()}"
-
-    DIAGNOSTICS = auto()
-    SUMMARIZATION = auto()
-    MTRS = auto()

agi_med_common/models/enums/moderation_label_enum.py

@@ -1,14 +0,0 @@
-from enum import IntEnum, auto
-
-
-class ModerationLabelEnum(IntEnum):
-    @staticmethod
-    def _generate_next_value_(name, start, count, last_values):
-        return count
-
-    OK = auto()
-    NON_MED = auto()
-    CHILD = auto()
-    ABSURD = auto()
-    GREETING = auto()
-    RECEIPT = auto()

agi_med_common/models/enums/mtrs_label_enum.py

@@ -1,10 +0,0 @@
-from enum import StrEnum, auto
-
-
-class MTRSLabelEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return name.upper()
-
-    LABORATORY = auto()
-    INSTRUMENTAL = auto()

agi_med_common/models/enums/mtrs_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class MTRSXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"mtrs_{name.lower()}"
-
-    NAME = auto()
-    LABEL = auto()
-    DESC = auto()

agi_med_common/models/enums/track_id_enum.py

@@ -1,18 +0,0 @@
-from enum import StrEnum, auto
-
-
-class TrackIdEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return "".join([word.title() for word in name.lower().split("_")])
-
-    DIAGNOSTIC = auto()
-    DIAGNOSTIC_SBERHEALTH = auto()
-    SECOND_OPINION = auto()
-    MEDICAL_TEST_DECRYPTION = auto()
-    CONSULTATION = auto()
-    DUMMY = auto()
-    COMMON_CONSULTATION = auto()
-    FINANCIER = auto()
-    FINANCIER_GENERIC = auto()
-    MULTIMODAL = auto()

agi_med_common-4.2.5.dist-info/RECORD

@@ -1,29 +0,0 @@
-agi_med_common/__init__.py,sha256=qpgKQvUWXCmAnbgdBIYqL-6NQcq25foavdorFLSzq6w,633
-agi_med_common/file_storage.py,sha256=4GW0CU0YtVzWmxV6vGjZg0DfyEgjlrgLRsZ7cEP-6Qk,1659
-agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
-agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
-agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
-agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
-agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
-agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
-agi_med_common/logger/logger.py,sha256=-VU69ULbw2-pNpOcxKiMpOeRAPe6H-EsGV9WRTv1SUo,839
-agi_med_common/models/__init__.py,sha256=D1-GV32Ap2qBzmJZC7RzkdJ3duqRMjzcZ4oUzbCZ66g,343
-agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-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/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTgdV43TZOcfHasHg,185
-agi_med_common/models/chat_item/chat_item.py,sha256=xP_uP14qK-u8z0ZUs2HFhhea-IP230UgOmjeqeC_N6Q,1180
-agi_med_common/models/chat_item/inner_context_item.py,sha256=2wIwDtZwicwhSyXeuOqH8-xONcskeEiw6QhjSnEEwZs,329
-agi_med_common/models/chat_item/outer_context_item.py,sha256=hSv-JMPlgLzH7kQzgjrDNZWSnm18TWHODziQixsqoIo,1269
-agi_med_common/models/chat_item/replica_item.py,sha256=D1goRTINLa1rUcQcwHOWUSiv_OR1UW0GVn5_TqK3dyk,1672
-agi_med_common/models/enums/__init__.py,sha256=NFgy8J-bET2GQqUyTaKYLzaLaM-jDR8-7TBTadWKm5o,341
-agi_med_common/models/enums/diagnostics_xml_tag_enum.py,sha256=aiUJpe5joh5EOcJ5sk1Xs003XkRry2CBL6H-bIEiapA,278
-agi_med_common/models/enums/doctor_choice_xml_tag_enum.py,sha256=MsSFI0vFHJVujhoj7fNX-fcGVV3jMF_nEDOJnjzh4ac,291
-agi_med_common/models/enums/moderation_label_enum.py,sha256=lbGG4Pu7cQp57uEyQEptNeYsQszK6yyxJ5AEhq4SGkE,292
-agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
-agi_med_common/models/enums/mtrs_xml_tag_enum.py,sha256=6OxuRsrx4b2uBjfrBgm4Y789Ly337_mQXL9VPRCpLyg,273
-agi_med_common/models/enums/track_id_enum.py,sha256=N3KUd97a4xDet_NKZ5URm0qWAAv3ts1l_foOjdlGY4c,532
-agi_med_common-4.2.5.dist-info/METADATA,sha256=TMhc3NYXrXLxhTV7LxKjpkaY5zkgn-WRpTrn2tVIuhM,546
-agi_med_common-4.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-4.2.5.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-4.2.5.dist-info/RECORD,,