mmar-mapi 1.0.10__py3-none-any.whl → 1.0.12__py3-none-any.whl

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/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/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

{mmar_mapi-1.0.10.dist-info → mmar_mapi-1.0.12.dist-info}/METADATA

@@ -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.dist-info → mmar_mapi-1.0.12.dist-info}/RECORD

@@ -1,17 +1,18 @@
-mmar_mapi/__init__.py,sha256=NxCsA3aMKjxIZ2f4pQTJ1LPpg7t2yz7PsiDeb_HrAJE,1109
+mmar_mapi/__init__.py,sha256=Q2WFyNje0mDzP3eK7k-900W4Uty82BlulKv9-FmkcZs,1177
 mmar_mapi/api.py,sha256=C9Sr8dISvf51xfEznPjccI_odaG4coQE3HI_0jVpjMQ,1677
 mmar_mapi/file_storage.py,sha256=GbahBabBdAKjlAnv1MszERUxxZyA9HGMiR9tz2a9dgY,4409
 mmar_mapi/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mmar_mapi/models/base.py,sha256=mKtXV2x51XVj7W-et9tjGcPMDUUUMelW-BywMgFc2p0,411
-mmar_mapi/models/chat.py,sha256=HC-zl32wu5xMyam3iZXZKc4vSfxtc4f-vMb9FgB8m1Q,13655
+mmar_mapi/models/chat.py,sha256=_M1Uy4QNyPkPLvu0vvufa9fs-3oqT59CtjEqWaOgNAM,13765
 mmar_mapi/models/chat_item.py,sha256=ZfCKvTqr7gpuJSAuHVxWRnlTefRwki_IVNA2N_CXGdg,5557
 mmar_mapi/models/enums.py,sha256=J-GNpql9MCnKnWiV9aJRQGI-pAybvV86923RZs99grA,1006
 mmar_mapi/models/tracks.py,sha256=HKDp-BX1p7AlDfSEKfOKCu0TRSK9cD4Dmq1vJt8oRjw,307
 mmar_mapi/models/widget.py,sha256=ue5o4AkN8SG09aA8eQLOOxbwah-mkZir0ZGL6OA8S9Q,1355
 mmar_mapi/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
 mmar_mapi/utils.py,sha256=oVrEAHGcfFBprwdTEzSbFCpdLtkDRVzWtOnYpS439os,394
+mmar_mapi/utils_import.py,sha256=GRCkiH0pIrl-xWo4rdxngcfuPzjfTIXA1rvDVxEZ3lI,1683
 mmar_mapi/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
-mmar_mapi-1.0.10.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
-mmar_mapi-1.0.10.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
-mmar_mapi-1.0.10.dist-info/METADATA,sha256=cL99l0W1GOtgnJOUNDyflzdkbMNHNfGwNpdiLnhFIz0,915
-mmar_mapi-1.0.10.dist-info/RECORD,,
+mmar_mapi-1.0.12.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
+mmar_mapi-1.0.12.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
+mmar_mapi-1.0.12.dist-info/METADATA,sha256=OKaqPCycfHy0GzpWwchWxlvNQymLrV1EVv-nDxna0xk,944
+mmar_mapi-1.0.12.dist-info/RECORD,,