agi-med-common 3.3.4__tar.gz → 3.5.3__tar.gz

{agi_med_common-3.3.4 → agi_med_common-3.5.3}/PKG-INFO

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

{agi_med_common-3.3.4 → agi_med_common-3.5.3}/src/agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.3.4"
+__version__ = "3.5.3"
 
 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-3.3.4 → agi_med_common-3.5.3}/src/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-3.3.4 → agi_med_common-3.5.3}/src/agi_med_common/models/chat_item/replica_item.py

@@ -4,7 +4,7 @@ from typing import Any
 from pydantic import Field
 
 from .. import _Base
-from ..enums import StateEnum, ActionEnum
+from ..enums import StateEnum, ActionEnum, ModerationLabelEnum
 
 _DATETIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
 _EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_DATETIME_FORMAT)
@@ -25,6 +25,12 @@ class ReplicaItem(_Base):
     action: ActionEnum = Field(
         ActionEnum.START, alias="Action", description="chat manager fsm action", examples=[ActionEnum.DIAGNOSIS]
     )
+    moderation: ModerationLabelEnum = Field(
+        ModerationLabelEnum.OK,
+        alias="Moderation",
+        description="chat manager moderated outcome type",
+        examples=[ModerationLabelEnum.NON_MED],
+    )
 
     def to_dict(self) -> dict[str, Any]:
         return self.model_dump(by_alias=True)

{agi_med_common-3.3.4 → agi_med_common-3.5.3}/src/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-3.5.3/src/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-3.5.3/src/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-3.3.4 → agi_med_common-3.5.3}/src/agi_med_common/models/enums/moderation_label_enum.py

@@ -11,3 +11,4 @@ class ModerationLabelEnum(IntEnum):
     CHILD = auto()
     ABSURD = auto()
     GREETING = auto()
+    RECEIPT = auto()

agi_med_common-3.5.3/src/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.4 → agi_med_common-3.5.3}/src/agi_med_common/xml_parser.py

@@ -1,9 +1,10 @@
 import re
+from enum import StrEnum
 
 
 class XMLParser:
     @staticmethod
-    def make_xml(body: str, tag: str) -> str:
+    def make_xml(body: str, tag: str | StrEnum) -> str:
         return f"<{tag}>{body}</{tag}>"
 
     @staticmethod
@@ -12,6 +13,6 @@ class XMLParser:
         return re.sub(re_xml, "", text).replace("  ", " ").strip()
 
     @staticmethod
-    def get_tag_list(response: str, tag: str) -> list[str]:
+    def get_tag_list(response: str, tag: str | StrEnum) -> list[str]:
         tag_content = re.findall(f"<{tag}>(.*?)</{tag}>", response, re.DOTALL)
         return [content.lower().strip() for content in tag_content]

{agi_med_common-3.3.4 → agi_med_common-3.5.3}/src/agi_med_common.egg-info/PKG-INFO

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

{agi_med_common-3.3.4 → agi_med_common-3.5.3}/src/agi_med_common.egg-info/SOURCES.txt

@@ -24,7 +24,10 @@ src/agi_med_common/models/chat_item/outer_context_item.py
 src/agi_med_common/models/chat_item/replica_item.py
 src/agi_med_common/models/enums/__init__.py
 src/agi_med_common/models/enums/action_enum.py
+src/agi_med_common/models/enums/diagnostics_xml_tag_enum.py
+src/agi_med_common/models/enums/doctor_choice_xml_tag_enum.py
 src/agi_med_common/models/enums/moderation_label_enum.py
 src/agi_med_common/models/enums/mtrs_label_enum.py
+src/agi_med_common/models/enums/mtrs_xml_tag_enum.py
 src/agi_med_common/models/enums/state_enum.py
 src/agi_med_common/models/enums/track_id_enum.py