agi-med-common 5.0.3__py3-none-any.whl → 5.0.5__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.3"
+__version__ = "5.0.5"
 
 from .logger import LogLevelEnum, logger_init, log_llm_error
 from .models import (

agi_med_common/file_storage.py

@@ -56,8 +56,8 @@ class FileStorage:
     def download_text(self, rid: ResourceId) -> str:
         return Path(rid).read_text(encoding='utf-8')
 
-    def is_valid(self, rid: ResourceId) -> bytes:
-        return Path(rid).exists() and Path(rid).is_file()
+    def is_valid(self, rid: ResourceId | None) -> bool:
+        return rid and Path(rid).exists() and Path(rid).is_file()
 
-    def get_dtype(self, rid: ResourceId) -> str:
-        return rid.rsplit('.')[-1]
+    def get_dtype(self, rid: ResourceId | None) -> str | None:
+        return rid and rid.rsplit('.')[-1]

agi_med_common/models/__init__.py

@@ -1,7 +1,6 @@
 from ._base import _Base
 
 from .enums import MTRSLabelEnum, DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
-
 from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
 from .base_config_models import GigaChatConfig
 from .tracks import TrackInfo, DomainInfo

agi_med_common/models/chat_item.py

@@ -1,8 +1,10 @@
+import json
 from datetime import datetime
 from typing import Any, List
 
 from agi_med_common.models.widget import Widget
-from pydantic import Field, ConfigDict
+from loguru import logger
+from pydantic import Field, ConfigDict, ValidationError
 
 from ._base import _Base
 
@@ -93,3 +95,32 @@ class ChatItem(_Base):
 
     def zip_history(self, field: str) -> list[Any]:
         return [replica.to_dict().get(field, None) for replica in self.inner_context.replicas]
+
+    @classmethod
+    def parse(cls, chat_text: str) -> "ChatItem":
+        return _parse_chat(chat_text)
+
+
+def _is_moderation_int_error(ex):
+    errs = ex.errors()
+    if len(errs) != 1:
+        return False
+    err = errs[0]
+    if err['type'] == 'string_type' and err['loc'] == ('InnerContext', 'Replicas', 0, 'Moderation') and err['msg'] == 'Input should be a valid string':
+        return True
+
+def _parse_chat(chat_text: str) -> ChatItem:
+    try:
+        return ChatItem.model_validate_json(chat_text)
+    except ValidationError as ex:
+        if _is_moderation_int_error(ex):
+            logger.warning("Failed to parse ChatItem, fallback to old version with `Moderation:int`")
+        else:
+            logger.error(f"Failed to parse: {ex}")
+
+    # old version
+    chat_dict = json.loads(chat_text)
+    for rep in chat_dict["InnerContext"]["Replicas"]:
+        rep["Moderation"] = "OK"
+
+    return ChatItem.model_validate(chat_dict)

{agi_med_common-5.0.3.dist-info → agi_med_common-5.0.5.dist-info}/METADATA

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

{agi_med_common-5.0.3.dist-info → agi_med_common-5.0.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
-agi_med_common/__init__.py,sha256=gQinv4oyHhBnCJnLBgyW4U4S5PiiZzT4P1r7yxnFnlA,759
-agi_med_common/file_storage.py,sha256=yAP65qT0NbkaasFLhwnNzXOpd5vAetF3wP6Sbf_tbak,1766
+agi_med_common/__init__.py,sha256=OIJN7vCtRdYDTVy7gcMP5wqT1MT-khzZF9YV44Mi-6w,759
+agi_med_common/file_storage.py,sha256=T0Hbs4W-pWO6HdWcmlVqABrQHYdq7lLZH4_Vu-YNVbw,1802
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/utils.py,sha256=rBNAECUrvvyf-UZ650IpTfHvx5I8lYSnsB8QPBKti48,1276
 agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
@@ -8,15 +8,15 @@ agi_med_common/api/chat_manager_api.py,sha256=e-LJSmnWCFW1Wj5R3uavlKaq9p5VBoErqP
 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=Bptf6O5TClqu8m3vYlcp6ecj_bchohoaQr6dKMbh4YE,293
+agi_med_common/models/__init__.py,sha256=UbZmFKH94-AsvLUeg2y2odJ6WMjumI0grOJWOWL-aVg,292
 agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-agi_med_common/models/chat_item.py,sha256=owdBZ_3P9muJYTw4WLi5keB1ABjXBHYSq9fwojsb55U,3862
+agi_med_common/models/chat_item.py,sha256=zrX3ktArwZ-gFfS6TCiKYs5imLtINvE_QFX3hnHFKbk,4859
 agi_med_common/models/enums.py,sha256=J-GNpql9MCnKnWiV9aJRQGI-pAybvV86923RZs99grA,1006
 agi_med_common/models/tracks.py,sha256=UP-jeWqDiCK6dyoMDfs7hemgl_xsJKee_DApjBf-XYc,311
 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-5.0.3.dist-info/METADATA,sha256=Y4TvigH10-9rvz5aDP1fKUvuuK98RRGLUKKVqh3cPqc,546
-agi_med_common-5.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-5.0.3.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-5.0.3.dist-info/RECORD,,
+agi_med_common-5.0.5.dist-info/METADATA,sha256=HCcnLGZkxpY7rcP-gEV9TkOlfFO6d_sLklkVvltePCU,546
+agi_med_common-5.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.0.5.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.0.5.dist-info/RECORD,,