agi-med-common 3.6.3__py3-none-any.whl → 4.0.1__py3-none-any.whl

agi_med_common/__init__.py

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

agi_med_common/models/chat_item/chat_item.py

@@ -18,7 +18,7 @@ class ChatItem(_Base):
 
     def update(self, replica_pair: ReplicaItemPair) -> None:
         self.inner_context.replicas.append(replica_pair.user_replica)
-        self.inner_context.replicas.append(replica_pair.model_replica)
+        self.inner_context.replicas.append(replica_pair.bot_replica)
 
     def count_questions(self) -> int:
         return len(

agi_med_common/models/chat_item/replica_item.py

@@ -6,6 +6,7 @@ from pydantic import Field, ConfigDict
 from .. import _Base
 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)
 
@@ -20,12 +21,8 @@ class ReplicaItem(_Base):
         examples=[_EXAMPLE_DATETIME],
         description=f"Format: {_DATETIME_FORMAT}",
     )
-    state: StateEnum = Field(
-        StateEnum.EMPTY, alias="State", description="chat manager fsm state", examples=[StateEnum.COLLECTION]
-    )
-    action: ActionEnum = Field(
-        ActionEnum.START, alias="Action", description="chat manager fsm action", examples=[ActionEnum.DIAGNOSIS]
-    )
+    state: StateEnum = Field("EMPTY", alias="State", description="chat manager fsm state", examples=["COLLECTION"])
+    action: ActionEnum = Field("START", alias="Action", description="chat manager fsm action", examples=["DIAGNOSIS"])
     moderation: ModerationLabelEnum = Field(
         ModerationLabelEnum.OK,
         alias="Moderation",
@@ -46,4 +43,4 @@ class ReplicaItemPair(_Base):
     model_config = ConfigDict(protected_namespaces=())
 
     user_replica: ReplicaItem = Field(alias="UserReplica")
-    model_replica: ReplicaItem = Field(alias="ModelReplica")
+    bot_replica: ReplicaItem = Field(alias="BotReplica")

agi_med_common/models/enums/__init__.py

@@ -1,9 +1,9 @@
 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
 
 from .diagnostics_xml_tag_enum import DiagnosticsXMLTagEnum
 from .mtrs_xml_tag_enum import MTRSXMLTagEnum
 from .doctor_choice_xml_tag_enum import DoctorChoiceXMLTagEnum
+
+StateEnum, ActionEnum = str, str

{agi_med_common-3.6.3.dist-info → agi_med_common-4.0.1.dist-info}/METADATA

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

{agi_med_common-3.6.3.dist-info → agi_med_common-4.0.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=IcL_L6jchfpfZUcCCWj3enH4rXPkeRuSa3MroRr7I7c,575
+agi_med_common/__init__.py,sha256=lWshIR6A2qvkqKno-AGFY1Y73YeaBZTUvY-csCt1XnM,575
 agi_med_common/file_storage.py,sha256=GR6_jKZ0o9uwUkOKZd1XaCVOSw0YmFlCOlC6EX8zz0I,1247
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
@@ -12,20 +12,18 @@ agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHm
 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
 agi_med_common/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTgdV43TZOcfHasHg,185
-agi_med_common/models/chat_item/chat_item.py,sha256=Ym-afB9-7NI6QquhMPqd2ZBkbcx-Txr-ZiXwn0PTO-0,1182
+agi_med_common/models/chat_item/chat_item.py,sha256=xP_uP14qK-u8z0ZUs2HFhhea-IP230UgOmjeqeC_N6Q,1180
 agi_med_common/models/chat_item/inner_context_item.py,sha256=YXmlNLabKN3O84UWJENsjfdtPq2XRH1VqKiyIxS-p64,255
 agi_med_common/models/chat_item/outer_context_item.py,sha256=IzrwhqbLKR8ZF8-Sd2CyNSa0eqb21NQH7LUqkg9se2Y,1046
-agi_med_common/models/chat_item/replica_item.py,sha256=OC3LCT3clGDFJ9pIr7FL8HXvldrplnuaLXEbkK6Hu9o,1737
-agi_med_common/models/enums/__init__.py,sha256=wnI9ub2F1L-TTB2jIpVF3OTLFr0QzcQjW3pX4P45HXo,377
-agi_med_common/models/enums/action_enum.py,sha256=J0S_CkCTDoebbdGbzfrnNTOXla-yMXsPsQI_dyXH2iE,348
+agi_med_common/models/chat_item/replica_item.py,sha256=D1goRTINLa1rUcQcwHOWUSiv_OR1UW0GVn5_TqK3dyk,1672
+agi_med_common/models/enums/__init__.py,sha256=NFgy8J-bET2GQqUyTaKYLzaLaM-jDR8-7TBTadWKm5o,341
 agi_med_common/models/enums/diagnostics_xml_tag_enum.py,sha256=aiUJpe5joh5EOcJ5sk1Xs003XkRry2CBL6H-bIEiapA,278
 agi_med_common/models/enums/doctor_choice_xml_tag_enum.py,sha256=MsSFI0vFHJVujhoj7fNX-fcGVV3jMF_nEDOJnjzh4ac,291
 agi_med_common/models/enums/moderation_label_enum.py,sha256=lbGG4Pu7cQp57uEyQEptNeYsQszK6yyxJ5AEhq4SGkE,292
 agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
 agi_med_common/models/enums/mtrs_xml_tag_enum.py,sha256=6OxuRsrx4b2uBjfrBgm4Y789Ly337_mQXL9VPRCpLyg,273
-agi_med_common/models/enums/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
 agi_med_common/models/enums/track_id_enum.py,sha256=N3KUd97a4xDet_NKZ5URm0qWAAv3ts1l_foOjdlGY4c,532
-agi_med_common-3.6.3.dist-info/METADATA,sha256=pN-Es5jecHbHSuWfYZTfJ3PshSnfyowErVAUl3STs9E,546
-agi_med_common-3.6.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-agi_med_common-3.6.3.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.6.3.dist-info/RECORD,,
+agi_med_common-4.0.1.dist-info/METADATA,sha256=ZPBBlimMkVTnQrJrevB-BX1lAbCN4hwNzSGBWYzfEaQ,546
+agi_med_common-4.0.1.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+agi_med_common-4.0.1.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.0.1.dist-info/RECORD,,

{agi_med_common-3.6.3.dist-info → agi_med_common-4.0.1.dist-info}/WHEEL

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

agi_med_common/models/enums/action_enum.py

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

agi_med_common/models/enums/state_enum.py

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