agi-med-common 1.4.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1 +1,16 @@
-__version__ = "1.4.0"
+__version__ = "2.0.0"
+
+from .logger import LogLevelEnum, logger_init
+from .models import (
+    TrackIdEnum,
+    StatesEnum,
+    ChatItem,
+    InnerContextItem,
+    OuterContextItem,
+    ReplicaItem,
+    ChatItemWithState,
+    InnerContextWithState,
+    ReplicaWithState,
+)
+from .utils import make_session_id, read_json, replace_n
+from .validators import is_file_exist, validate_prompt

agi_med_common/models/__init__.py

@@ -1,3 +1,5 @@
 from ._base import _Base
-from .enums import TrackIdEnum
+from .enums import TrackIdEnum, StatesEnum
 from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
+from .chat_item_with_state import ChatItemWithState, InnerContextWithState, ReplicaWithState
+from .base_config_models import GigaChatConfig

agi_med_common/models/base_config_models/__init__.py

@@ -0,0 +1 @@
+from .gigachat_config import GigaChatConfig

agi_med_common/models/base_config_models/gigachat_config.py

@@ -0,0 +1,15 @@
+from typing import Self
+
+from pydantic import BaseModel, model_validator, SecretStr
+
+
+class GigaChatConfig(BaseModel):
+
+    client_id: SecretStr = SecretStr("")
+    client_secret: SecretStr = SecretStr("")
+
+    @model_validator(mode="after")
+    def empty_validator(self) -> Self:
+        if not (self.client_id and self.client_secret):
+            raise ValueError("creds for gigachat is not filled")
+        return self

agi_med_common/models/chat_item/replica_item.py

@@ -1,10 +1,31 @@
-from pydantic import Field
+from datetime import datetime
+
+from pydantic import Field, field_validator, field_serializer
 
 from .. import _Base
 
+_TIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
+_EXAMPLE_DATETIME: datetime = datetime(1970, 1, 1, 0, 0, 0)
+
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
-    date_time: str = Field("1970-01-01-00-00-00", alias="DateTime", examples=["2024-12-10-18-03-46"])
-    previous_score: float | None = Field(None, alias="PreviousScore")
+    date_time: datetime = Field(
+        _EXAMPLE_DATETIME,
+        alias="DateTime",
+        examples=[_EXAMPLE_DATETIME.strftime(_TIME_FORMAT)],
+        description=f"Format: {_TIME_FORMAT}",
+    )
+    previous_score: float | None = Field(None, alias="PreviousScore", exclude=True, deprecated=True)
+
+    @field_validator("date_time", mode="before")
+    @classmethod
+    def convert_date_time(cls, timestamp: str | datetime) -> datetime:
+        if isinstance(timestamp, datetime):
+            return timestamp
+        return datetime.strptime(timestamp, _TIME_FORMAT)
+
+    @field_serializer("date_time")
+    def serialize_date_time(self, timestamp: datetime) -> str:
+        return timestamp.strftime(_TIME_FORMAT)

agi_med_common/models/chat_item_with_state/__init__.py

@@ -0,0 +1,3 @@
+from .replica_with_state import ReplicaWithState
+from .inner_context_with_state import InnerContextWithState
+from .chat_with_state import ChatItemWithState

agi_med_common/models/chat_item_with_state/chat_with_state.py

@@ -0,0 +1,8 @@
+from ..chat_item import ChatItem
+from pydantic import Field
+
+from . import InnerContextWithState
+
+
+class ChatItemWithState(ChatItem):
+    inner_context: InnerContextWithState = Field(alias="InnerContext")

agi_med_common/models/chat_item_with_state/inner_context_with_state.py

@@ -0,0 +1,8 @@
+from ..chat_item import InnerContextItem
+from pydantic import Field
+
+from . import ReplicaWithState
+
+
+class InnerContextWithState(InnerContextItem):
+    replicas: list[ReplicaWithState] = Field(alias="Replicas")

agi_med_common/models/chat_item_with_state/replica_with_state.py

@@ -0,0 +1,8 @@
+from ..chat_item import ReplicaItem
+from pydantic import Field
+
+from ..enums import StatesEnum
+
+
+class ReplicaWithState(ReplicaItem):
+    state: StatesEnum | None = Field(None, alias="State")

agi_med_common/models/enums/__init__.py

@@ -1 +1,2 @@
 from .track_id_enum import TrackIdEnum
+from .states_enum import StatesEnum

agi_med_common/models/enums/states_enum.py

@@ -0,0 +1,29 @@
+from enum import StrEnum, auto
+
+
+class StatesEnum(StrEnum):
+    """
+    Класс для хранения имен состояний.
+    """
+
+    BEGIN = auto()
+    START = auto()
+
+    IS_CHILD = auto()
+    IS_ABSURD = auto()
+    CRITICAL = auto()
+    WHAT_COMPLAINTS = auto()
+    NOT_MEDICAL = auto()
+
+    ERROR_STATE = auto()
+
+    CONSULTATION_TRANSIT = auto()
+    NOT_MEDICAL_DOC = auto()
+    CONSULTATION = auto()
+
+    INFO_COLLECTION = auto()
+    MAKE_DIAGNOSIS = auto()
+
+    SUMMARIZATION = auto()
+
+    ANALYSIS_CONSULTATION = auto()

agi_med_common/utils.py

@@ -0,0 +1,18 @@
+import codecs
+import json
+import os
+from datetime import datetime
+from pathlib import Path
+
+
+def make_session_id() -> str:
+    return f"{datetime.now():%y%m%d%H%M%S}"
+
+
+def read_json(path: Path | os.PathLike[str] | str) -> list | dict:
+    with codecs.open(path, "r", encoding="utf8") as file:
+        return json.load(file)
+
+
+def replace_n(value: str) -> str:
+    return value.replace("\\n", "\n")

agi_med_common/validators.py

@@ -0,0 +1,18 @@
+import os
+import re
+from pathlib import Path
+
+
+def is_file_exist(filepath: str | os.PathLike[str] | Path) -> str | os.PathLike | Path:
+    if not os.path.exists(filepath):
+        raise ValueError(f"File {filepath} is not exist. Check the path")
+    return filepath
+
+
+def validate_prompt(prompt: str, prompt_required_keys: set[str]) -> str:
+    exist_keys: set[str] = set(re.findall(r"{(.*?)}", prompt))
+    if missed_keys := prompt_required_keys.difference(exist_keys):
+        raise ValueError(f"Missing required key in prompt: {missed_keys}")
+    if extern_keys := exist_keys.difference(prompt_required_keys):
+        raise ValueError(f"You have more keys for prompt: {extern_keys}")
+    return prompt

{agi_med_common-1.4.0.dist-info → agi_med_common-2.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: agi_med_common
-Version: 1.4.0
+Version: 2.0.0
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.13

agi_med_common-2.0.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+agi_med_common/__init__.py,sha256=0GJulrzkBDkB17nr_utQW7ghrK5yCMq30GZKK9aV3wY,385
+agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
+agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
+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=BrLFWe_103S7uFHX2UeGpZdF-wuz7up4l6gfQwQjJoA,1263
+agi_med_common/models/__init__.py,sha256=kBCFA9ctcBRm9JrrEs4HHGYcXERMaoEI8p5N6kN2JAY,289
+agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
+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/models/chat_item/__init__.py,sha256=MoZFnZQV-tWJTHmH1OhRBRkrc5cGWAfdJ6QfHDg6TDk,168
+agi_med_common/models/chat_item/chat_item.py,sha256=aCRR4OwL-YS6hvVSNkIg7FNpfvEGo8BESCY8BhiAlNk,255
+agi_med_common/models/chat_item/inner_context_item.py,sha256=s17frAcwyi8WcHnBt-5LF8nUJeO-hnu7YMobYQO5a-c,166
+agi_med_common/models/chat_item/outer_context_item.py,sha256=IprCIh8ki3PUueFwdmPw-MYn0-prx_G523Y-IXN62ZY,792
+agi_med_common/models/chat_item/replica_item.py,sha256=_OvnGmbP-wzxi16c6FGz-_Y-QtudV1Pi6I-zCJ7aQfA,1131
+agi_med_common/models/chat_item_with_state/__init__.py,sha256=KrXgsvJnLzS1kSi9_Y1z6_egxIqKL7epY1r-PIyrnno,156
+agi_med_common/models/chat_item_with_state/chat_with_state.py,sha256=IgZh0CRoar56eM8QOblPx8tsIm9NPO5tOj0mVC8yj40,205
+agi_med_common/models/chat_item_with_state/inner_context_with_state.py,sha256=qIwqCHn-JuQemcVQqLDzkpuaPRNCjIL_-h1wm1Jv5WU,212
+agi_med_common/models/chat_item_with_state/replica_with_state.py,sha256=YvUcCo1HTfxYUEwZ5T7UqLW2WkZ8SFvVNQZf-5mK5Dg,192
+agi_med_common/models/enums/__init__.py,sha256=yHUFI7n5r-LY5y4CbliZlXDrmr38bsz8dh4iyNo5Ti8,75
+agi_med_common/models/enums/states_enum.py,sha256=IerXA4uaCNnE0zt4JF4bveqwL1wpUUBMXMh6kqFV_XM,543
+agi_med_common/models/enums/track_id_enum.py,sha256=fHGQz9AACIS6SfTFCrSau1HKee70r2bap-OcBxD3YRE,423
+agi_med_common-2.0.0.dist-info/METADATA,sha256=VTyClWQ9zBYJdascshkCrq_hXlMsvCFyU3qxY8nz4Sg,520
+agi_med_common-2.0.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agi_med_common-2.0.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-2.0.0.dist-info/RECORD,,

{agi_med_common-1.4.0.dist-info → agi_med_common-2.0.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

agi_med_common-1.4.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-agi_med_common/__init__.py,sha256=8UhoYEXHs1Oai7BW_ExBmuwWnRI-yMG_u1fQAXMizHQ,22
-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=BrLFWe_103S7uFHX2UeGpZdF-wuz7up4l6gfQwQjJoA,1263
-agi_med_common/models/__init__.py,sha256=MxRz8pUM0U7QfLSDsjDX4dMQQHEDJu6179RibFWDYQc,137
-agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-agi_med_common/models/chat_item/__init__.py,sha256=MoZFnZQV-tWJTHmH1OhRBRkrc5cGWAfdJ6QfHDg6TDk,168
-agi_med_common/models/chat_item/chat_item.py,sha256=aCRR4OwL-YS6hvVSNkIg7FNpfvEGo8BESCY8BhiAlNk,255
-agi_med_common/models/chat_item/inner_context_item.py,sha256=s17frAcwyi8WcHnBt-5LF8nUJeO-hnu7YMobYQO5a-c,166
-agi_med_common/models/chat_item/outer_context_item.py,sha256=IprCIh8ki3PUueFwdmPw-MYn0-prx_G523Y-IXN62ZY,792
-agi_med_common/models/chat_item/replica_item.py,sha256=c561A7EpjjaB7fr7DPd6hTCphAH1uFXGSiF74E5L-jY,419
-agi_med_common/models/enums/__init__.py,sha256=ooZLUfkRKeJDeTROAAqdRsMoqKBKmgnuGZXWQf9r83Y,39
-agi_med_common/models/enums/track_id_enum.py,sha256=fHGQz9AACIS6SfTFCrSau1HKee70r2bap-OcBxD3YRE,423
-agi_med_common-1.4.0.dist-info/METADATA,sha256=nUj-kQJAJ5TXGleZPhKPAiTh43vplQPGr3d7I_xHZfA,520
-agi_med_common-1.4.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-agi_med_common-1.4.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-1.4.0.dist-info/RECORD,,