mmar-mapi 1.0.9__tar.gz → 1.0.11__tar.gz

{mmar_mapi-1.0.9 → mmar_mapi-1.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mmar-mapi
-Version: 1.0.9
+Version: 1.0.11
 Summary: Common pure/IO utilities for multi-modal architectures team
 Keywords: 
 Author: Eugene Tagin
@@ -20,6 +20,7 @@ Classifier: Topic :: Software Development
 Classifier: Topic :: Utilities
 Classifier: Typing :: Typed
 Requires-Dist: pydantic~=2.11.7
+Requires-Dist: loguru~=0.7.3
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 

{mmar_mapi-1.0.9 → mmar_mapi-1.0.11}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mmar-mapi"
 # dynamic version is not supported yet on uv_build
-version = "1.0.9"
+version = "1.0.11"
 description = "Common pure/IO utilities for multi-modal architectures team"
 authors = [{name = "Eugene Tagin", email = "tagin@airi.net"}]
 license = "MIT"
@@ -25,6 +25,7 @@ classifiers = [
 ]
 dependencies = [
     "pydantic~=2.11.7",
+    "loguru~=0.7.3",
 ]
 
 [build-system]
@@ -51,7 +52,6 @@ ci = [
     "mypy>=1.10",
     "types-markdown>=3.6",
     "types-pyyaml>=6.0",
-    "more-itertools>=10.8.0",
 ]
 
 [tool.uv]

{mmar_mapi-1.0.9 → mmar_mapi-1.0.11}/src/mmar_mapi/__init__.py

@@ -15,8 +15,9 @@ from .models.chat_item import ChatItem, OuterContextItem, InnerContextItem, Repl
 from .models.enums import MTRSLabelEnum, DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnum
 from .models.tracks import TrackInfo, DomainInfo
 from .models.widget import Widget
-from .utils import make_session_id
+from .utils import make_session_id, chunked
 from .xml_parser import XMLParser
+from .utils_import import load_main_objects
 
 __all__ = [
     "AIMessage",
@@ -42,6 +43,8 @@ __all__ = [
     "TrackInfo",
     "Widget",
     "XMLParser",
+    "chunked",
+    "load_main_objects",
     "make_content",
     "make_session_id",
 ]

{mmar_mapi-1.0.9 → mmar_mapi-1.0.11}/src/mmar_mapi/models/widget.py

@@ -1,6 +1,6 @@
 from typing import Self, Literal
 
-from more_itertools import chunked
+from mmar_mapi.utils import chunked
 from pydantic import BaseModel, model_validator
 
 

mmar_mapi-1.0.11/src/mmar_mapi/utils.py

@@ -0,0 +1,16 @@
+from itertools import islice
+from collections.abc import Iterable
+from datetime import datetime
+
+
+def make_session_id() -> str:
+    return f"{datetime.now():%Y-%m-%d--%H-%M-%S}"
+
+
+def chunked(items: Iterable, n) -> list:
+    "behavior like in more_itertools.chunked"
+    iterator = iter(items)
+    res = []
+    while chunk := list(islice(iterator, n)):
+        res.append(chunk)
+    return res

mmar_mapi-1.0.11/src/mmar_mapi/utils_import.py

@@ -0,0 +1,42 @@
+from pkgutil import iter_modules
+from importlib import import_module
+from types import ModuleType
+
+from loguru import logger
+
+def convert_snake_to_pascal(name: str) -> str:
+    """ snake_to_pascal -> SnakeToPascal """
+    return "".join(map(str.capitalize, name.split("_")))
+
+
+def get_main_object_name(module: type[ModuleType]) -> str:
+    module_name = module.__name__.rsplit(".", 1)[-1]
+    main_object_name = convert_snake_to_pascal(module_name)
+    return main_object_name
+
+
+def get_main_object(module: type[ModuleType], obj_type):
+    """tries to find SomeObject in object src.some_object"""
+    main_object_name = get_main_object_name(module)
+    try:
+        res = getattr(module, main_object_name)
+        if not isinstance(res, obj_type) and not issubclass(res, obj_type):
+            logger.error(f"Failed to load {module}.{main_object_name}: expected {obj_type} but found {type(res)}")
+            return None
+        return res
+    except AttributeError as ex:
+        logger.error(f"Failed to load {module}.{main_object_name}: {ex}")
+        return None
+
+
+def load_modules(package_name: str) -> list[type[ModuleType]]:
+    package = import_module(package_name)
+    res = [import_module(module_name) for _, module_name, _ in iter_modules(package.__path__, package_name + ".")]
+    return res
+
+
+def load_main_objects(package_name: str, obj_type: type) -> dict[str, object]:
+    modules = load_modules(package_name)
+    main_objects = [get_main_object(m, obj_type) for m in modules]
+    res = {obj.__name__: obj for obj in main_objects}
+    return res

mmar_mapi-1.0.9/src/mmar_mapi/utils.py

@@ -1,5 +0,0 @@
-from datetime import datetime
-
-
-def make_session_id() -> str:
-    return f"{datetime.now():%Y-%m-%d--%H-%M-%S}"