agi-med-common 4.2.12__py3-none-any.whl → 4.2.15__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,10 +1,8 @@
-__version__ = "4.2.12"
+__version__ = "4.2.15"
 
 from .logger import LogLevelEnum, logger_init, log_llm_error
 from .models import (
-    StateEnum,
     MTRSLabelEnum,
-    ActionEnum,
     ModerationLabelEnum,
     ChatItem,
     InnerContextItem,
@@ -15,7 +13,7 @@ from .models import (
 from .models.widget import Widget
 from .file_storage import FileStorage, ResourceId
 from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
-from .utils import make_session_id, read_json, try_parse_json, try_parse_int, pretty_line
+from .utils import make_session_id, read_json, try_parse_json, try_parse_int, try_parse_float, pretty_line
 from .validators import ExistingPath, ExistingFile, ExistingDir, StrNotEmpty, SecretStrNotEmpty, Prompt, Message
 from .xml_parser import XMLParser
 from .parallel_map import parallel_map

agi_med_common/models/__init__.py

@@ -1,7 +1,6 @@
 from ._base import _Base
 
-from .enums import StateEnum, MTRSLabelEnum, ActionEnum, ModerationLabelEnum
-from .enums import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
+from .enums import MTRSLabelEnum, ModerationLabelEnum, DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 
 from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem, ReplicaItemPair
 from .base_config_models import GigaChatConfig

agi_med_common/models/chat_item.py

@@ -4,7 +4,7 @@ from typing import Any
 from pydantic import Field, ConfigDict
 from agi_med_common.models.widget import Widget
 
-from .enums import ModerationLabelEnum, StateEnum, ActionEnum
+from .enums import ModerationLabelEnum
 from ._base import _Base
 
 
@@ -34,6 +34,7 @@ class OuterContextItem(_Base):
     client_id: str = Field("", alias="ClientId", examples=["543216789"])
     track_id: str = Field(default="Consultation", alias="TrackId")
     entrypoint_key: str = Field("", alias="EntrypointKey", examples=["giga"])
+    language_code: str = Field("ru", alias="LanguageCode", examples=["ru"])
 
     def create_id(self, short: bool = False, clean: bool = False) -> str:
         uid, sid, cid = self.user_id, self.session_id, self.client_id
@@ -59,8 +60,8 @@ class ReplicaItem(_Base):
         examples=[_EXAMPLE_DATETIME],
         description=f"Format: {_DATETIME_FORMAT}",
     )
-    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"])
+    state: str = Field("EMPTY", alias="State", description="chat manager fsm state", examples=["COLLECTION"])
+    action: str = Field("START", alias="Action", description="chat manager fsm action", examples=["DIAGNOSIS"])
     moderation: ModerationLabelEnum = Field(
         ModerationLabelEnum.OK,
         alias="Moderation",

agi_med_common/models/enums.py

@@ -1,9 +1,6 @@
 from enum import IntEnum, StrEnum, auto
 
 
-StateEnum, ActionEnum = str, str
-
-
 class ModerationLabelEnum(IntEnum):
     @staticmethod
     def _generate_next_value_(name, start, count, last_values):

agi_med_common/utils.py

@@ -15,11 +15,14 @@ def read_json(path: Path | os.PathLike[str] | str) -> list | dict:
 
 
 def try_parse_json(text: str | None) -> str | None:
+    # taken from agi_med_common
+    if not isinstance(text, str):
+        return None
     if not text:
         return None
-    if not text.startswith("{"):
+    if text[0] not in "{[":
         return None
-    if not text.endswith("}"):
+    if text[-1] not in "}]":
         return None
     try:
         return json.loads(text)
@@ -33,13 +36,12 @@ def try_parse_int(text: str) -> int | None:
     except ValueError:
         return None
 
-
 def pretty_line(text: str, cut_count: int = 100) -> str:
     if len(text) > 100:
         text_cut = text[:cut_count]
         size = len(text)
-        text_pretty = f'{text_cut}..(total {size} characters)'
+        text_pretty = f"{text_cut}..(total {size} characters)"
     else:
         text_pretty = text
-    text_pretty = text_pretty.replace('\n', '\\n')
+    text_pretty = text_pretty.replace("\n", "\\n")
     return text_pretty

{agi_med_common-4.2.12.dist-info → agi_med_common-4.2.15.dist-info}/METADATA

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

{agi_med_common-4.2.12.dist-info → agi_med_common-4.2.15.dist-info}/RECORD

@@ -1,20 +1,20 @@
-agi_med_common/__init__.py,sha256=7GZe14D48F8klVwUrPho7sj0WVGQYrKj4aGrjNhXjcU,722
+agi_med_common/__init__.py,sha256=1MDnHkF-GzakV5vMzftoMrTB5zuOUmdDlu2Zy0g9gsY,708
 agi_med_common/file_storage.py,sha256=yAP65qT0NbkaasFLhwnNzXOpd5vAetF3wP6Sbf_tbak,1766
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
-agi_med_common/utils.py,sha256=pG4ayRBBgYKfEBULV6gZgl7AbI_K7VfByoSsB1BCIEI,1044
+agi_med_common/utils.py,sha256=4juaxXWRBD8sqJ8mCHeP8O-T9d79ixjNvBEdrgPtLrw,1122
 agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
 agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
 agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
 agi_med_common/logger/logger.py,sha256=-VU69ULbw2-pNpOcxKiMpOeRAPe6H-EsGV9WRTv1SUo,839
-agi_med_common/models/__init__.py,sha256=KXnXxD2c5v2TVXrc-n7gJipTAyvtMasbocxThINIojs,330
+agi_med_common/models/__init__.py,sha256=Kek8n74nuozGZRL-brljKpYyXb1uWFEDu5ufrzrDq5o,289
 agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-agi_med_common/models/chat_item.py,sha256=0-CWepFcFKEyJS-Vbq21xpwLO0wbdFYHiWGqA6UKFFI,4376
-agi_med_common/models/enums.py,sha256=RxPPbkW1NcHAGO-gHuzwIvOjv0_ZBsr9b1aq_wk4WFM,1313
+agi_med_common/models/chat_item.py,sha256=gvJlRI-EIBMqsA7Cu_dBgjqsc4-aD80G-ONQ6PsyDJE,4416
+agi_med_common/models/enums.py,sha256=zv8BwaloRMp6TFEzdGsdcxnz97a4jKZ9zOSzWdG2QcY,1278
 agi_med_common/models/widget.py,sha256=g7Bobyk_gANghGEhjDjvi4i-YRoHo5Bm1Pr4eF-zf0U,661
 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-4.2.12.dist-info/METADATA,sha256=OrORy6dyLhEKWcE75DzpjkcAJLRfaNLWJ8lmUZf35jk,547
-agi_med_common-4.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-4.2.12.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-4.2.12.dist-info/RECORD,,
+agi_med_common-4.2.15.dist-info/METADATA,sha256=XmdlGA3cfIpF3ovXIkKnoP0dc_9prF5mg1YC6YlB0Bw,547
+agi_med_common-4.2.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-4.2.15.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.2.15.dist-info/RECORD,,