agi-med-common 3.5.4__py3-none-any.whl → 3.5.6__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.5.4"
+__version__ = "3.5.6"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -16,6 +16,6 @@ from .models import (
 
 from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 
-from .utils import make_session_id, read_json, replace_n
+from .utils import make_session_id, read_json
 from .validators import is_file_exist, validate_prompt
 from .xml_parser import XMLParser

agi_med_common/file_storage.py

@@ -0,0 +1,39 @@
+import string
+from hashlib import md5
+from pathlib import Path
+
+ResourceId = str
+ASCII_DIGITS = set(string.ascii_lowercase + string.digits)
+
+
+def _validate_exist(files_dir):
+    if not files_dir.exists():
+        err = f"Failed to access file-storage directory: {files_dir}"
+        raise OSError(err)
+
+
+def _validate_dtype(dtype: str):
+    if all(map(ASCII_DIGITS.__contains__, dtype)):
+        return
+    raise ValueError(f'Bad dtype: {dtype}')
+
+
+class FileStorage:
+    def __init__(self, files_dir):
+        self.files_dir = Path(files_dir)
+        self.files_dir.mkdir(exist_ok=True)
+        _validate_exist(self.files_dir)
+
+    def upload(self, content: bytes, dtype: str) -> ResourceId | None:
+        _validate_dtype(dtype)
+        fname_hash = md5(content).hexdigest()
+        fname = f'{fname_hash}.{dtype}'
+        fpath = self.files_dir / fname
+        fpath.write_bytes(content)
+        return str(fpath)
+
+    def download(self, rid: ResourceId) -> bytes:
+        return Path(rid).read_bytes()
+
+    def is_valid(self, rid: ResourceId) -> bytes:
+        return Path(rid).exists() and Path(rid).is_file()

agi_med_common/models/chat_item/replica_item.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any
+from typing import Any, Optional
 
 from pydantic import Field
 
@@ -12,6 +12,7 @@ _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>"])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
     date_time: str = Field(
         _EXAMPLE_DATETIME,

agi_med_common/models/enums/track_id_enum.py

@@ -14,3 +14,4 @@ class TrackIdEnum(StrEnum):
     DUMMY = auto()
     COMMON_CONSULTATION = auto()
     FINANCIER = auto()
+    MULTIMODAL = auto()

agi_med_common/utils.py

@@ -12,7 +12,3 @@ def make_session_id() -> str:
 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-3.5.4.dist-info → agi_med_common-3.5.6.dist-info}/METADATA

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

{agi_med_common-3.5.4.dist-info → agi_med_common-3.5.6.dist-info}/RECORD

@@ -1,5 +1,6 @@
-agi_med_common/__init__.py,sha256=xl4bes4OjoEwc3IBkPUrKT3tn4NdiUxKNT-3Swjtcgc,511
-agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
+agi_med_common/__init__.py,sha256=zKCQJV8k1zNUd57rQqySzZGi4iUOAjiu9uGOSQ_Zfa8,500
+agi_med_common/file_storage.py,sha256=W6MOWBrO20y-xnW3FIR8vABK2PazRk3-Jm853mVXttk,1115
+agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
 agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
@@ -13,7 +14,7 @@ agi_med_common/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTg
 agi_med_common/models/chat_item/chat_item.py,sha256=Ym-afB9-7NI6QquhMPqd2ZBkbcx-Txr-ZiXwn0PTO-0,1182
 agi_med_common/models/chat_item/inner_context_item.py,sha256=YXmlNLabKN3O84UWJENsjfdtPq2XRH1VqKiyIxS-p64,255
 agi_med_common/models/chat_item/outer_context_item.py,sha256=IzrwhqbLKR8ZF8-Sd2CyNSa0eqb21NQH7LUqkg9se2Y,1046
-agi_med_common/models/chat_item/replica_item.py,sha256=O4QJlXQ_HaxtH3CEAYdxULCaMbJ6nTHPEDsZ6ZeoDDw,1507
+agi_med_common/models/chat_item/replica_item.py,sha256=1Sa73Y-LaGhsjTr5YfQ_ay7gai7VAgKuY7ZoOgMrS4Y,1606
 agi_med_common/models/enums/__init__.py,sha256=wnI9ub2F1L-TTB2jIpVF3OTLFr0QzcQjW3pX4P45HXo,377
 agi_med_common/models/enums/action_enum.py,sha256=-aFNyHQ1yyCJedSrFubcHuLAiLZLP6KSs_5ff9lrR3I,320
 agi_med_common/models/enums/diagnostics_xml_tag_enum.py,sha256=aiUJpe5joh5EOcJ5sk1Xs003XkRry2CBL6H-bIEiapA,278
@@ -22,8 +23,8 @@ agi_med_common/models/enums/moderation_label_enum.py,sha256=lbGG4Pu7cQp57uEyQEpt
 agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
 agi_med_common/models/enums/mtrs_xml_tag_enum.py,sha256=6OxuRsrx4b2uBjfrBgm4Y789Ly337_mQXL9VPRCpLyg,273
 agi_med_common/models/enums/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
-agi_med_common/models/enums/track_id_enum.py,sha256=P2b89MP7E3ePZ8bZR2Db12su_Z6r6K45e3090fugFUY,477
-agi_med_common-3.5.4.dist-info/METADATA,sha256=SCUgVS-kUV9BeGrpSdcVUMQIAJCXgdbk0GJxTGhWHFM,520
-agi_med_common-3.5.4.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-agi_med_common-3.5.4.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.5.4.dist-info/RECORD,,
+agi_med_common/models/enums/track_id_enum.py,sha256=KM4DbJ6fzK4xPSghKBaTdAhz_SlTSLxJvnKJoMjc9QE,501
+agi_med_common-3.5.6.dist-info/METADATA,sha256=aFw75dqX_AI5a28CNpQ6miZxyyS-4mg-VDCsWPt_uyw,520
+agi_med_common-3.5.6.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+agi_med_common-3.5.6.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.5.6.dist-info/RECORD,,

{agi_med_common-3.5.4.dist-info → agi_med_common-3.5.6.dist-info}/WHEEL

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