agi-med-common 4.0.6__py3-none-any.whl → 4.2.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "4.0.6"
+__version__ = "4.2.0"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -16,6 +16,6 @@ from .models import (
 from .file_storage import FileStorage
 from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 from .utils import make_session_id, read_json
-from .validators import is_file_exist, validate_prompt
+from .validators import ExistingPath, ExistingFile, StrNotEmpty, Prompt
 from .xml_parser import XMLParser
 from .parallel_map import parallel_map

agi_med_common/models/chat_item/outer_context_item.py

@@ -1,4 +1,4 @@
-from pydantic import Field
+from pydantic import Field, ConfigDict
 from typing import Any
 
 from .. import _Base
@@ -6,12 +6,16 @@ from ..enums import TrackIdEnum
 
 
 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: TrackIdEnum | str = Field(default=TrackIdEnum.CONSULTATION, alias="TrackId")
+    model_id: str = Field("", alias="ModelId", examples=["gigachat-2"])
 
     def create_id(self, short: bool = False, clean: bool = False) -> str:
         if short:

agi_med_common/validators.py

@@ -1,14 +1,38 @@
 import os
 import re
 from pathlib import Path
+from typing import Annotated, TypeVar
+from pydantic import AfterValidator
 
 
-def is_file_exist(filepath: str | os.PathLike[str] | Path) -> str | os.PathLike | Path:
+PathLike = str | os.PathLike[str] | Path
+
+
+def validate_existing_path(filepath: PathLike) -> PathLike:
     if not os.path.exists(filepath):
-        raise ValueError(f"File {filepath} is not exist. Check the path")
+        raise ValueError(f"Path {filepath} is not exist")
+    return filepath
+
+
+def validate_existing_file(filepath: PathLike) -> PathLike:
+    if not os.path.isfile(filepath):
+        raise ValueError(f"Path {filepath} is not file")
     return filepath
 
 
+ExistingPath = Annotated[Path, AfterValidator(validate_existing_path)]
+ExistingFile = Annotated[Path, AfterValidator(validate_existing_file)]
+
+
+def validate_not_empty(text: any) -> any:
+    if not text:
+        raise ValueError("Expected not empty")
+    return text
+
+
+StrNotEmpty = Annotated[str, AfterValidator(validate_not_empty)]
+
+
 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):
@@ -16,3 +40,36 @@ def validate_prompt(prompt: str, prompt_required_keys: set[str]) -> str:
     if extern_keys := exist_keys.difference(prompt_required_keys):
         raise ValueError(f"You have more keys for prompt: {extern_keys}")
     return prompt
+
+
+def validate_keys(*keys):
+    assert keys and all(isinstance(key, str) for key in keys)
+
+    def validate(prompt: str) -> str:
+        prompt_required_keys: set[str] = set(keys)
+        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
+
+    return validate
+
+
+K = TypeVar("K", bound=str)
+
+
+class Prompt:
+    # use inside pydantic models: prompt: Prompt['base_context']
+    def __class_getitem__(cls, keys: K | tuple[K, ...]) -> type[str]:
+        if isinstance(keys, tuple):
+            keys_tuple = keys
+        else:
+            keys_tuple = (keys,)
+
+        # Validate that all keys are strings
+        if not all(isinstance(key, str) for key in keys_tuple):
+            raise TypeError("All keys must be strings")
+
+        return Annotated[str, AfterValidator(validate_keys(*keys_tuple))]

{agi_med_common-4.0.6.dist-info → agi_med_common-4.2.0.dist-info}/METADATA

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

{agi_med_common-4.0.6.dist-info → agi_med_common-4.2.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
-agi_med_common/__init__.py,sha256=ieERcHJV_y_jHN7e9DFg6VSz2TAszb423gDMusC7q24,575
+agi_med_common/__init__.py,sha256=IvElxZ-gSgSUhXTYjdzg0e5p8C74xFqsslUXktPQf5w,592
 agi_med_common/file_storage.py,sha256=uyZE3ePKrbVLNVcvxw9lbgumy1UPwDGIwQMs1Bd8HFI,1502
 agi_med_common/parallel_map.py,sha256=Qx6xe7DqlEUDpSucp5Hm8r9y9Iquwh9JvX7lOqHhnOw,921
 agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
-agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
+agi_med_common/validators.py,sha256=sWXWsJn9FBlbyUHCNBNSMiEnD9ZenXhzMBNTXjC7yRE,2413
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
 agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
 agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
@@ -14,7 +14,7 @@ agi_med_common/models/base_config_models/gigachat_config.py,sha256=WNSCTO8Fjpxc1
 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=6YD3u5Xa-zKC9XSaum4b2zMB5-nWmTkGn05DMdWFxF4,1060
+agi_med_common/models/chat_item/outer_context_item.py,sha256=_snB7FI4sAp17U29r_SYI6UMKm9yveO031HjCtQQ9Bo,1263
 agi_med_common/models/chat_item/replica_item.py,sha256=D1goRTINLa1rUcQcwHOWUSiv_OR1UW0GVn5_TqK3dyk,1672
 agi_med_common/models/enums/__init__.py,sha256=NFgy8J-bET2GQqUyTaKYLzaLaM-jDR8-7TBTadWKm5o,341
 agi_med_common/models/enums/diagnostics_xml_tag_enum.py,sha256=aiUJpe5joh5EOcJ5sk1Xs003XkRry2CBL6H-bIEiapA,278
@@ -23,7 +23,7 @@ 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/track_id_enum.py,sha256=N3KUd97a4xDet_NKZ5URm0qWAAv3ts1l_foOjdlGY4c,532
-agi_med_common-4.0.6.dist-info/METADATA,sha256=RwjXqgPp2Pw0591Sjy0hUs73zkYEkgwCP69KCqV11wE,546
-agi_med_common-4.0.6.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-agi_med_common-4.0.6.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-4.0.6.dist-info/RECORD,,
+agi_med_common-4.2.0.dist-info/METADATA,sha256=oSn4FoibCezpwEn5Qx1jUmfVAIOdnUraA3gxA_iVy0s,546
+agi_med_common-4.2.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+agi_med_common-4.2.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-4.2.0.dist-info/RECORD,,

{agi_med_common-4.0.6.dist-info → agi_med_common-4.2.0.dist-info}/WHEEL

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