agi-med-common 3.2.1__py3-none-any.whl → 3.3.1__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.2.1"
+__version__ = "3.3.1"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -11,9 +11,8 @@ from .models import (
     InnerContextItem,
     OuterContextItem,
     ReplicaItem,
-    ChatItemWithState,
-    InnerContextWithState,
-    ReplicaWithState,
+    ReplicaItemPair,
 )
 from .utils import make_session_id, read_json, replace_n
 from .validators import is_file_exist, validate_prompt
+from .xml_parser import XMLParser

agi_med_common/logger/logger.py

@@ -7,7 +7,7 @@ from . import LogLevelEnum
 
 def logger_init(log_level: LogLevelEnum) -> None:
     logger.remove()
-    extra = {"request_id": "NOT_A_REQUEST"}
+    extra = {"request_id": "SYSTEM_REQUEST"}
     format_ = "{time:DD-MM-YYYY HH:mm:ss} | <level>{level: <8}</level> | {extra[request_id]}"
     format_ = f"{format_} | <level>{{message}}</level>"
     logger.add(sys.stdout, colorize=True, format=format_, level=log_level)

agi_med_common/models/__init__.py

@@ -1,5 +1,5 @@
 from ._base import _Base
 from .enums import TrackIdEnum, StateEnum, MTRSLabelEnum, ActionEnum, ModerationLabelEnum
-from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem
+from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem, ReplicaItemPair
 from .chat_item_with_state import ChatItemWithState, InnerContextWithState, ReplicaWithState
 from .base_config_models import GigaChatConfig

agi_med_common/models/chat_item/__init__.py

@@ -1,4 +1,4 @@
-from .replica_item import ReplicaItem
+from .replica_item import ReplicaItem, ReplicaItemPair
 from .outer_context_item import OuterContextItem
 from .inner_context_item import InnerContextItem
 from .chat_item import ChatItem

agi_med_common/models/chat_item/chat_item.py

@@ -1,9 +1,34 @@
 from pydantic import Field
+from typing import Any
 
-from . import OuterContextItem, InnerContextItem
+from . import OuterContextItem, InnerContextItem, ReplicaItemPair
+from ..enums import ActionEnum
 from .. import _Base
 
 
 class ChatItem(_Base):
     outer_context: OuterContextItem = Field(alias="OuterContext")
     inner_context: InnerContextItem = Field(alias="InnerContext")
+
+    def create_id(self, short: bool = False, clean: bool = False) -> str:
+        return self.outer_context.create_id(short, clean)
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)
+
+    def update(self, replica_pair: ReplicaItemPair) -> None:
+        self.inner_context.replicas.append(replica_pair.user_replica)
+        self.inner_context.replicas.append(replica_pair.model_replica)
+
+    def count_questions(self) -> int:
+        return len(
+            list(
+                filter(
+                    lambda replica: replica.role and replica.action == ActionEnum.QUESTION,
+                    self.inner_context.replicas,
+                )
+            )
+        )
+
+    def zip_history(self, field: str) -> list[Any]:
+        return [replica.to_dict().get(field, None) for replica in self.inner_context.replicas]

agi_med_common/models/chat_item/inner_context_item.py

@@ -6,3 +6,6 @@ from .. import _Base
 
 class InnerContextItem(_Base):
     replicas: list[ReplicaItem] = Field(alias="Replicas")
+
+    def to_dict(self) -> dict[str, list]:
+        return self.model_dump(by_alias=True)

agi_med_common/models/chat_item/outer_context_item.py

@@ -1,4 +1,5 @@
 from pydantic import Field
+from typing import Any
 
 from .. import _Base
 from ..enums import TrackIdEnum
@@ -10,9 +11,14 @@ class OuterContextItem(_Base):
     user_id: str = Field("", alias="UserId", examples=["123456789"])
     session_id: str = Field("", alias="SessionId", examples=["987654321"])
     client_id: str = Field("", alias="ClientId", examples=["543216789"])
-    track_id: TrackIdEnum = Field(TrackIdEnum.DIAGNOSTIC, alias="TrackId")
+    track_id: TrackIdEnum = Field(TrackIdEnum.CONSULTATION, alias="TrackId")
 
-    def create_id(self, short: bool = False) -> str:
+    def create_id(self, short: bool = False, clean: bool = False) -> str:
         if short:
             return f"{self.user_id}_{self.session_id}_{self.client_id}"
-        return f"user_{self.user_id}_session_{self.session_id}_client_{self.client_id}"
+        if not clean:
+            return f"user_{self.user_id}_session_{self.session_id}_client_{self.client_id}"
+        return f"user_{self.user_id}_session_{self.session_id}_client_{self.client_id}_clean"
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)

agi_med_common/models/chat_item/replica_item.py

@@ -1,31 +1,35 @@
 from datetime import datetime
+from typing import Any
 
-from pydantic import Field, field_validator, field_serializer
+from pydantic import Field
 
 from .. import _Base
+from ..enums import StateEnum, ActionEnum
 
 _TIME_FORMAT: str = "%Y-%m-%d-%H-%M-%S"
-_EXAMPLE_DATETIME: datetime = datetime(1970, 1, 1, 0, 0, 0)
+_EXAMPLE_DATETIME: str = datetime(1970, 1, 1, 0, 0, 0).strftime(_TIME_FORMAT)
 
 
 class ReplicaItem(_Base):
     body: str = Field("", alias="Body", examples=["Привет"])
     role: bool = Field(False, alias="Role", description="True = ai, False = client", examples=[False])
-    date_time: datetime = Field(
+    date_time: str = Field(
         _EXAMPLE_DATETIME,
         alias="DateTime",
-        examples=[_EXAMPLE_DATETIME.strftime(_TIME_FORMAT)],
+        examples=[_EXAMPLE_DATETIME],
         description=f"Format: {_TIME_FORMAT}",
     )
-    previous_score: float | None = Field(None, alias="PreviousScore", exclude=True, deprecated=True)
-
-    @field_validator("date_time", mode="before")
-    @classmethod
-    def convert_date_time(cls, timestamp: str | datetime) -> datetime:
-        if isinstance(timestamp, datetime):
-            return timestamp
-        return datetime.strptime(timestamp, _TIME_FORMAT)
-
-    @field_serializer("date_time")
-    def serialize_date_time(self, timestamp: datetime) -> str:
-        return timestamp.strftime(_TIME_FORMAT)
+    state: StateEnum = Field(
+        StateEnum.EMPTY, alias="State", description="chat manager fsm state", examples=[StateEnum.COLLECTION]
+    )
+    action: ActionEnum = Field(
+        ActionEnum.START, alias="Action", description="chat manager fsm action", examples=[ActionEnum.DIAGNOSIS]
+    )
+
+    def to_dict(self) -> dict[str, Any]:
+        return self.model_dump(by_alias=True)
+
+
+class ReplicaItemPair(_Base):
+    user_replica: ReplicaItem = Field(alias="UserReplica")
+    model_replica: ReplicaItem = Field(alias="ModelReplica")

agi_med_common/xml_parser.py

@@ -0,0 +1,17 @@
+import re
+
+
+class XMLParser:
+    @staticmethod
+    def make_xml(body: str, tag: str) -> 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) -> list[str]:
+        tag_content = re.findall(f"<{tag}>(.*?)</{tag}>", response, re.DOTALL)
+        return [content.lower().strip() for content in tag_content]

{agi_med_common-3.2.1.dist-info → agi_med_common-3.3.1.dist-info}/METADATA

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

agi_med_common-3.3.1.dist-info/RECORD

@@ -0,0 +1,26 @@
+agi_med_common/__init__.py,sha256=x2GH-mmgggvXfhat4sDBZqwYjHqvjzspwSHoLZLsQlk,427
+agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
+agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
+agi_med_common/xml_parser.py,sha256=1MYE9yajX1Enrn0A6DD1cXagZuYdjCvp3vN4D8fIge4,523
+agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
+agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
+agi_med_common/logger/logger.py,sha256=u1qjQXsE1a1-YZ7iWuypzPnJ0e2rB6HVVrtiUGnoERo,843
+agi_med_common/models/__init__.py,sha256=ODfLq-mVK4-94TAjufH00YjFB1HKlMy0x0Jhw4_VfDg,353
+agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
+agi_med_common/models/base_config_models/__init__.py,sha256=KjS_bSCka8BOMsigwcIML-e6eNB2ouMU6gxlhRmzeuY,44
+agi_med_common/models/base_config_models/gigachat_config.py,sha256=WNSCTO8Fjpxc1v2LRUHfKqo9aeMDpXltTHYBFgTD2N0,422
+agi_med_common/models/chat_item/__init__.py,sha256=NG1optUYwjnxfpTXntUfTDOA0eKTgdV43TZOcfHasHg,185
+agi_med_common/models/chat_item/chat_item.py,sha256=Ym-afB9-7NI6QquhMPqd2ZBkbcx-Txr-ZiXwn0PTO-0,1182
+agi_med_common/models/chat_item/inner_context_item.py,sha256=YXmlNLabKN3O84UWJENsjfdtPq2XRH1VqKiyIxS-p64,255
+agi_med_common/models/chat_item/outer_context_item.py,sha256=IzrwhqbLKR8ZF8-Sd2CyNSa0eqb21NQH7LUqkg9se2Y,1046
+agi_med_common/models/chat_item/replica_item.py,sha256=1BIKdqVfpWoWQP6xhtfr2c__Z-LJVDOkGcy5gbGQ_DA,1171
+agi_med_common/models/enums/__init__.py,sha256=7TyOshcmklii1Cvx76Kxw4Ly9Wei2hsX1aHk6kzkVrY,207
+agi_med_common/models/enums/action_enum.py,sha256=ZZtSBlOsQWvJ1UYqmsMsVLrzO1kJdImqJMqI0_KSV9M,287
+agi_med_common/models/enums/moderation_label_enum.py,sha256=Ixz-7_6RneNwGzDzO9pV38V6ZutUQa2pUS2torMDY8s,271
+agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
+agi_med_common/models/enums/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
+agi_med_common/models/enums/track_id_enum.py,sha256=YoP15BTGjaFDxCOtjsqfwuMTW-BhcaQyEdxLfK7RQUA,454
+agi_med_common-3.3.1.dist-info/METADATA,sha256=s9JQvs_Sy6XqY6-3zamsq9_tEQuhJnAl8dxUnqSYVJs,520
+agi_med_common-3.3.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+agi_med_common-3.3.1.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.3.1.dist-info/RECORD,,

agi_med_common/models/chat_item_with_state/__init__.py

@@ -1,3 +0,0 @@
-from .replica_with_state import ReplicaWithState
-from .inner_context_with_state import InnerContextWithState
-from .chat_with_state import ChatItemWithState

agi_med_common/models/chat_item_with_state/chat_with_state.py

@@ -1,8 +0,0 @@
-from ..chat_item import ChatItem
-from pydantic import Field
-
-from . import InnerContextWithState
-
-
-class ChatItemWithState(ChatItem):
-    inner_context: InnerContextWithState = Field(alias="InnerContext")

agi_med_common/models/chat_item_with_state/inner_context_with_state.py

@@ -1,8 +0,0 @@
-from ..chat_item import InnerContextItem
-from pydantic import Field
-
-from . import ReplicaWithState
-
-
-class InnerContextWithState(InnerContextItem):
-    replicas: list[ReplicaWithState] = Field(alias="Replicas")

agi_med_common/models/chat_item_with_state/replica_with_state.py

@@ -1,8 +0,0 @@
-from ..chat_item import ReplicaItem
-from pydantic import Field
-
-from ..enums import StatesEnum
-
-
-class ReplicaWithState(ReplicaItem):
-    state: StatesEnum | None = Field(None, alias="State")

agi_med_common-3.2.1.dist-info/RECORD

@@ -1,29 +0,0 @@
-agi_med_common/__init__.py,sha256=uY4L-vBlFDl44XcDzT9bA281p_JW7129mwccpsY-npQ,444
-agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
-agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
-agi_med_common/logger/__init__.py,sha256=RW_0VZtbeJ4RsLqUXZUQWl5CtV9g840rU7qRlf5u49E,96
-agi_med_common/logger/log_level_enum.py,sha256=lWuSMho9I0v_xf1RuwpERx5o8NJXaavjwxSdh8fxMqE,477
-agi_med_common/logger/logger.py,sha256=_tmBPpZr1UaIycYzuvwrnPrVeu5kcwJBXSbTb7InGjI,842
-agi_med_common/models/__init__.py,sha256=ebG_iVj_cQ8Em6pdsLarl8zSfotEF3KBV1XMpDyS5YY,336
-agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
-agi_med_common/models/base_config_models/__init__.py,sha256=KjS_bSCka8BOMsigwcIML-e6eNB2ouMU6gxlhRmzeuY,44
-agi_med_common/models/base_config_models/gigachat_config.py,sha256=WNSCTO8Fjpxc1v2LRUHfKqo9aeMDpXltTHYBFgTD2N0,422
-agi_med_common/models/chat_item/__init__.py,sha256=MoZFnZQV-tWJTHmH1OhRBRkrc5cGWAfdJ6QfHDg6TDk,168
-agi_med_common/models/chat_item/chat_item.py,sha256=aCRR4OwL-YS6hvVSNkIg7FNpfvEGo8BESCY8BhiAlNk,255
-agi_med_common/models/chat_item/inner_context_item.py,sha256=s17frAcwyi8WcHnBt-5LF8nUJeO-hnu7YMobYQO5a-c,166
-agi_med_common/models/chat_item/outer_context_item.py,sha256=IprCIh8ki3PUueFwdmPw-MYn0-prx_G523Y-IXN62ZY,792
-agi_med_common/models/chat_item/replica_item.py,sha256=_OvnGmbP-wzxi16c6FGz-_Y-QtudV1Pi6I-zCJ7aQfA,1131
-agi_med_common/models/chat_item_with_state/__init__.py,sha256=KrXgsvJnLzS1kSi9_Y1z6_egxIqKL7epY1r-PIyrnno,156
-agi_med_common/models/chat_item_with_state/chat_with_state.py,sha256=IgZh0CRoar56eM8QOblPx8tsIm9NPO5tOj0mVC8yj40,205
-agi_med_common/models/chat_item_with_state/inner_context_with_state.py,sha256=qIwqCHn-JuQemcVQqLDzkpuaPRNCjIL_-h1wm1Jv5WU,212
-agi_med_common/models/chat_item_with_state/replica_with_state.py,sha256=YvUcCo1HTfxYUEwZ5T7UqLW2WkZ8SFvVNQZf-5mK5Dg,192
-agi_med_common/models/enums/__init__.py,sha256=7TyOshcmklii1Cvx76Kxw4Ly9Wei2hsX1aHk6kzkVrY,207
-agi_med_common/models/enums/action_enum.py,sha256=ZZtSBlOsQWvJ1UYqmsMsVLrzO1kJdImqJMqI0_KSV9M,287
-agi_med_common/models/enums/moderation_label_enum.py,sha256=Ixz-7_6RneNwGzDzO9pV38V6ZutUQa2pUS2torMDY8s,271
-agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
-agi_med_common/models/enums/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
-agi_med_common/models/enums/track_id_enum.py,sha256=YoP15BTGjaFDxCOtjsqfwuMTW-BhcaQyEdxLfK7RQUA,454
-agi_med_common-3.2.1.dist-info/METADATA,sha256=NH-OaoLWdsngpKmpcehrQAq3gqxPbFqzd4h67ZJ7Eck,520
-agi_med_common-3.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-agi_med_common-3.2.1.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.2.1.dist-info/RECORD,,