agi-med-common 3.1.2__tar.gz → 3.2.0__tar.gz

{agi_med_common-3.1.2 → agi_med_common-3.2.0}/PKG-INFO

@@ -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 → agi_med_common-3.2.0}/src/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-3.1.2 → agi_med_common-3.2.0}/src/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-3.2.0/src/agi_med_common/models/enums/__init__.py

@@ -0,0 +1,5 @@
+from .track_id_enum import TrackIdEnum
+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-3.2.0/src/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-3.2.0/src/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-3.2.0/src/agi_med_common/models/enums/mtrs_label_enum.py

@@ -0,0 +1,10 @@
+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-3.2.0/src/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 → agi_med_common-3.2.0}/src/agi_med_common.egg-info/PKG-INFO

@@ -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 → agi_med_common-3.2.0}/src/agi_med_common.egg-info/SOURCES.txt

@@ -26,6 +26,8 @@ src/agi_med_common/models/chat_item_with_state/chat_with_state.py
 src/agi_med_common/models/chat_item_with_state/inner_context_with_state.py
 src/agi_med_common/models/chat_item_with_state/replica_with_state.py
 src/agi_med_common/models/enums/__init__.py
+src/agi_med_common/models/enums/action_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/states_enum.py
+src/agi_med_common/models/enums/state_enum.py
 src/agi_med_common/models/enums/track_id_enum.py

agi_med_common-3.1.2/src/agi_med_common/models/enums/__init__.py

@@ -1,3 +0,0 @@
-from .track_id_enum import TrackIdEnum
-from .states_enum import StatesEnum
-from .mtrs_label_enum import MTRSLabelEnum

agi_med_common-3.1.2/src/agi_med_common/models/enums/mtrs_label_enum.py

@@ -1,6 +0,0 @@
-from enum import StrEnum
-
-
-class MTRSLabelEnum(StrEnum):
-    LABORATORY = "LABORATORY"
-    INSTRUMENTAL = "INSTRUMENTAL"

agi_med_common-3.1.2/src/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()