agi-med-common 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -1 +1 @@
-__version__ = "1.0.0"
+__version__ = "1.1.0"

agi_med_common/logger/__init__.py

@@ -0,0 +1,2 @@
+from .log_level_enum import LogLevelEnum
+from .logger import logger_init, log_llm_error, logger

agi_med_common/logger/log_level_enum.py

@@ -0,0 +1,15 @@
+from enum import StrEnum, auto
+
+
+class LogLevelEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return name.upper()
+
+    TRACE = auto()  # logger.trace()
+    DEBUG = auto()  # logger.debug()
+    INFO = auto()  # logger.info()
+    SUCCESS = auto()  # logger.success()
+    WARNING = auto()  # logger.warning()
+    ERROR = auto()  # logger.error()
+    CRITICAL = auto()  # logger.critical()

agi_med_common/logger/logger.py

@@ -0,0 +1,38 @@
+import sys
+from uuid import UUID
+
+from loguru import logger
+
+from . import LogLevelEnum
+
+
+def logger_init(
+    log_level: LogLevelEnum, service_name: str, include_outer_context: bool = True, include_extra_uuid: bool = True
+) -> None:
+    logger.remove()
+    default_uuid = UUID(int=0)
+    format_ = f"{{time:dd-MM-YYYY HH:mm:ss:ms}} | {service_name} | {{extra[uuid]}}"
+    extra: dict[str, UUID | str] = {"uuid": default_uuid}
+    if include_outer_context:
+        format_ = f"{format_} | {{extra[outer_context]}}"
+        extra["outer_context"] = default_uuid
+    if include_extra_uuid:
+        format_ = f"{format_} | {{extra[extra_uuid]}}"
+        extra["extra_uuid"] = default_uuid
+    format_ = f"{format_} | <level>{{message}}</level>"
+    logger.add(sys.stdout, colorize=True, format=format_, level=log_level)
+    logger.configure(extra=extra)
+
+
+def log_llm_error(
+    text: str | None = None,
+    vector: list[float] | None = None,
+    model: str = "gigachat",
+) -> None:
+    if text is not None and not text:
+        logger.error(f"No response from {model}!!!")
+        return None
+    if vector is not None and all(not item for item in vector):
+        logger.error(f"No response from {model} encoder!!!")
+        return None
+    return None

{agi_med_common-1.0.0.dist-info → agi_med_common-1.1.0.dist-info}/METADATA

@@ -1,11 +1,12 @@
 Metadata-Version: 2.1
 Name: agi_med_common
-Version: 1.0.0
+Version: 1.1.0
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 Requires-Dist: pydantic<=3.0.0,>=2.9.2
+Requires-Dist: loguru==0.7.2
 
 # agi-med-common
 

{agi_med_common-1.0.0.dist-info → agi_med_common-1.1.0.dist-info}/RECORD

@@ -1,4 +1,7 @@
-agi_med_common/__init__.py,sha256=J-j-u0itpEFT6irdmWmixQqYMadNl1X91TxUmoiLHMI,22
+agi_med_common/__init__.py,sha256=LGVQyDsWifdACo7qztwb8RWWHds1E7uQ-ZqD8SAjyw4,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=9yQqXRWcIJ8QV9EgcjHrnvhqyzJRpeYhlEyKMJ-gOes,1252
 agi_med_common/models/__init__.py,sha256=v4Iwdw7EdGBoBf50LyFsxn99sDABf6z9hOko7Lmi9NM,106
 agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
 agi_med_common/models/chat_item/__init__.py,sha256=MoZFnZQV-tWJTHmH1OhRBRkrc5cGWAfdJ6QfHDg6TDk,168
@@ -8,7 +11,7 @@ agi_med_common/models/chat_item/outer_context_item.py,sha256=YMkR_ZAXPK6AKQUnRy4
 agi_med_common/models/chat_item/replica_item.py,sha256=MCk0B8fHMQiwFYqdWCNnJgKhbJvQ2ujQP7nu-ztN87I,236
 agi_med_common/models/enums/__init__.py,sha256=ooZLUfkRKeJDeTROAAqdRsMoqKBKmgnuGZXWQf9r83Y,39
 agi_med_common/models/enums/track_id_enum.py,sha256=flKTRX4Th-bbMdqYiYj4Rj4Mku-p586bdKZS1YqfK6E,394
-agi_med_common-1.0.0.dist-info/METADATA,sha256=YPHa887inQzItVEyOOwiYTObCwUwYlcs-dvsbPWMSVk,484
-agi_med_common-1.0.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-agi_med_common-1.0.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-1.0.0.dist-info/RECORD,,
+agi_med_common-1.1.0.dist-info/METADATA,sha256=bthjmnWC1iTN1ywpTNkPlc6REmHj2aCWMqwnZ05pdA0,513
+agi_med_common-1.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+agi_med_common-1.1.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-1.1.0.dist-info/RECORD,,