agi-med-common 1.0.0__py3-none-any.whl

agi_med_common/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

agi_med_common/models/__init__.py

@@ -0,0 +1,2 @@
+from ._base import _Base
+from .chat_item import ChatItem, OuterContextItem, InnerContextItem, ReplicaItem

agi_med_common/models/_base.py

@@ -0,0 +1,5 @@
+from pydantic import ConfigDict, BaseModel
+
+
+class _Base(BaseModel):
+    model_config = ConfigDict(populate_by_name=True, str_strip_whitespace=True)

agi_med_common/models/chat_item/__init__.py

@@ -0,0 +1,4 @@
+from .replica_item import ReplicaItem
+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

@@ -0,0 +1,9 @@
+from pydantic import Field
+
+from . import OuterContextItem, InnerContextItem
+from .. import _Base
+
+
+class ChatItem(_Base):
+    outer_context: OuterContextItem = Field(alias="OuterContext")
+    inner_context: InnerContextItem = Field(alias="InnerContext")

agi_med_common/models/chat_item/inner_context_item.py

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

agi_med_common/models/chat_item/outer_context_item.py

@@ -0,0 +1,18 @@
+from pydantic import Field
+
+from .. import _Base
+from ..enums import TrackIdEnum
+
+
+class OuterContextItem(_Base):
+    sex: bool = Field(alias="Sex", description="True = male, False = female")
+    age: int = Field(alias="Age")
+    user_id: str = Field(alias="UserId")
+    session_id: str = Field(alias="SessionId")
+    client_id: str = Field(alias="ClientId")
+    track_id: TrackIdEnum = Field(TrackIdEnum.DIAGNOSTIC, alias="TrackId")
+
+    def create_id(self, short: 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}"

agi_med_common/models/chat_item/replica_item.py

@@ -0,0 +1,9 @@
+from pydantic import Field
+
+from .. import _Base
+
+
+class ReplicaItem(_Base):
+    body: str = Field(alias="Body")
+    role: bool = Field(alias="Role", description="True = ai, False = client")
+    date_time: str = Field(alias="DateTime")

agi_med_common/models/enums/__init__.py

@@ -0,0 +1 @@
+from .track_id_enum import TrackIdEnum

agi_med_common/models/enums/track_id_enum.py

@@ -0,0 +1,13 @@
+from enum import StrEnum, auto
+
+
+class TrackIdEnum(StrEnum):
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[str]) -> str:
+        return "".join([word.title() for word in name.lower().split("_")])
+
+    DIAGNOSTIC = auto()
+    SECOND_OPINION = auto()
+    MEDICAL_TEST_DECRYPTION = auto()
+    CONSULTATION = auto()
+    SUMMARIZATION = auto()

agi_med_common-1.0.0.dist-info/METADATA

@@ -0,0 +1,25 @@
+Metadata-Version: 2.1
+Name: agi_med_common
+Version: 1.0.0
+Summary: Сommon for agi-med team
+Author: AGI-MED-TEAM
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic<=3.0.0,>=2.9.2
+
+# agi-med-common
+
+Common компонент
+
+## Ответственный разработчик
+
+@zhelvakov
+
+## Общая информация
+
+- Общий ChatItem для сервисов
+- Общей TrackId
+- 
+## Тесты
+
+- `sudo docker compose up --build`

agi_med_common-1.0.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+agi_med_common/__init__.py,sha256=J-j-u0itpEFT6irdmWmixQqYMadNl1X91TxUmoiLHMI,22
+agi_med_common/models/__init__.py,sha256=v4Iwdw7EdGBoBf50LyFsxn99sDABf6z9hOko7Lmi9NM,106
+agi_med_common/models/_base.py,sha256=qNdH8x3x3mYbo5XgWtR9VpEarxsEvXvzynadUlDvHmU,149
+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=YMkR_ZAXPK6AKQUnRy4mFqVNGFu4jtsQ7pOlf3xXrLs,666
+agi_med_common/models/chat_item/replica_item.py,sha256=MCk0B8fHMQiwFYqdWCNnJgKhbJvQ2ujQP7nu-ztN87I,236
+agi_med_common/models/enums/__init__.py,sha256=ooZLUfkRKeJDeTROAAqdRsMoqKBKmgnuGZXWQf9r83Y,39
+agi_med_common/models/enums/track_id_enum.py,sha256=flKTRX4Th-bbMdqYiYj4Rj4Mku-p586bdKZS1YqfK6E,394
+agi_med_common-1.0.0.dist-info/METADATA,sha256=YPHa887inQzItVEyOOwiYTObCwUwYlcs-dvsbPWMSVk,484
+agi_med_common-1.0.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+agi_med_common-1.0.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-1.0.0.dist-info/RECORD,,

agi_med_common-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.6.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

agi_med_common-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+agi_med_common