agi-med-common 3.6.2__py3-none-any.whl → 4.0.0__py3-none-any.whl

agi_med_common/__init__.py

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

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",

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/parallel_map.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, TypeVar
+from typing import Any, Callable, TypeVar, Iterable
 import concurrent.futures
 from tqdm import tqdm
 
@@ -8,12 +8,12 @@ X = TypeVar("X")
 
 def parallel_map(
     func: Callable[..., X],
-    items: list[Any],
+    items: Iterable[Any],
     process: bool = False,
     multiple_args: bool = False,
     max_workers: int = 2,
     show_tqdm: bool = False,
-    desc: str = "",
+    desc: str = ""
 ) -> list[X]:
     pool = (
         concurrent.futures.ProcessPoolExecutor(max_workers=max_workers)
@@ -27,5 +27,7 @@ def parallel_map(
                 futures.append(executor.submit(func, *item))
             else:
                 futures.append(executor.submit(func, item))
-        results: list[X] = [future.result() for future in tqdm(futures, disable=(not show_tqdm), desc=desc)]
+        results: list[X] = [
+            future.result() for future in tqdm(futures, disable=(not show_tqdm), desc=desc)
+        ]
     return results

{agi_med_common-3.6.2.dist-info → agi_med_common-4.0.0.dist-info}/METADATA

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 3.6.2
+Version: 4.0.0
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: pydantic<=3.0.0,>=2.9.2
 Requires-Dist: loguru~=0.7.2
+Requires-Dist: tqdm==4.67.*
 
 # agi-med-common
 
@@ -14,7 +15,7 @@ Common компонент
 
 ## Ответственный разработчик
 
-@zhelvakov
+@bakulin
 
 ## Общая информация
 

{agi_med_common-3.6.2.dist-info → agi_med_common-4.0.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-agi_med_common/__init__.py,sha256=4YmS-FOJBhMUtRG29AgzaLvuqtvBhcfq3UQZmk_N190,575
+agi_med_common/__init__.py,sha256=NU3IwKvif4JVeoBOH5-VLR9nTUifoee2Dyi50DuUElc,575
 agi_med_common/file_storage.py,sha256=GR6_jKZ0o9uwUkOKZd1XaCVOSw0YmFlCOlC6EX8zz0I,1247
-agi_med_common/parallel_map.py,sha256=bC7YGM5sWxSNvaA54Qpu8rlFJTOlRyMw0eWlttkTsW8,886
+agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
@@ -15,17 +15,15 @@ agi_med_common/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTg
 agi_med_common/models/chat_item/chat_item.py,sha256=Ym-afB9-7NI6QquhMPqd2ZBkbcx-Txr-ZiXwn0PTO-0,1182
 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=2N586TCQ33YoBOeghvvPnIPHSEtnWMQOr_ZldNfLH3o,1676
+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.2.dist-info/METADATA,sha256=Se5lBEGM9qmjZbqKsAT0UrOD67AkaxYXFWdYLdA5RQg,520
-agi_med_common-3.6.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-agi_med_common-3.6.2.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.6.2.dist-info/RECORD,,
+agi_med_common-4.0.0.dist-info/METADATA,sha256=j8n2jal1fUydQ8rD9Y5t5Ra9UmcxilK-j6cCdUjY5OU,546
+agi_med_common-4.0.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+agi_med_common-4.0.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.0.0.dist-info/RECORD,,

{agi_med_common-3.6.2.dist-info → agi_med_common-4.0.0.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()