agi-med-common 1.5.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.5.0"
+__version__ = "2.0.0"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (

agi_med_common/models/chat_item/replica_item.py

@@ -1,10 +1,31 @@
-from pydantic import Field
+from datetime import datetime
+
+from pydantic import Field, field_validator, field_serializer
 
 from .. import _Base
 
+_TIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
+_EXAMPLE_DATETIME: datetime = datetime(1970, 1, 1, 0, 0, 0)
+
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
-    date_time: str = Field("1970-01-01-00-00-00", alias="DateTime", examples=["2024-12-10-18-03-46"])
+    date_time: datetime = Field(
+        _EXAMPLE_DATETIME,
+        alias="DateTime",
+        examples=[_EXAMPLE_DATETIME.strftime(_TIME_FORMAT)],
+        description=f"Format: {_TIME_FORMAT}",
+    )
     previous_score: float | None = Field(None, alias="PreviousScore", exclude=True, deprecated=True)
+
+    @field_validator("date_time", mode="before")
+    @classmethod
+    def convert_date_time(cls, timestamp: str | datetime) -> datetime:
+        if isinstance(timestamp, datetime):
+            return timestamp
+        return datetime.strptime(timestamp, _TIME_FORMAT)
+
+    @field_serializer("date_time")
+    def serialize_date_time(self, timestamp: datetime) -> str:
+        return timestamp.strftime(_TIME_FORMAT)

agi_med_common/models/enums/states_enum.py

@@ -8,14 +8,22 @@ class StatesEnum(StrEnum):
 
     BEGIN = auto()
     START = auto()
-    NOT_MEDICAL = auto()
+
     IS_CHILD = auto()
     IS_ABSURD = auto()
     CRITICAL = auto()
     WHAT_COMPLAINTS = auto()
-    ANALYSIS_CONSULTATION = auto()
-    COMMON_CONSULTATION = auto()
+    NOT_MEDICAL = auto()
+
+    ERROR_STATE = auto()
+
+    CONSULTATION_TRANSIT = auto()
+    NOT_MEDICAL_DOC = auto()
+    CONSULTATION = auto()
+
     INFO_COLLECTION = auto()
     MAKE_DIAGNOSIS = auto()
-    COMMENT_DIAGNOSIS = auto()
-    EXIT_INNER_CYCLE = auto()
+
+    SUMMARIZATION = auto()
+
+    ANALYSIS_CONSULTATION = auto()

{agi_med_common-1.5.0.dist-info → agi_med_common-2.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: agi_med_common
-Version: 1.5.0
+Version: 2.0.0
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.13

{agi_med_common-1.5.0.dist-info → agi_med_common-2.0.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=nJzhnz1BvLLyFjKRZMXZg8eQeAb174cb5uLRhCLsfZw,385
+agi_med_common/__init__.py,sha256=0GJulrzkBDkB17nr_utQW7ghrK5yCMq30GZKK9aV3wY,385
 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
@@ -12,15 +12,15 @@ agi_med_common/models/chat_item/__init__.py,sha256=MoZFnZQV-tWJTHmH1OhRBRkrc5cGW
 agi_med_common/models/chat_item/chat_item.py,sha256=aCRR4OwL-YS6hvVSNkIg7FNpfvEGo8BESCY8BhiAlNk,255
 agi_med_common/models/chat_item/inner_context_item.py,sha256=s17frAcwyi8WcHnBt-5LF8nUJeO-hnu7YMobYQO5a-c,166
 agi_med_common/models/chat_item/outer_context_item.py,sha256=IprCIh8ki3PUueFwdmPw-MYn0-prx_G523Y-IXN62ZY,792
-agi_med_common/models/chat_item/replica_item.py,sha256=cv6eeLjKwuITeFbkjTcAfh90Dzf2b47YaE0UeVxdZf8,450
+agi_med_common/models/chat_item/replica_item.py,sha256=_OvnGmbP-wzxi16c6FGz-_Y-QtudV1Pi6I-zCJ7aQfA,1131
 agi_med_common/models/chat_item_with_state/__init__.py,sha256=KrXgsvJnLzS1kSi9_Y1z6_egxIqKL7epY1r-PIyrnno,156
 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=yHUFI7n5r-LY5y4CbliZlXDrmr38bsz8dh4iyNo5Ti8,75
-agi_med_common/models/enums/states_enum.py,sha256=_aX8IVB8PeXjw_HgFVivRyqWgghTud7lspLCSalbrKM,490
+agi_med_common/models/enums/states_enum.py,sha256=IerXA4uaCNnE0zt4JF4bveqwL1wpUUBMXMh6kqFV_XM,543
 agi_med_common/models/enums/track_id_enum.py,sha256=fHGQz9AACIS6SfTFCrSau1HKee70r2bap-OcBxD3YRE,423
-agi_med_common-1.5.0.dist-info/METADATA,sha256=eQ59D8d3hH8sX0YFHZ65dzU66PA6UjDcwXpGnpYE_ZY,520
-agi_med_common-1.5.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-agi_med_common-1.5.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-1.5.0.dist-info/RECORD,,
+agi_med_common-2.0.0.dist-info/METADATA,sha256=VTyClWQ9zBYJdascshkCrq_hXlMsvCFyU3qxY8nz4Sg,520
+agi_med_common-2.0.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agi_med_common-2.0.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-2.0.0.dist-info/RECORD,,

{agi_med_common-1.5.0.dist-info → agi_med_common-2.0.0.dist-info}/WHEEL

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