agi-med-common 3.5.4__tar.gz → 3.5.6__tar.gz

{agi_med_common-3.5.4 → agi_med_common-3.5.6}/PKG-INFO

@@ -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 → agi_med_common-3.5.6}/src/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-3.5.6/src/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-3.5.4 → agi_med_common-3.5.6}/src/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-3.5.4 → agi_med_common-3.5.6}/src/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-3.5.4 → agi_med_common-3.5.6}/src/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 → agi_med_common-3.5.6}/src/agi_med_common.egg-info/PKG-INFO

@@ -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 → agi_med_common-3.5.6}/src/agi_med_common.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 requirements.txt
 src/agi_med_common/__init__.py
+src/agi_med_common/file_storage.py
 src/agi_med_common/utils.py
 src/agi_med_common/validators.py
 src/agi_med_common/xml_parser.py