agi-med-common 3.1.2__py3-none-any.whl → 3.2.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,10 +1,12 @@
-__version__ = "3.1.2"
+__version__ = "3.2.0"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
     TrackIdEnum,
     StatesEnum,
     MTRSLabelEnum,
+    ActionEnum,
+    ModerationLabelEnum,
     ChatItem,
     InnerContextItem,
     OuterContextItem,

agi_med_common/models/__init__.py

@@ -1,5 +1,5 @@
 from ._base import _Base
-from .enums import TrackIdEnum, StatesEnum, MTRSLabelEnum
+from .enums import TrackIdEnum, StateEnum, MTRSLabelEnum, ActionEnum, ModerationLabelEnum
 from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
 from .chat_item_with_state import ChatItemWithState, InnerContextWithState, ReplicaWithState
 from .base_config_models import GigaChatConfig

agi_med_common/models/enums/__init__.py

@@ -1,3 +1,5 @@
 from .track_id_enum import TrackIdEnum
-from .states_enum import StatesEnum
+from .state_enum import StateEnum
 from .mtrs_label_enum import MTRSLabelEnum
+from .action_enum import ActionEnum
+from .moderation_label_enum import ModerationLabelEnum

agi_med_common/models/enums/action_enum.py

@@ -0,0 +1,17 @@
+from enum import StrEnum, auto
+
+
+class ActionEnum(StrEnum):
+    START = auto()
+
+    GREETING = auto()
+    MODERATION = auto()
+    ANSWER = auto()
+    QUESTION = auto()
+    CRITICAL = auto()
+    MED_TEST_DECRYPTION = auto()
+
+    DIAGNOSIS = auto()
+    REPORT = auto()
+
+    FINAL = auto()

agi_med_common/models/enums/moderation_label_enum.py

@@ -0,0 +1,13 @@
+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()

agi_med_common/models/enums/mtrs_label_enum.py

@@ -1,6 +1,10 @@
-from enum import StrEnum
+from enum import StrEnum, auto
 
 
 class MTRSLabelEnum(StrEnum):
-    LABORATORY = "LABORATORY"
-    INSTRUMENTAL = "INSTRUMENTAL"
+    @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/state_enum.py

@@ -0,0 +1,9 @@
+from enum import StrEnum, auto
+
+
+class StateEnum(StrEnum):
+    EMPTY = auto()
+    COLLECTION = auto()
+    READINESS = auto()
+    FINAL = auto()
+    ERROR = auto()

{agi_med_common-3.1.2.dist-info → agi_med_common-3.2.0.dist-info}/METADATA

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

{agi_med_common-3.1.2.dist-info → agi_med_common-3.2.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
-agi_med_common/__init__.py,sha256=oQDCDS_sbDl4LArCLLH3OjW3LrbY8oi3bXIqXp1zV4w,404
+agi_med_common/__init__.py,sha256=O8I_AkCdDjoOJmeQAiaxknGGwFP6dXNjJoOMKYS6R8k,445
 agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 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=_tmBPpZr1UaIycYzuvwrnPrVeu5kcwJBXSbTb7InGjI,842
-agi_med_common/models/__init__.py,sha256=yxD4vQNlPHOYMfpZcMDTBSGCV8wQs-ueVuxuLGZk_S0,304
+agi_med_common/models/__init__.py,sha256=ebG_iVj_cQ8Em6pdsLarl8zSfotEF3KBV1XMpDyS5YY,336
 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
@@ -17,11 +17,13 @@ agi_med_common/models/chat_item_with_state/__init__.py,sha256=KrXgsvJnLzS1kSi9_Y
 agi_med_common/models/chat_item_with_state/chat_with_state.py,sha256=IgZh0CRoar56eM8QOblPx8tsIm9NPO5tOj0mVC8yj40,205
 agi_med_common/models/chat_item_with_state/inner_context_with_state.py,sha256=qIwqCHn-JuQemcVQqLDzkpuaPRNCjIL_-h1wm1Jv5WU,212
 agi_med_common/models/chat_item_with_state/replica_with_state.py,sha256=YvUcCo1HTfxYUEwZ5T7UqLW2WkZ8SFvVNQZf-5mK5Dg,192
-agi_med_common/models/enums/__init__.py,sha256=i6xUMYZee-7lJO7hnmz1EqELyZBF8upVKKZEqNlsjmY,118
-agi_med_common/models/enums/mtrs_label_enum.py,sha256=yFH4jwh_6OiwgiVeUCCCHSZ1it7go33qd52Ox4uq9wo,121
-agi_med_common/models/enums/states_enum.py,sha256=wA5iUNJd3zKDxMg8yL4IzxhbS-wRsb95YZAOLHVGPzg,584
+agi_med_common/models/enums/__init__.py,sha256=7TyOshcmklii1Cvx76Kxw4Ly9Wei2hsX1aHk6kzkVrY,207
+agi_med_common/models/enums/action_enum.py,sha256=ZZtSBlOsQWvJ1UYqmsMsVLrzO1kJdImqJMqI0_KSV9M,287
+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/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
 agi_med_common/models/enums/track_id_enum.py,sha256=YoP15BTGjaFDxCOtjsqfwuMTW-BhcaQyEdxLfK7RQUA,454
-agi_med_common-3.1.2.dist-info/METADATA,sha256=iFsWan5LUGLkD0s4JQVBNExZ-bwGljWpd9mTBNBxgtg,520
-agi_med_common-3.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-agi_med_common-3.1.2.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.1.2.dist-info/RECORD,,
+agi_med_common-3.2.0.dist-info/METADATA,sha256=dJezI5EWow9eyeK5d2nSYi4ied8g5RcXX_bcTA9RXVI,520
+agi_med_common-3.2.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agi_med_common-3.2.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.2.0.dist-info/RECORD,,

agi_med_common/models/enums/states_enum.py

@@ -1,30 +0,0 @@
-from enum import StrEnum, auto
-
-
-class StatesEnum(StrEnum):
-    """
-    Класс для хранения имен состояний.
-    """
-
-    BEGIN = auto()
-    START = auto()
-
-    IS_CHILD = auto()
-    IS_ABSURD = auto()
-    CRITICAL = auto()
-    WHAT_COMPLAINTS = auto()
-    NOT_MEDICAL = auto()
-
-    ERROR_STATE = auto()
-
-    CONSULTATION_TRANSIT = auto()
-    NOT_MEDICAL_DOC = auto()
-    CONSULTATION = auto()
-
-    INFO_COLLECTION = auto()
-    MAKE_DIAGNOSIS = auto()
-
-    SUMMARIZATION = auto()
-    MEDICAL_TEST_RECOMMENDATION = auto()
-
-    ANALYSIS_CONSULTATION = auto()