agi-med-common 4.2.5__tar.gz → 4.2.6__tar.gz

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/PKG-INFO

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

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common/__init__.py

@@ -1,8 +1,7 @@
-__version__ = "4.2.5"
+__version__ = "4.2.6"
 
-from .logger import LogLevelEnum, logger_init
+from .logger import LogLevelEnum, logger_init, log_llm_error
 from .models import (
-    TrackIdEnum,
     StateEnum,
     MTRSLabelEnum,
     ActionEnum,
@@ -13,6 +12,7 @@ from .models import (
     ReplicaItem,
     ReplicaItemPair,
 )
+from .models.widget import Widget
 from .file_storage import FileStorage
 from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 from .utils import make_session_id, read_json

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common/models/__init__.py

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

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common/models/chat_item/outer_context_item.py

@@ -2,7 +2,6 @@ from pydantic import Field, ConfigDict
 from typing import Any
 
 from .. import _Base
-from ..enums import TrackIdEnum
 
 
 class OuterContextItem(_Base):
@@ -14,7 +13,7 @@ class OuterContextItem(_Base):
     user_id: str = Field("", alias="UserId", examples=["123456789"])
     session_id: str = Field("", alias="SessionId", examples=["987654321"])
     client_id: str = Field("", alias="ClientId", examples=["543216789"])
-    track_id: TrackIdEnum | str = Field(default=TrackIdEnum.CONSULTATION, alias="TrackId")
+    track_id: str = Field(default="Consultation", alias="TrackId")
     entrypoint_key: str = Field("", alias="EntrypointKey", examples=["giga"])
 
     def create_id(self, short: bool = False, clean: bool = False) -> str:

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common/models/chat_item/replica_item.py

@@ -1,10 +1,11 @@
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field, ConfigDict
 
 from .. import _Base
 from ..enums import StateEnum, ActionEnum, ModerationLabelEnum
+from agi_med_common.models.widget import Widget
 
 
 _DATETIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
@@ -13,7 +14,8 @@ _EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_DATETIME_FORMAT
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
-    resource_id: Optional[str] = Field(None, alias="ResourceId", examples=["<link-id>"])
+    resource_id: str | None = Field(None, alias="ResourceId", examples=["<link-id>"])
+    widget: Widget | None = Field(None, alias="Widget", exampels=[None])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
     date_time: str = Field(
         _EXAMPLE_DATETIME,
@@ -37,6 +39,10 @@ class ReplicaItem(_Base):
     def DATETIME_FORMAT() -> str:
         return _DATETIME_FORMAT
 
+    def with_now_datetime(self):
+        dt = datetime.now().strftime(ReplicaItem.DATETIME_FORMAT())
+        return self.model_copy(update=dict(date_time=dt))
+
 
 class ReplicaItemPair(_Base):
     # remove annoying warning for protected `model_` namespace

agi_med_common-4.2.6/src/agi_med_common/models/enums.py

@@ -0,0 +1,58 @@
+from enum import IntEnum, StrEnum, auto
+
+
+StateEnum, ActionEnum = str, str
+
+
+class ModerationLabelEnum(IntEnum):
+    @staticmethod
+    def _generate_next_value_(name, start, count, last_values):
+        return count
+
+    OK = auto()
+    NON_MED = auto()
+    CHILD = auto()
+    ABSURD = auto()
+    GREETING = auto()
+    RECEIPT = auto()
+
+
+class DoctorChoiceXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"{name.lower()}"
+
+    DIAGNOSTICS = auto()
+    SUMMARIZATION = auto()
+    MTRS = auto()
+
+
+class MTRSXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"mtrs_{name.lower()}"
+
+    NAME = auto()
+    LABEL = auto()
+    DESC = auto()
+
+
+class MTRSLabelEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return name.upper()
+
+    LABORATORY = auto()
+    INSTRUMENTAL = auto()
+
+
+class DiagnosticsXMLTagEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return f"diag_{name.lower()}"
+
+    DIAG = auto()
+    DOC = auto()
+    DESC = auto()
+
+

agi_med_common-4.2.6/src/agi_med_common/models/widget.py

@@ -0,0 +1,20 @@
+from typing import List, Self
+from pydantic import BaseModel, model_validator
+
+
+class Widget(BaseModel):
+    buttons: List[List[str]] | None = None
+    ibuttons: List[List[str]] | None = None
+
+    @model_validator(mode="after")
+    def check(self) -> Self:
+        if not self.buttons and not self.ibuttons:
+            raise ValueError("Empty widget is not allowed!")
+        if not self.ibuttons:
+            return self
+        for row in self.ibuttons:
+            for btn in row:
+                if ":" in btn:
+                    continue
+                raise ValueError(f"Expected buttons like `<callback>:<caption>`, found: {btn}")
+        return self

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common.egg-info/PKG-INFO

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

{agi_med_common-4.2.5 → agi_med_common-4.2.6}/src/agi_med_common.egg-info/SOURCES.txt

@@ -17,17 +17,12 @@ src/agi_med_common/logger/log_level_enum.py
 src/agi_med_common/logger/logger.py
 src/agi_med_common/models/__init__.py
 src/agi_med_common/models/_base.py
+src/agi_med_common/models/enums.py
+src/agi_med_common/models/widget.py
 src/agi_med_common/models/base_config_models/__init__.py
 src/agi_med_common/models/base_config_models/gigachat_config.py
 src/agi_med_common/models/chat_item/__init__.py
 src/agi_med_common/models/chat_item/chat_item.py
 src/agi_med_common/models/chat_item/inner_context_item.py
 src/agi_med_common/models/chat_item/outer_context_item.py
-src/agi_med_common/models/chat_item/replica_item.py
-src/agi_med_common/models/enums/__init__.py
-src/agi_med_common/models/enums/diagnostics_xml_tag_enum.py
-src/agi_med_common/models/enums/doctor_choice_xml_tag_enum.py
-src/agi_med_common/models/enums/moderation_label_enum.py
-src/agi_med_common/models/enums/mtrs_label_enum.py
-src/agi_med_common/models/enums/mtrs_xml_tag_enum.py
-src/agi_med_common/models/enums/track_id_enum.py
+src/agi_med_common/models/chat_item/replica_item.py

agi_med_common-4.2.5/src/agi_med_common/models/enums/__init__.py

@@ -1,9 +0,0 @@
-from .track_id_enum import TrackIdEnum
-from .mtrs_label_enum import MTRSLabelEnum
-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-4.2.5/src/agi_med_common/models/enums/diagnostics_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class DiagnosticsXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"diag_{name.lower()}"
-
-    DIAG = auto()
-    DOC = auto()
-    DESC = auto()

agi_med_common-4.2.5/src/agi_med_common/models/enums/doctor_choice_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class DoctorChoiceXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"{name.lower()}"
-
-    DIAGNOSTICS = auto()
-    SUMMARIZATION = auto()
-    MTRS = auto()

agi_med_common-4.2.5/src/agi_med_common/models/enums/moderation_label_enum.py

@@ -1,14 +0,0 @@
-from enum import IntEnum, auto
-
-
-class ModerationLabelEnum(IntEnum):
-    @staticmethod
-    def _generate_next_value_(name, start, count, last_values):
-        return count
-
-    OK = auto()
-    NON_MED = auto()
-    CHILD = auto()
-    ABSURD = auto()
-    GREETING = auto()
-    RECEIPT = auto()

agi_med_common-4.2.5/src/agi_med_common/models/enums/mtrs_label_enum.py

@@ -1,10 +0,0 @@
-from enum import StrEnum, auto
-
-
-class MTRSLabelEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return name.upper()
-
-    LABORATORY = auto()
-    INSTRUMENTAL = auto()

agi_med_common-4.2.5/src/agi_med_common/models/enums/mtrs_xml_tag_enum.py

@@ -1,11 +0,0 @@
-from enum import StrEnum, auto
-
-
-class MTRSXMLTagEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return f"mtrs_{name.lower()}"
-
-    NAME = auto()
-    LABEL = auto()
-    DESC = auto()

agi_med_common-4.2.5/src/agi_med_common/models/enums/track_id_enum.py

@@ -1,18 +0,0 @@
-from enum import StrEnum, auto
-
-
-class TrackIdEnum(StrEnum):
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
-        return "".join([word.title() for word in name.lower().split("_")])
-
-    DIAGNOSTIC = auto()
-    DIAGNOSTIC_SBERHEALTH = auto()
-    SECOND_OPINION = auto()
-    MEDICAL_TEST_DECRYPTION = auto()
-    CONSULTATION = auto()
-    DUMMY = auto()
-    COMMON_CONSULTATION = auto()
-    FINANCIER = auto()
-    FINANCIER_GENERIC = auto()
-    MULTIMODAL = auto()