agi-med-common 1.5.0__tar.gz → 2.0.0__tar.gz

{agi_med_common-1.5.0 → agi_med_common-2.0.0}/PKG-INFO

@@ -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 → agi_med_common-2.0.0}/src/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-2.0.0/src/agi_med_common/models/chat_item/replica_item.py

@@ -0,0 +1,31 @@
+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: 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-1.5.0 → agi_med_common-2.0.0}/src/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 → agi_med_common-2.0.0}/src/agi_med_common.egg-info/PKG-INFO

@@ -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/src/agi_med_common/models/chat_item/replica_item.py

@@ -1,10 +0,0 @@
-from pydantic import Field
-
-from .. import _Base
-
-
-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"])
-    previous_score: float | None = Field(None, alias="PreviousScore", exclude=True, deprecated=True)