agi-med-common 5.2.0__py3-none-any.whl → 5.2.2__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.2.0"
+__version__ = "5.2.2"
 
 from .file_storage import FileStorage, ResourceId
 from .models.base import Base

agi_med_common/api.py

@@ -1,5 +1,3 @@
-from typing import List, Tuple
-
 from agi_med_common.models.chat import Chat, ChatMessage
 from agi_med_common.models.tracks import DomainInfo, TrackInfo
 from pydantic import BaseModel
@@ -11,13 +9,13 @@ ResourceId = str
 
 
 class ChatManagerAPI:
-    def get_domains(self, *, client_id: str, language_code: str = "ru") -> List[DomainInfo]:
+    def get_domains(self, *, client_id: str, language_code: str = "ru") -> list[DomainInfo]:
         raise NotImplementedError
 
-    def get_tracks(self, *, client_id: str, language_code: str = "ru") -> List[TrackInfo]:
+    def get_tracks(self, *, client_id: str, language_code: str = "ru") -> list[TrackInfo]:
         raise NotImplementedError
 
-    def get_response(self, *, chat: Chat) -> List[ChatMessage]:
+    def get_response(self, *, chat: Chat) -> list[ChatMessage]:
         raise NotImplementedError
 
 
@@ -40,7 +38,7 @@ class ContentInterpreterRemoteAPI:
 
 
 class ClassifierAPI:
-    def get_values(self) -> List[Value]:
+    def get_values(self) -> list[Value]:
         raise NotImplementedError
 
     def evaluate(self, *, chat: Chat) -> Value:
@@ -55,10 +53,10 @@ class CriticAPI:
 class ContentInterpreterAPI:
     def interpret(
         self, *, kind: str, query: str, resource_id: str = "", chat: Chat | None = None
-    ) -> Tuple[Interpretation, ResourceId | None]:
+    ) -> tuple[Interpretation, ResourceId | None]:
         raise NotImplementedError
 
 
 class TextProcessorAPI:
-    def process(self, text: str, chat: Chat | None = None) -> str:
+    def process(self, *, text: str, chat: Chat | None = None) -> str:
         raise NotImplementedError

agi_med_common/models/chat.py

@@ -1,7 +1,8 @@
 import warnings
 from copy import deepcopy
 from datetime import datetime
-from typing import Any, List, Dict, Literal, TypeVar, Callable
+from typing import Any, Literal, TypeVar
+from collections.abc import Callable
 
 from agi_med_common.models.chat_item import ChatItem, ReplicaItem, OuterContextItem
 from agi_med_common.models.widget import Widget
@@ -14,9 +15,9 @@ 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]
+StrDict = dict[str, Any]
 ContentBase = str | Widget | StrDict
-Content = ContentBase | List[ContentBase]
+Content = ContentBase | list[ContentBase]
 T = TypeVar("T")
 
 
@@ -184,7 +185,7 @@ ChatMessage = TypeUnion[HumanMessage, AIMessage, MiscMessage]
 
 class Chat(Base):
     context: Context = Field(default_factory=Context)
-    messages: List[ChatMessage] = Field(default_factory=list)
+    messages: list[ChatMessage] = Field(default_factory=list)
 
     class Config:
         extra = Extra.ignore
@@ -208,11 +209,11 @@ class Chat(Base):
     def add_message(self, message: ChatMessage):
         self.messages.append(message)
 
-    def add_messages(self, messages: List[ChatMessage]):
+    def add_messages(self, messages: list[ChatMessage]):
         for message in messages:
             self.messages.append(message)
 
-    def replace_messages(self, messages: List[ChatMessage]):
+    def replace_messages(self, messages: list[ChatMessage]):
         return self.model_copy(update=dict(messages=messages))
 
     def get_last_state(self, default: str = "empty") -> str:

agi_med_common/models/chat_item.py

@@ -1,5 +1,6 @@
 from datetime import datetime
-from typing import Annotated, Any, List, Callable
+from typing import Annotated, Any
+from collections.abc import Callable
 
 from agi_med_common.models.widget import Widget
 from pydantic import Field, ConfigDict, BeforeValidator, AfterValidator
@@ -126,11 +127,11 @@ class ChatItem(Base):
     def add_replica(self, replica: ReplicaItem):
         self.inner_context.replicas.append(replica)
 
-    def add_replicas(self, replicas: List[ReplicaItem]):
+    def add_replicas(self, replicas: list[ReplicaItem]):
         for replica in replicas:
             self.inner_context.replicas.append(replica)
 
-    def replace_replicas(self, replicas: List[ReplicaItem]):
+    def replace_replicas(self, replicas: list[ReplicaItem]):
         return self.model_copy(update=dict(inner_context=InnerContextItem(replicas=replicas)))
 
     def get_last_state(self, default: str = "empty") -> str:

agi_med_common/models/widget.py

@@ -1,12 +1,12 @@
-from typing import List, Self, Literal
+from typing import Self, Literal
 
 from pydantic import BaseModel, model_validator
 
 
 class Widget(BaseModel):
     type: Literal["widget"] = "widget"
-    buttons: List[List[str]] | None = None
-    ibuttons: List[List[str]] | None = None
+    buttons: list[list[str]] | None = None
+    ibuttons: list[list[str]] | None = None
 
     @model_validator(mode="after")
     def check(self) -> Self:

agi_med_common/parallel_map.py

@@ -1,4 +1,6 @@
-from typing import Any, Callable, TypeVar, Iterable
+from typing import Any, TypeVar
+from collections.abc import Callable
+from collections.abc import Iterable
 import concurrent.futures
 from tqdm import tqdm
 

agi_med_common/utils.py

@@ -3,7 +3,7 @@ import json
 import os
 from datetime import datetime
 from pathlib import Path
-from typing import Iterable
+from collections.abc import Iterable
 
 
 def make_session_id() -> str:

{agi_med_common-5.2.0.dist-info → agi_med_common-5.2.2.dist-info}/METADATA

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

{agi_med_common-5.2.0.dist-info → agi_med_common-5.2.2.dist-info}/RECORD

@@ -1,21 +1,21 @@
-agi_med_common/__init__.py,sha256=hIN6RedLcKaVjlFQpFbg0b-zzOtszrC2Pxv0PsYLm9E,828
-agi_med_common/api.py,sha256=0tnmyG-ROSEE_b4rmDznuD_2ZuoLP2XnwUAEDn7nAm4,1716
+agi_med_common/__init__.py,sha256=RTzZYi5dcgt97c2nh9CYdAhXoaH4kb8MQc6HMCzh_L4,828
+agi_med_common/api.py,sha256=83Ew5SvcypjvAWQ6c1kAcnvHsPxZ8BeBHnVHXLkUJvU,1687
 agi_med_common/file_storage.py,sha256=lEh9TyzUu8dS9WkaEcshSUxESXQ1i7PAyXqejdambpg,1802
-agi_med_common/parallel_map.py,sha256=drqv6yBfF43ShLl-U_8W1_B6tp3jQadxSnsf2AyTgXo,1037
+agi_med_common/parallel_map.py,sha256=O6pxgPlHT9OVqqgry1yHy_0bV0U2AwvGJ42iG2e834U,1091
 agi_med_common/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
-agi_med_common/utils.py,sha256=EeeXRv5BjrB-_GtHzI3V5swyVO0KPR-SMpjHRx2r2aM,1560
+agi_med_common/utils.py,sha256=9nV01aSqUcudWmDI_Jzvb_8e0LJM3WJ6vsf9Vcv8DB0,1569
 agi_med_common/validators.py,sha256=vMoPN42XzC8re-zdjekk5_lNQYHuTiAWD56YLvj2Z2w,2824
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
 agi_med_common/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agi_med_common/models/base.py,sha256=mKtXV2x51XVj7W-et9tjGcPMDUUUMelW-BywMgFc2p0,411
-agi_med_common/models/chat.py,sha256=LcmclAC16wBRQJ4YfehM-bLb24RpYQKi5ENz_bcKJWA,12327
-agi_med_common/models/chat_item.py,sha256=03YW6Exx3pDC4IsI2lsS_3_cYAwbqjYml2l9CpPMHVs,5541
+agi_med_common/models/chat.py,sha256=1ednqUNnJVRxJ4CWhaACETSazTNkj-rQo3tt3Jc1MB0,12342
+agi_med_common/models/chat_item.py,sha256=9Z4cfDSc2SQ2OTShOYdnNe4JEJMtdAY8bFCPT1eAyZY,5562
 agi_med_common/models/enums.py,sha256=J-GNpql9MCnKnWiV9aJRQGI-pAybvV86923RZs99grA,1006
 agi_med_common/models/tracks.py,sha256=HKDp-BX1p7AlDfSEKfOKCu0TRSK9cD4Dmq1vJt8oRjw,307
-agi_med_common/models/widget.py,sha256=aJZ2vWx_PTFN02Wz16eokz9IIVrxqNuZYVDqLG36toE,710
+agi_med_common/models/widget.py,sha256=KXSP3d4yQNeificMfzRxROdByw9IkxwCjkDD0kc7tcQ,704
 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=QvKTnp0VioXzd3_PUgNBeYx5RDTZT-45j-Ll-wXEI_o,421
-agi_med_common-5.2.0.dist-info/METADATA,sha256=hCKf-l7rnB4MyCEFmPVALtd2LSWGMCXUZ-bpjHbag_M,517
-agi_med_common-5.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agi_med_common-5.2.0.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-5.2.0.dist-info/RECORD,,
+agi_med_common-5.2.2.dist-info/METADATA,sha256=9Kk0ayatnwaQEei0hq5sJE7mmiUnl7aEJA5FWBnPzk0,517
+agi_med_common-5.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agi_med_common-5.2.2.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-5.2.2.dist-info/RECORD,,