agi-med-common 5.0.4__py3-none-any.whl → 5.0.6__py3-none-any.whl

agi_med_common/__init__.py

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

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,33 @@ 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(err):
+    return err['type'] == 'string_type' and err['loc'] == ('InnerContext', 'Replicas', 0, 'Moderation') and err['msg'] == 'Input should be a valid string'
+
+def _is_moderation_int_errors(ex):
+    errs = ex.errors()
+    if len(errs) != 1:
+        return False
+    return all(map(_is_moderation_int_errors, errs))
+
+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.4.dist-info → agi_med_common-5.0.6.dist-info}/METADATA

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

{agi_med_common-5.0.4.dist-info → agi_med_common-5.0.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agi_med_common/__init__.py,sha256=P-VenrZNAAXke5KdFVPgPsj2n0X2IViXQ9TXeA5AiHw,759
+agi_med_common/__init__.py,sha256=a_kPHvngH3TR2fHLkFx62CYfzN-I_AkbcKlvi7s--AU,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
@@ -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=5mcHMx5it6cU91TYkwbT-fwKyJEqJ4crmnr5jVFNeDs,4914
 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.4.dist-info/METADATA,sha256=lNa_4Gsif9jHQyRw2j16Ca2MmWmPP2QS_v_OyENDNa0,546
-agi_med_common-5.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-5.0.4.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-5.0.4.dist-info/RECORD,,
+agi_med_common-5.0.6.dist-info/METADATA,sha256=f1_RfTWoxG2gwOaF7hjy9PML2-yU7oDrAo69A4TljFM,546
+agi_med_common-5.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.0.6.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.0.6.dist-info/RECORD,,