agi-med-common 4.2.9__py3-none-any.whl → 4.2.11__py3-none-any.whl

agi_med_common/__init__.py

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

agi_med_common/file_storage.py

@@ -53,6 +53,9 @@ class FileStorage:
     def download(self, rid: ResourceId) -> bytes:
         return Path(rid).read_bytes()
 
+    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()
 

agi_med_common/models/chat_item.py

@@ -0,0 +1,115 @@
+from datetime import datetime
+from typing import Any
+
+from pydantic import Field, ConfigDict
+from agi_med_common.models.widget import Widget
+
+from .enums import ModerationLabelEnum, StateEnum, ActionEnum
+from ._base import _Base
+
+
+_DATETIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
+_EXAMPLE_DATETIME_0 = datetime(1970, 1, 1, 0, 0, 0)
+_EXAMPLE_DATETIME: str = _EXAMPLE_DATETIME_0.strftime(_DATETIME_FORMAT)
+
+
+class OuterContextItem(_Base):
+    # remove annoying warning for protected `model_` namespace
+    model_config = ConfigDict(protected_namespaces=())
+
+    sex: bool = Field(
+        False,
+        alias="Sex",
+        description="True = male, False = female",
+        examples=[True]
+    )
+    age: int = Field(0, alias="Age", examples=[20])
+    user_id: str = Field("", alias="UserId", examples=["123456789"])
+    parent_session_id: str | None = Field(
+        None,
+        alias="ParentSessionId",
+        examples=["987654320"]
+    )
+    session_id: str = Field("", alias="SessionId", examples=["987654321"])
+    client_id: str = Field("", alias="ClientId", examples=["543216789"])
+    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:
+        uid, sid, cid = self.user_id, self.session_id, self.client_id
+        if short:
+            return f"{uid}_{sid}_{cid}"
+        if not clean:
+            return f"user_{uid}_session_{sid}_client_{cid}"
+        return f"user_{uid}_session_{sid}_client_{cid}_clean"
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)
+
+
+class ReplicaItem(_Base):
+    body: str = Field("", alias="Body", examples=["Привет"])
+    resource_id: str | None = Field(None, alias="ResourceId", examples=["<link-id>"])
+    widget: Widget | None = Field(None, alias="Widget", examples=[None])
+    command: dict | None = Field(None, alias="Command", examples=[None])
+    role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
+    date_time: str = Field(
+        _EXAMPLE_DATETIME,
+        alias="DateTime",
+        examples=[_EXAMPLE_DATETIME],
+        description=f"Format: {_DATETIME_FORMAT}",
+    )
+    state: StateEnum = Field("EMPTY", alias="State", description="chat manager fsm state", examples=["COLLECTION"])
+    action: ActionEnum = Field("START", alias="Action", description="chat manager fsm action", examples=["DIAGNOSIS"])
+    moderation: ModerationLabelEnum = Field(
+        ModerationLabelEnum.OK,
+        alias="Moderation",
+        description="chat manager moderated outcome type",
+        examples=[ModerationLabelEnum.NON_MED],
+    )
+    extra: dict | None = Field(None, alias="Extra", examples=[None])
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)
+
+    @staticmethod
+    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 InnerContextItem(_Base):
+    replicas: list[ReplicaItem] = Field(alias="Replicas")
+    attrs: dict[str, str | int] | None = Field(default={}, alias="Attrs")
+
+    def to_dict(self) -> dict[str, list]:
+        return self.model_dump(by_alias=True)
+
+
+class ReplicaItemPair(_Base):
+    # remove annoying warning for protected `model_` namespace
+    model_config = ConfigDict(protected_namespaces=())
+
+    user_replica: ReplicaItem = Field(alias="UserReplica")
+    bot_replica: ReplicaItem = Field(alias="BotReplica")
+
+
+class ChatItem(_Base):
+    outer_context: OuterContextItem = Field(alias="OuterContext")
+    inner_context: InnerContextItem = Field(alias="InnerContext")
+
+    def create_id(self, short: bool = False, clean: bool = False) -> str:
+        return self.outer_context.create_id(short, clean)
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)
+
+    def update(self, replica_pair: ReplicaItemPair) -> None:
+        self.inner_context.replicas.append(replica_pair.user_replica)
+        self.inner_context.replicas.append(replica_pair.bot_replica)
+
+    def zip_history(self, field: str) -> list[Any]:
+        return [replica.to_dict().get(field, None) for replica in self.inner_context.replicas]

{agi_med_common-4.2.9.dist-info → agi_med_common-4.2.11.dist-info}/METADATA

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

{agi_med_common-4.2.9.dist-info → agi_med_common-4.2.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
-agi_med_common/__init__.py,sha256=5KSML4nPmtbPIbePp_Cdjp0hPgVoRzVtQDMp8SnYHrM,708
-agi_med_common/file_storage.py,sha256=4GW0CU0YtVzWmxV6vGjZg0DfyEgjlrgLRsZ7cEP-6Qk,1659
+agi_med_common/__init__.py,sha256=Vp254TnyEq5ms2wa8pY4mksfVPI_UwriJbj6O_qoXF4,709
+agi_med_common/file_storage.py,sha256=yAP65qT0NbkaasFLhwnNzXOpd5vAetF3wP6Sbf_tbak,1766
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/utils.py,sha256=42okDsaCIkxcBlBojyTOqPH1QUGkginjvgyPhSnRq9w,725
 agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
@@ -9,16 +9,12 @@ agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjw
 agi_med_common/logger/logger.py,sha256=-VU69ULbw2-pNpOcxKiMpOeRAPe6H-EsGV9WRTv1SUo,839
 agi_med_common/models/__init__.py,sha256=KXnXxD2c5v2TVXrc-n7gJipTAyvtMasbocxThINIojs,330
 agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
+agi_med_common/models/chat_item.py,sha256=0-CWepFcFKEyJS-Vbq21xpwLO0wbdFYHiWGqA6UKFFI,4376
 agi_med_common/models/enums.py,sha256=RxPPbkW1NcHAGO-gHuzwIvOjv0_ZBsr9b1aq_wk4WFM,1313
 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/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTgdV43TZOcfHasHg,185
-agi_med_common/models/chat_item/chat_item.py,sha256=xP_uP14qK-u8z0ZUs2HFhhea-IP230UgOmjeqeC_N6Q,1180
-agi_med_common/models/chat_item/inner_context_item.py,sha256=2wIwDtZwicwhSyXeuOqH8-xONcskeEiw6QhjSnEEwZs,329
-agi_med_common/models/chat_item/outer_context_item.py,sha256=P2ulGbKnEXBJ50Ou-_p_qKCdO1lB4SUvdSu1ZG1yRdo,1213
-agi_med_common/models/chat_item/replica_item.py,sha256=cGnAyUxNM50004EWKc8q4nYvy9cqZ8fxPdTilja6fBU,2013
-agi_med_common-4.2.9.dist-info/METADATA,sha256=axCVcE7uged5BEl3_h_GSzNTlplY1Ed20rPr_vfC1fs,546
-agi_med_common-4.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-4.2.9.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-4.2.9.dist-info/RECORD,,
+agi_med_common-4.2.11.dist-info/METADATA,sha256=jsQrpTz0nUWpKIYstKOHWL5Xwgtmw7O62RWlSEoL4zA,547
+agi_med_common-4.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-4.2.11.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.2.11.dist-info/RECORD,,

agi_med_common/models/chat_item/__init__.py

@@ -1,4 +0,0 @@
-from .replica_item import ReplicaItem, ReplicaItemPair
-from .outer_context_item import OuterContextItem
-from .inner_context_item import InnerContextItem
-from .chat_item import ChatItem

agi_med_common/models/chat_item/chat_item.py

@@ -1,34 +0,0 @@
-from pydantic import Field
-from typing import Any
-
-from . import OuterContextItem, InnerContextItem, ReplicaItemPair
-from ..enums import ActionEnum
-from .. import _Base
-
-
-class ChatItem(_Base):
-    outer_context: OuterContextItem = Field(alias="OuterContext")
-    inner_context: InnerContextItem = Field(alias="InnerContext")
-
-    def create_id(self, short: bool = False, clean: bool = False) -> str:
-        return self.outer_context.create_id(short, clean)
-
-    def to_dict(self) -> dict[str, Any]:
-        return self.model_dump(by_alias=True)
-
-    def update(self, replica_pair: ReplicaItemPair) -> None:
-        self.inner_context.replicas.append(replica_pair.user_replica)
-        self.inner_context.replicas.append(replica_pair.bot_replica)
-
-    def count_questions(self) -> int:
-        return len(
-            list(
-                filter(
-                    lambda replica: replica.role and replica.action == ActionEnum.QUESTION,
-                    self.inner_context.replicas,
-                )
-            )
-        )
-
-    def zip_history(self, field: str) -> list[Any]:
-        return [replica.to_dict().get(field, None) for replica in self.inner_context.replicas]

agi_med_common/models/chat_item/inner_context_item.py

@@ -1,12 +0,0 @@
-from pydantic import Field
-
-from . import ReplicaItem
-from .. import _Base
-
-
-class InnerContextItem(_Base):
-    replicas: list[ReplicaItem] = Field(alias="Replicas")
-    attrs: dict[str, str | int] | None = Field(default={}, alias="Attrs")
-
-    def to_dict(self) -> dict[str, list]:
-        return self.model_dump(by_alias=True)

agi_med_common/models/chat_item/outer_context_item.py

@@ -1,27 +0,0 @@
-from pydantic import Field, ConfigDict
-from typing import Any
-
-from .. import _Base
-
-
-class OuterContextItem(_Base):
-    # remove annoying warning for protected `model_` namespace
-    model_config = ConfigDict(protected_namespaces=())
-
-    sex: bool = Field(False, alias="Sex", description="True = male, False = female", examples=[True])
-    age: int = Field(0, alias="Age", examples=[20])
-    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: 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:
-        if short:
-            return f"{self.user_id}_{self.session_id}_{self.client_id}"
-        if not clean:
-            return f"user_{self.user_id}_session_{self.session_id}_client_{self.client_id}"
-        return f"user_{self.user_id}_session_{self.session_id}_client_{self.client_id}_clean"
-
-    def to_dict(self) -> dict[str, Any]:
-        return self.model_dump(by_alias=True)

agi_med_common/models/chat_item/replica_item.py

@@ -1,53 +0,0 @@
-from datetime import datetime
-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"
-_EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_DATETIME_FORMAT)
-
-
-class ReplicaItem(_Base):
-    body: str = Field("", alias="Body", examples=["Привет"])
-    resource_id: str | None = Field(None, alias="ResourceId", examples=["<link-id>"])
-    widget: Widget | None = Field(None, alias="Widget", examples=[None])
-    command: dict | None = Field(None, alias="Command", examples=[None])
-    role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
-    date_time: str = Field(
-        _EXAMPLE_DATETIME,
-        alias="DateTime",
-        examples=[_EXAMPLE_DATETIME],
-        description=f"Format: {_DATETIME_FORMAT}",
-    )
-    state: StateEnum = Field("EMPTY", alias="State", description="chat manager fsm state", examples=["COLLECTION"])
-    action: ActionEnum = Field("START", alias="Action", description="chat manager fsm action", examples=["DIAGNOSIS"])
-    moderation: ModerationLabelEnum = Field(
-        ModerationLabelEnum.OK,
-        alias="Moderation",
-        description="chat manager moderated outcome type",
-        examples=[ModerationLabelEnum.NON_MED],
-    )
-
-    def to_dict(self) -> dict[str, Any]:
-        return self.model_dump(by_alias=True)
-
-    @staticmethod
-    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
-    model_config = ConfigDict(protected_namespaces=())
-
-    user_replica: ReplicaItem = Field(alias="UserReplica")
-    bot_replica: ReplicaItem = Field(alias="BotReplica")