mmar-mapi 1.0.23__tar.gz → 1.0.24__tar.gz

{mmar_mapi-1.0.23 → mmar_mapi-1.0.24}/PKG-INFO

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

{mmar_mapi-1.0.23 → mmar_mapi-1.0.24}/pyproject.toml

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

{mmar_mapi-1.0.23 → mmar_mapi-1.0.24}/src/mmar_mapi/api.py

@@ -159,6 +159,7 @@ class ExtractedImageMetadata(BaseModel):
 
 
 class ExtractedPicture(ExtractedImage, ExtractedImageMetadata):
+    " Image of part of page "
     pass
 
 
@@ -167,6 +168,7 @@ class ExtractedTable(ExtractedImage, ExtractedImageMetadata):
 
 
 class ExtractedPageImage(ExtractedImage):
+    " Image of all page "
     pass
 
 

{mmar_mapi-1.0.23 → mmar_mapi-1.0.24}/src/mmar_mapi/file_storage.py

@@ -78,7 +78,7 @@ class FileStorage:
 
     def upload_dir(self, resource_ids: list[ResourceId]) -> ResourceId:
         content = "\n".join(resource_ids)
-        res = self.upload(content, "dir")
+        res = self.upload(content=content, fname=".dir")
         return res
 
     def download(self, resource_id: ResourceId) -> bytes:

{mmar_mapi-1.0.23 → mmar_mapi-1.0.24}/src/mmar_mapi/models/chat.py

@@ -244,8 +244,8 @@ class Chat(Base):
     def parse(chat_obj: str | dict | ChatItem) -> "Chat":
         return _parse_chat_compat(chat_obj)
 
-    def to_chat_item(self) -> ChatItem:
-        return convert_chat_to_chat_item(self)
+    def to_chat_item(self, failsafe: bool=False) -> ChatItem:
+        return convert_chat_to_chat_item(self, failsafe)
 
     def add_message(self, message: ChatMessage):
         self.messages.append(message)
@@ -392,19 +392,24 @@ def convert_chat_item_to_chat(chat_item: ChatItem) -> Chat:
     return res
 
 
-def convert_context_to_outer_context(context: Context) -> OuterContextItem:
+def convert_context_to_outer_context(context: Context, failsafe: bool=False) -> OuterContextItem:
     # legacy: eliminate after migration
     extra = context.extra or {}
+    if failsafe:
+        extra['sex'] = extra.get('sex') or True
+        extra['age'] = extra.get('age') or 42
+        extra['language_code'] = extra.get('language_code') or ''
+        extra['entrypoint_key'] = extra.get('entrypoint_key') or ''
     return OuterContextItem(
         client_id=context.client_id,
         user_id=context.user_id,
         session_id=context.session_id,
         track_id=context.track_id,
-        sex=extra.get("sex"),
-        age=extra.get("age"),
+        sex=extra["sex"],
+        age=extra["age"],
+        entrypoint_key=extra["entrypoint_key"],
+        language_code=extra["language_code"],
         parent_session_id=extra.get("parent_session_id"),
-        entrypoint_key=extra.get("entrypoint_key"),
-        language_code=extra.get("language_code"),
     )
 
 
@@ -436,10 +441,10 @@ def convert_message_to_replica_item(message: ChatMessage) -> ReplicaItem | None:
     return ReplicaItem(**kwargs)
 
 
-def convert_chat_to_chat_item(chat: Chat) -> ChatItem:
+def convert_chat_to_chat_item(chat: Chat, failsafe: bool=False) -> ChatItem:
     # legacy: eliminate after migration
     res = ChatItem(
-        outer_context=convert_context_to_outer_context(chat.context),
+        outer_context=convert_context_to_outer_context(chat.context, failsafe=failsafe),
         inner_context=dict(replicas=list(map(convert_message_to_replica_item, chat.messages))),
     )
     return res