agi-med-common 5.0.16__tar.gz → 5.0.18__tar.gz

{agi_med_common-5.0.16 → agi_med_common-5.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 5.0.16
+Version: 5.0.18
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-5.0.16 → agi_med_common-5.0.18}/src/agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.16"
+__version__ = "5.0.18"
 
 from .models import (
     MTRSLabelEnum,
@@ -26,3 +26,4 @@ from .api.content_interpreter_api import ContentInterpreterAPI, Interpretation
 from .api.content_interpreter_remote_api import ContentInterpreterRemoteAPI
 from .api.text_generator_api import TextGeneratorAPI
 from .api.classifier_api import ClassifierAPI
+from .api.critic_api import CriticAPI

agi_med_common-5.0.18/src/agi_med_common/api/critic_api.py

@@ -0,0 +1,6 @@
+from agi_med_common.models import ChatItem
+
+
+class CriticAPI:
+    def evaluate(self, text: str, chat: ChatItem | None = None, request_id: str = "") -> float:
+        raise NotImplementedError

agi_med_common-5.0.18/src/agi_med_common/models/chat.py

@@ -0,0 +1,267 @@
+from datetime import datetime
+from typing import Any, List, Dict, Literal
+
+from agi_med_common.models.chat_item import ChatItem, ReplicaItem, OuterContextItem
+from agi_med_common.models.widget import Widget
+from agi_med_common.type_union import TypeUnion
+from agi_med_common.utils import first_nonnull
+from pydantic import Field
+
+from ._base import _Base
+
+
+_DT_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
+_EXAMPLE_DT: str = datetime(year=1970, month=1, day=1).strftime(_DT_FORMAT)
+StrDict = Dict[str, Any]
+ContentBase = str | Widget | StrDict
+Content = ContentBase | List[ContentBase]
+
+
+def now_pretty() -> str:
+    return datetime.now().strftime(_DT_FORMAT)
+
+
+class Context(_Base):
+    client_id: str = Field("", examples=["543216789"])
+    user_id: str = Field("", examples=["123456789"])
+    session_id: str = Field("", examples=["987654321"])
+    track_id: str = Field(examples=["Hello"])
+    extra: StrDict | None = Field(None, examples=[None])
+
+    def create_id(self, short: bool = False) -> str:
+        uid, sid, cid = self.user_id, self.session_id, self.client_id
+        if short:
+            return f"{cid}_{uid}_{sid}"
+        return f"client_{cid}_user_{uid}_session_{sid}"
+
+
+def _get_str_field(obj: dict, field) -> str | None:
+    if not isinstance(obj, dict):
+        return None
+    text = obj.get(field)
+    if text is not None and isinstance(text, str):
+        return text
+    return None
+
+
+def _get_text(obj: Content) -> str:
+    if isinstance(obj, str):
+        return obj
+    if isinstance(obj, list):
+        return "".join(map(_get_text, obj))
+    if isinstance(obj, dict) and obj.get("type") == "text":
+        return _get_str_field(obj, "text") or ""
+    return ""
+
+
+def _get_resource_id(obj: Content) -> str | None:
+    if isinstance(obj, list):
+        return first_nonnull(map(_get_resource_id, obj))
+    if isinstance(obj, dict) and obj.get("type") == "resource_id":
+        return _get_str_field(obj, "resource_id")
+    return None
+
+
+def _get_command(obj: Content) -> dict | None:
+    if isinstance(obj, list):
+        return first_nonnull(map(_get_command, obj))
+    if isinstance(obj, dict) and obj.get("type") == "command":
+        return _get_str_field(obj, "command")
+    return None
+
+
+def _get_widget(obj: Content) -> Widget | None:
+    if isinstance(obj, list):
+        return first_nonnull(map(_get_widget, obj))
+    if isinstance(obj, Widget):
+        return obj
+    return None
+
+
+# todo fix: generalize functions above
+
+
+class BaseMessage(_Base):
+    type: str
+    content: Content = Field("", examples=["Привет"])
+    date_time: str = Field(default_factory=now_pretty, examples=[_EXAMPLE_DT])
+    extra: StrDict | None = Field(None, examples=[None])
+
+    @property
+    def text(self) -> str:
+        return _get_text(self.content)
+
+    @property
+    def resource_id(self) -> str | None:
+        return _get_resource_id(self.content)
+
+    @property
+    def command(self) -> dict | None:
+        return _get_command(self.content)
+
+    @property
+    def widget(self) -> Widget | None:
+        return _get_widget(self.content)
+
+    @staticmethod
+    def DATETIME_FORMAT() -> str:
+        return _DT_FORMAT
+
+    def with_now_datetime(self):
+        return self.model_copy(update=dict(date_time=now_pretty()))
+
+
+class HumanMessage(BaseMessage):
+    type: Literal["human"] = "human"
+
+
+class AIMessage(BaseMessage):
+    type: Literal["ai"] = "ai"
+    state: str = Field("", examples=["COLLECTION"])
+
+
+class MiscMessage(BaseMessage):
+    type: Literal["misc"] = "misc"
+
+
+ChatMessage = TypeUnion[HumanMessage, AIMessage, MiscMessage]
+
+
+class Chat(_Base):
+    context: Context
+    messages: List[ChatMessage] = []
+
+    def create_id(self, short: bool = False) -> str:
+        return self.context.create_id(short)
+
+    @classmethod
+    def parse(self, chat_obj: str | dict) -> "Chat":
+        return _parse_chat_compat(chat_obj)
+
+
+def convert_replica_item_to_message(replica: ReplicaItem) -> ChatMessage:
+    resource_id = (replica.resource_id or None) and {"type": "resource_id", "resource_id": replica.resource_id}
+    body = replica.body
+    command = replica.command
+    widget = replica.widget
+
+    content = list(filter(None, [body, resource_id, command, widget]))
+    if len(content) == 0:
+        content = ""
+    elif len(content) == 1:
+        content = content[0]
+
+    kwargs = dict(
+        content=content,
+        date_time=replica.date_time,
+        extra=replica.extra,
+    )
+    if not replica.role:
+        return HumanMessage(**kwargs)
+    return AIMessage(
+        **kwargs,
+        state=replica.state,
+        extra=dict(
+            action=replica.action,
+            moderation=replica.moderation,
+        ),
+    )
+
+
+def convert_outer_context_to_context(octx: OuterContextItem) -> Context:
+    # legacy: eliminate
+    context = Context(
+        client_id=octx.client_id,
+        user_id=octx.user_id,
+        session_id=octx.session_id,
+        track_id=octx.track_id,
+        extra=dict(
+            sex=octx.sex,
+            age=octx.age,
+            parent_session_id=octx.parent_session_id,
+            entrypoint_key=octx.entrypoint_key,
+            language_code=octx.language_code,
+        ),
+    )
+    return context
+
+
+def convert_chat_item_to_chat(chat_item: ChatItem) -> Chat:
+    # legacy: eliminate
+    context = convert_outer_context_to_context(chat_item.outer_context)
+    messages = [convert_replica_item_to_message(replica) for replica in chat_item.inner_context.replicas]
+    res = Chat(context=context, messages=messages)
+    return res
+
+
+def convert_context_to_outer_context(context: Context) -> OuterContextItem:
+    # legacy: eliminate
+    extra = context.extra 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"),
+        parent_session_id=extra.get("parent_session_id"),
+        entrypoint_key=extra.get("entrypoint_key"),
+        language_code=extra.get("language_code"),
+    )
+
+
+def convert_message_to_replica_item(message: ChatMessage) -> ReplicaItem | None:
+    # legacy: eliminate
+    m_type = message.type
+    if m_type in {"ai", "human"}:
+        role = m_type == "ai"
+    else:
+        return None
+
+    extra = message.extra or {}
+    action = extra.pop("action")
+    moderation = extra.pop("moderation")
+
+    kwargs = dict(
+        role=role,
+        body=message.body,
+        resource_id=message.resource_id,
+        command=message.command,
+        widget=message.widget,
+        date_time=message.date_time,
+        extra=extra or None,
+        state=getattr(message, "state", ""),
+        action=action,
+        moderation=moderation,
+    )
+    return ReplicaItem(**kwargs)
+
+
+def convert_chat_to_chat_item(chat: Chat) -> ChatItem:
+    # legacy: eliminate
+    return ChatItem(
+        outer_context=convert_context_to_outer_context(chat.context),
+        inner_context=dict(replicas=list(map(convert_message_to_replica_item, chat.messages))),
+    )
+
+
+def parse_chat_item_as_chat(chat_obj: str | dict) -> Chat:
+    # legacy: eliminate
+    chat_item = ChatItem.parse(chat_obj)
+    res = convert_chat_item_to_chat(chat_item)
+    return res
+
+
+def _parse_chat(chat_obj: str | dict) -> Chat:
+    if isinstance(chat_obj, dict):
+        return Chat.model_validate(chat_obj)
+
+    return Chat.model_validate_json(chat_obj)
+
+
+def _parse_chat_compat(chat_obj: str | dict) -> Chat:
+    # legacy: eliminate
+    try:
+        return _parse_chat(chat_obj)
+    except Exception:
+        return parse_chat_item_as_chat(chat_obj)

{agi_med_common-5.0.16 → agi_med_common-5.0.18}/src/agi_med_common/models/chat_item.py

@@ -1,11 +1,8 @@
-import json
-import warnings
 from datetime import datetime
-from typing import Any, List
+from typing import Annotated, Any, List
 
 from agi_med_common.models.widget import Widget
-from loguru import logger
-from pydantic import Field, ConfigDict, ValidationError
+from pydantic import Field, ConfigDict, BeforeValidator
 
 from ._base import _Base
 
@@ -42,6 +39,24 @@ class OuterContextItem(_Base):
     def to_dict(self) -> dict[str, Any]:
         return self.model_dump(by_alias=True)
 
+LABELS = {
+    0: "OK",
+    1: "NON_MED",
+    2: "CHILD",
+    3: "ABSURD",
+    4: "GREETING",
+    5: "RECEIPT",
+}
+
+
+def fix_deprecated_moderation(moderation):
+    if isinstance(moderation, int):
+        return LABELS.get(moderation, "OK")
+    elif isinstance(moderation, str):
+        return moderation
+    else:
+        raise ValueError(f"Unsupported moderation: {moderation} :: {type(moderation)}")
+
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
@@ -55,7 +70,9 @@ class ReplicaItem(_Base):
     state: str = Field("", alias="State", description="chat manager fsm state", examples=["COLLECTION"])
     action: str = Field("", alias="Action", description="chat manager fsm action", examples=["DIAGNOSIS"])
     # todo fix: support loading from `moderation: int`
-    moderation: str = Field("OK", alias="Moderation", description="moderation outcome", examples=["OK"])
+    moderation: Annotated[str, BeforeValidator(str)] = Field(
+        "OK", alias="Moderation", description="moderation outcome", examples=["OK"]
+    )
     extra: dict | None = Field(None, alias="Extra", examples=[None])
 
     def to_dict(self) -> dict[str, Any]:
@@ -99,40 +116,11 @@ class ChatItem(_Base):
 
     @classmethod
     def parse(cls, chat_obj: str | dict) -> "ChatItem":
-        return _parse_chat(chat_obj)
-
+        return _parse_chat_item(chat_obj)
 
-def _is_moderation_int_error(err):
-    # fmt: off
-    if err['type'] != 'string_type': return False
-    if err['loc'][0] != 'InnerContext': return False
-    if err['loc'][1] != 'Replicas': return False
-    if err['loc'][3] != 'Moderation': return False
-    if err['msg'] != 'Input should be a valid string': return False
-    # fmt: on
-    return True
 
-
-def _is_moderation_int_errors(ex):
-    errs = ex.errors()
-    return all(map(_is_moderation_int_error, errs))
-
-
-def _parse_chat(chat_obj: str | dict) -> ChatItem:
+def _parse_chat_item(chat_obj: str | dict) -> ChatItem:
     if isinstance(chat_obj, dict):
-        return ChatItem(**chat_obj)
-    try:
-        return ChatItem.model_validate_json(chat_obj)
-    except ValidationError as ex:
-        if _is_moderation_int_errors(ex):
-            msg = "Failed to parse ChatItem, fallback to old version with `Moderation:int`"
-        else:
-            msg = f"Failed to parse: {ex}"
-        warnings.warn(msg)
-
-    # old version
-    chat_dict = json.loads(chat_obj)
-    for rep in chat_dict["InnerContext"]["Replicas"]:
-        rep["Moderation"] = "OK"
-
-    return ChatItem.model_validate(chat_dict)
+        return ChatItem.model_validate(chat_obj)
+
+    return ChatItem.model_validate_json(chat_obj)

{agi_med_common-5.0.16 → agi_med_common-5.0.18}/src/agi_med_common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 5.0.16
+Version: 5.0.18
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-5.0.16 → agi_med_common-5.0.18}/src/agi_med_common.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ src/agi_med_common/api/chat_manager_api.py
 src/agi_med_common/api/classifier_api.py
 src/agi_med_common/api/content_interpreter_api.py
 src/agi_med_common/api/content_interpreter_remote_api.py
+src/agi_med_common/api/critic_api.py
 src/agi_med_common/api/text_generator_api.py
 src/agi_med_common/models/__init__.py
 src/agi_med_common/models/_base.py

agi_med_common-5.0.16/src/agi_med_common/models/chat.py

@@ -1,107 +0,0 @@
-from datetime import datetime
-from typing import Any, List, Dict, Literal
-
-from agi_med_common.models.widget import Widget
-from agi_med_common.type_union import TypeUnion
-from agi_med_common.utils import first_nonnull
-from pydantic import Field
-
-from ._base import _Base
-
-
-_DT_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
-_EXAMPLE_DT: str = datetime(year=1970, month=1, day=1).strftime(_DT_FORMAT)
-StrDict = Dict[str, Any]
-ContentBase = str | Widget | StrDict
-Content = ContentBase | List[ContentBase]
-
-
-def now_pretty() -> str:
-    return datetime.now().strftime(_DT_FORMAT)
-
-
-class Context(_Base):
-    client_id: str = Field("", examples=["543216789"])
-    user_id: str = Field("", examples=["123456789"])
-    session_id: str = Field("", examples=["987654321"])
-    track_id: str = Field(examples=["Hello"])
-    extra: StrDict | None = Field(None, examples=[None])
-
-    def create_id(self, short: bool = False) -> str:
-        uid, sid, cid = self.user_id, self.session_id, self.client_id
-        if short:
-            return f"{cid}_{uid}_{sid}"
-        return f"client_{cid}_user_{uid}_session_{sid}"
-
-
-def _get_str_field(obj: dict, field) -> str | None:
-    if not isinstance(obj, dict):
-        return None
-    text = obj.get(field)
-    if text is not None and isinstance(text, str):
-        return text
-    return None
-
-
-def _get_text(obj: Content) -> str:
-    if isinstance(obj, str):
-        return obj
-    if isinstance(obj, list):
-        return "".join(map(_get_text, obj))
-    if isinstance(obj, dict) and obj.get("type") == "text":
-        return _get_str_field(obj, "text") or ""
-    return ""
-
-
-def _get_resource_id(obj: Content) -> str | None:
-    if isinstance(obj, list):
-        return first_nonnull(_get_str_field(el, "resource_id") for el in obj)
-    if isinstance(obj, dict) and obj.get("type") == "resource_id":
-        return _get_str_field(obj, "resource_id")
-    return None
-
-
-class BaseMessage(_Base):
-    type: str
-    content: Content = Field("", examples=["Привет"])
-    date_time: str = Field(default_factory=now_pretty, examples=[_EXAMPLE_DT])
-    extra: StrDict | None = Field(None, examples=[None])
-
-    @property
-    def text(self) -> str:
-        return _get_text(self.content)
-
-    @property
-    def resource_id(self) -> str | None:
-        return _get_resource_id(self.content)
-
-    @staticmethod
-    def DATETIME_FORMAT() -> str:
-        return _DT_FORMAT
-
-    def with_now_datetime(self):
-        return self.model_copy(update=dict(date_time=now_pretty()))
-
-
-class HumanMessage(BaseMessage):
-    type: Literal["human"] = "human"
-
-
-class AIMessage(BaseMessage):
-    type: Literal["ai"] = "ai"
-    state: str = Field("", examples=["COLLECTION"])
-
-
-class MiscMessage(BaseMessage):
-    type: Literal["misc"] = "misc"
-
-
-ChatMessage = TypeUnion[HumanMessage, AIMessage, MiscMessage]
-
-
-class Chat(_Base):
-    context: Context
-    messages: List[ChatMessage] = []
-
-    def create_id(self, short: bool = False) -> str:
-        return self.context.create_id(short)