agi-med-common 3.3.3__py3-none-any.whl → 3.5.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.3.3"
+__version__ = "3.5.0"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -13,6 +13,9 @@ from .models import (
     ReplicaItem,
     ReplicaItemPair,
 )
+
+from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
+
 from .utils import make_session_id, read_json, replace_n
 from .validators import is_file_exist, validate_prompt
 from .xml_parser import XMLParser

agi_med_common/models/__init__.py

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

agi_med_common/models/chat_item/replica_item.py

@@ -6,8 +6,8 @@ from pydantic import Field
 from .. import _Base
 from ..enums import StateEnum, ActionEnum
 
-_TIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
-_EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_TIME_FORMAT)
+_DATETIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
+_EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_DATETIME_FORMAT)
 
 
 class ReplicaItem(_Base):
@@ -17,7 +17,7 @@ class ReplicaItem(_Base):
         _EXAMPLE_DATETIME,
         alias="DateTime",
         examples=[_EXAMPLE_DATETIME],
-        description=f"Format: {_TIME_FORMAT}",
+        description=f"Format: {_DATETIME_FORMAT}",
     )
     state: StateEnum = Field(
         StateEnum.EMPTY, alias="State", description="chat manager fsm state", examples=[StateEnum.COLLECTION]
@@ -29,6 +29,10 @@ class ReplicaItem(_Base):
     def to_dict(self) -> dict[str, Any]:
         return self.model_dump(by_alias=True)
 
+    @staticmethod
+    def DATETIME_FORMAT() -> str:
+        return _DATETIME_FORMAT
+
 
 class ReplicaItemPair(_Base):
     user_replica: ReplicaItem = Field(alias="UserReplica")

agi_med_common/models/enums/__init__.py

@@ -3,3 +3,7 @@ from .state_enum import StateEnum
 from .mtrs_label_enum import MTRSLabelEnum
 from .action_enum import ActionEnum
 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

agi_med_common/models/enums/diagnostics_xml_tag_enum.py

@@ -0,0 +1,11 @@
+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

@@ -0,0 +1,11 @@
+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/mtrs_xml_tag_enum.py

@@ -0,0 +1,11 @@
+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-3.3.3.dist-info → agi_med_common-3.5.0.dist-info}/METADATA

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

{agi_med_common-3.3.3.dist-info → agi_med_common-3.5.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
-agi_med_common/__init__.py,sha256=4QcrtmRdnNtnqUNfgu-enI8IXAKD0gMXY5uPIBbs5ZA,427
+agi_med_common/__init__.py,sha256=xDj7PLO_pv-CqgTxQRjMN-Jy9wOMCftsLS8dh2pMlEE,511
 agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 agi_med_common/xml_parser.py,sha256=1MYE9yajX1Enrn0A6DD1cXagZuYdjCvp3vN4D8fIge4,523
 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=acJcPaHS9X-GUhjaD_8Cg1KZBMtOZ3hEnQlglHWN7-A,260
+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
@@ -13,14 +13,17 @@ agi_med_common/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTg
 agi_med_common/models/chat_item/chat_item.py,sha256=Ym-afB9-7NI6QquhMPqd2ZBkbcx-Txr-ZiXwn0PTO-0,1182
 agi_med_common/models/chat_item/inner_context_item.py,sha256=YXmlNLabKN3O84UWJENsjfdtPq2XRH1VqKiyIxS-p64,255
 agi_med_common/models/chat_item/outer_context_item.py,sha256=IzrwhqbLKR8ZF8-Sd2CyNSa0eqb21NQH7LUqkg9se2Y,1046
-agi_med_common/models/chat_item/replica_item.py,sha256=1BIKdqVfpWoWQP6xhtfr2c__Z-LJVDOkGcy5gbGQ_DA,1171
-agi_med_common/models/enums/__init__.py,sha256=7TyOshcmklii1Cvx76Kxw4Ly9Wei2hsX1aHk6kzkVrY,207
+agi_med_common/models/chat_item/replica_item.py,sha256=s9UP8K8wecXP-3e_ifc3Vv9ai8B-EBySwklK1tgjZPg,1268
+agi_med_common/models/enums/__init__.py,sha256=wnI9ub2F1L-TTB2jIpVF3OTLFr0QzcQjW3pX4P45HXo,377
 agi_med_common/models/enums/action_enum.py,sha256=ZZtSBlOsQWvJ1UYqmsMsVLrzO1kJdImqJMqI0_KSV9M,287
+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=Ixz-7_6RneNwGzDzO9pV38V6ZutUQa2pUS2torMDY8s,271
 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/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
 agi_med_common/models/enums/track_id_enum.py,sha256=YoP15BTGjaFDxCOtjsqfwuMTW-BhcaQyEdxLfK7RQUA,454
-agi_med_common-3.3.3.dist-info/METADATA,sha256=fhl2GogahhCpsbQ1NBsZYFG_v8lR6bcs_DbuaU94yIY,520
-agi_med_common-3.3.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-agi_med_common-3.3.3.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.3.3.dist-info/RECORD,,
+agi_med_common-3.5.0.dist-info/METADATA,sha256=IZtayKZ7zv-EY6JkKe_gz_394R-SvyScK7xxzqix3i4,520
+agi_med_common-3.5.0.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
+agi_med_common-3.5.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.5.0.dist-info/RECORD,,

{agi_med_common-3.3.3.dist-info → agi_med_common-3.5.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (75.8.1)
 Root-Is-Purelib: true
 Tag: py3-none-any