mmar-mapi 1.0.10__tar.gz → 1.0.12__tar.gz

{mmar_mapi-1.0.10 → mmar_mapi-1.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mmar-mapi
-Version: 1.0.10
+Version: 1.0.12
 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.10 → mmar_mapi-1.0.12}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mmar-mapi"
 # dynamic version is not supported yet on uv_build
-version = "1.0.10"
+version = "1.0.12"
 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]

{mmar_mapi-1.0.10 → mmar_mapi-1.0.12}/src/mmar_mapi/__init__.py

@@ -17,7 +17,7 @@ from .models.tracks import TrackInfo, DomainInfo
 from .models.widget import Widget
 from .utils import make_session_id, chunked
 from .xml_parser import XMLParser
-
+from .utils_import import load_main_objects
 
 __all__ = [
     "AIMessage",
@@ -44,6 +44,7 @@ __all__ = [
     "Widget",
     "XMLParser",
     "chunked",
+    "load_main_objects",
     "make_content",
     "make_session_id",
 ]

{mmar_mapi-1.0.10 → mmar_mapi-1.0.12}/src/mmar_mapi/models/chat.py

@@ -185,6 +185,8 @@ class AIMessage(BaseMessage):
     def action(self) -> str:
         return (self.extra or {}).get("action", "")
 
+    def with_state(self, state: str) -> "AIMessage":
+        return self.model_copy(update=dict(state=state))
 
 class MiscMessage(BaseMessage):
     type: Literal["misc"] = "misc"

mmar_mapi-1.0.12/src/mmar_mapi/utils_import.py

@@ -0,0 +1,46 @@
+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]]:
+    try:
+        package = import_module(package_name)
+    except ModuleNotFoundError:
+        logger.error(f"Not found module: {package_name}")
+        return []
+    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