agi-med-common 5.0.4__tar.gz → 5.0.5__tar.gz

{agi_med_common-5.0.4 → agi_med_common-5.0.5}/PKG-INFO

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

{agi_med_common-5.0.4 → agi_med_common-5.0.5}/src/agi_med_common/__init__.py

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

{agi_med_common-5.0.4 → agi_med_common-5.0.5}/src/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-5.0.4 → agi_med_common-5.0.5}/src/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.4 → agi_med_common-5.0.5}/src/agi_med_common.egg-info/PKG-INFO

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