mmar-mapi 1.0.1__tar.gz → 1.0.3__tar.gz

{mmar_mapi-1.0.1 → mmar_mapi-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mmar-mapi
-Version: 1.0.1
+Version: 1.0.3
 Summary: Common pure/IO utilities for multi-modal architectures team
 Keywords: 
 Author: Eugene Tagin

{mmar_mapi-1.0.1 → mmar_mapi-1.0.3}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mmar-mapi"
 # dynamic version is not supported yet on uv_build
-version = "1.0.1"
+version = "1.0.3"
 description = "Common pure/IO utilities for multi-modal architectures team"
 authors = [{name = "Eugene Tagin", email = "tagin@airi.net"}]
 license = "MIT"

{mmar_mapi-1.0.1 → mmar_mapi-1.0.3}/src/mmar_mapi/__init__.py

@@ -3,9 +3,10 @@ __version__ = "5.2.4"
 from .file_storage import FileStorage, ResourceId
 from .models.base import Base
 from .models.base_config_models import GigaChatConfig
-from .models.chat import Chat, Context, ChatMessage, AIMessage, HumanMessage, MiscMessage, make_content
+from .models.chat import Chat, Context, ChatMessage, AIMessage, HumanMessage, MiscMessage, make_content, Content
 from .models.chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
 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 .xml_parser import XMLParser

mmar_mapi-1.0.3/src/mmar_mapi/xml_parser.py

@@ -0,0 +1,18 @@
+import re
+from enum import StrEnum
+
+
+class XMLParser:
+    @staticmethod
+    def make_xml(body: str, tag: str | StrEnum) -> str:
+        return f"<{tag}>{body}</{tag}>"
+
+    @staticmethod
+    def remove_xml(text: str) -> str:
+        re_xml = re.compile(r"<(.*?)>")
+        return re.sub(re_xml, "", text).replace("  ", " ").strip()
+
+    @staticmethod
+    def get_tag_list(response: str, tag: str | StrEnum) -> list[str]:
+        tag_content = re.findall(f"<{tag}>(.*?)</{tag}>", response, re.DOTALL)
+        return [content.lower().strip() for content in tag_content]