aidial-adapter-anthropic 0.1.0__py3-none-any.whl → 0.2.0rc0__py3-none-any.whl

aidial_adapter_anthropic/__init__.py

@@ -0,0 +1,5 @@
+import os
+import warnings
+
+if os.getenv("PYDANTIC_V2", "0").lower() not in ("1", "true"):
+    warnings.warn("PYDANTIC_V2 env variable is expected to be set to True")

aidial_adapter_anthropic/_utils/json.py

@@ -57,7 +57,7 @@ def _to_dict(obj: Any, **kwargs) -> Any:
         return tuple(rec(element) for element in obj)
 
     if isinstance(obj, BaseModel):
-        return rec(obj.dict())
+        return rec(obj.model_dump())
 
     if hasattr(obj, "to_dict"):
         return rec(obj.to_dict())

aidial_adapter_anthropic/adapter/_base.py

@@ -15,10 +15,7 @@ from aidial_adapter_anthropic.dial.request import (
 )
 
 
-class ChatCompletionAdapter(ABC, BaseModel):
-    class Config:
-        arbitrary_types_allowed = True
-
+class ChatCompletionAdapter(ABC):
     @abstractmethod
     async def chat(
         self,

aidial_adapter_anthropic/adapter/_claude/adapter.py

@@ -210,6 +210,7 @@ async def create_adapter(
     )(model)
 
 
+@dataclass
 class Adapter(ChatCompletionAdapter):
     deployment: str
     storage: Optional[FileStorage]

aidial_adapter_anthropic/adapter/_claude/config.py

@@ -2,9 +2,11 @@ from typing import List, Literal
 
 from anthropic.types.anthropic_beta_param import AnthropicBetaParam
 from anthropic.types.beta import BetaThinkingConfigParam as ThinkingConfigParam
-from pydantic import Field
+from pydantic import BaseModel, ConfigDict, Field
 
-from aidial_adapter_anthropic._utils.pydantic import ExtraForbidModel
+
+class ExtraForbidModel(BaseModel):
+    model_config = ConfigDict(extra="forbid")
 
 
 class ThinkingConfigEnabled(ExtraForbidModel):

aidial_adapter_anthropic/adapter/_claude/state.py

@@ -21,7 +21,7 @@ class MessageState(BaseModel):
     claude_message_content: List[ParsedContentBlock] | List[ContentBlock]
 
     def to_dict(self) -> dict:
-        return self.dict(
+        return self.model_dump(
             # FIXME: a hack to exclude the private __json_buf field
             exclude={"claude_message_content": {"__all__": {"__json_buf"}}},
             # Excluding `citations: null`, since they could not be even parsed
@@ -35,7 +35,7 @@ def get_message_content_from_state(
 ) -> List[ContentBlockParam] | None:
     if (cc := message.custom_content) and (state_dict := cc.state):
         try:
-            state = MessageState.parse_obj(state_dict)
+            state = MessageState.model_validate(state_dict)
             return [block.to_dict() for block in state.claude_message_content]  # type: ignore
         except pydantic.ValidationError as e:
             _log.error(

aidial_adapter_anthropic/adapter/_decorator/base.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Callable, List
 
 from aidial_sdk.chat_completion import Message
@@ -9,6 +10,7 @@ from aidial_adapter_anthropic.dial.consumer import Consumer
 from aidial_adapter_anthropic.dial.request import ModelParameters
 
 
+@dataclass
 class ChatCompletionDecorator(ChatCompletionAdapter):
     adapter: ChatCompletionAdapter
 

aidial_adapter_anthropic/adapter/_decorator/preprocess.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Callable, List
 
 from aidial_sdk.chat_completion import Message
@@ -20,6 +21,7 @@ def preprocess_messages_decorator(
     )
 
 
+@dataclass
 class PreprocessMessagesDecorator(ChatCompletionDecorator):
     on_messages: Callable[[List[Message]], ListProjection[Message]]
 

aidial_adapter_anthropic/adapter/_decorator/replicator.py

@@ -22,7 +22,7 @@ class ReplicatorDecorator(ChatCompletionDecorator):
         params: ModelParameters,
         messages: List[Message],
     ) -> None:
-        params1 = params.copy()
+        params1 = params.model_copy()
         params1.n = 1
 
         async def _chat(root_consumer: Consumer):

aidial_adapter_anthropic/adapter/_truncate_prompt.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+from dataclasses import dataclass
 from typing import Awaitable, Callable, List, Optional, Set, Tuple, TypeVar
 
 from aidial_sdk.exceptions import ContextLengthExceededError
@@ -7,7 +8,6 @@ from aidial_sdk.exceptions import (
     InvalidRequestError,
     TruncatePromptSystemAndLastUserError,
 )
-from pydantic import BaseModel
 
 from aidial_adapter_anthropic._utils.list import (
     omit_by_indices,
@@ -15,7 +15,7 @@ from aidial_adapter_anthropic._utils.list import (
 )
 
 
-class TruncatePromptError(ABC, BaseModel):
+class TruncatePromptError(ABC):
     @abstractmethod
     def to_dial_exception(self) -> DialException:
         pass
@@ -24,6 +24,7 @@ class TruncatePromptError(ABC, BaseModel):
         return self.to_dial_exception().message
 
 
+@dataclass
 class InconsistentLimitsError(TruncatePromptError):
     user_limit: int
     model_limit: int
@@ -35,6 +36,7 @@ class InconsistentLimitsError(TruncatePromptError):
         )
 
 
+@dataclass
 class ModelLimitOverflow(TruncatePromptError):
     model_limit: int
     token_count: int
@@ -43,6 +45,7 @@ class ModelLimitOverflow(TruncatePromptError):
         return ContextLengthExceededError(self.model_limit, self.token_count)
 
 
+@dataclass
 class UserLimitOverflow(TruncatePromptError):
     user_limit: int
     token_count: int

aidial_adapter_anthropic/dial/_attachments.py

@@ -50,10 +50,8 @@ class HandlerWithConfig(Protocol, Generic[_T, _Config]):
     def __call__(self, resource: Resource, config: _Config | None) -> _T: ...
 
 
-class AttachmentProcessor(BaseModel, Generic[_T, _Config]):
-    class Config:
-        arbitrary_types_allowed = True
-
+@dataclass
+class AttachmentProcessor(Generic[_T, _Config]):
     supported_types: Dict[str, Set[str]]
     """MIME type to file extensions mapping"""
 
@@ -89,11 +87,12 @@ class WithResources(Generic[_T]):
         return WithResources(payload=payload, resources=resources)
 
 
-class AttachmentProcessors(BaseModel, Generic[_Txt, _T, _Config]):
-    config: _Config | None = None
+@dataclass
+class AttachmentProcessors(Generic[_Txt, _T, _Config]):
     attachment_processors: Sequence[AttachmentProcessor[_T, _Config]]
     text_handler: Callable[[str], _Txt]
     file_storage: FileStorage | None
+    config: _Config | None = field(default=None)
 
     @property
     def supported_types(self) -> Dict[str, Set[str]]:

aidial_adapter_anthropic/dial/consumer.py

@@ -179,7 +179,7 @@ class ChoiceConsumer(Consumer):
         self._citations[document_id] = (display_index, document)
 
         if document:
-            document = document.copy()
+            document = document.model_copy()
             document.title = f"[{display_index}] {document.title or ''}".strip()
             document.reference_type = document.reference_type or document.type
             document.reference_url = document.reference_url or document.url

aidial_adapter_anthropic/dial/request.py

@@ -20,7 +20,7 @@ from aidial_sdk.chat_completion.request import (
 )
 from aidial_sdk.exceptions import RequestValidationError
 from pydantic import BaseModel
-from pydantic.v1 import ValidationError as PydanticValidationError
+from pydantic import ValidationError as PydanticValidationError
 
 from aidial_adapter_anthropic.adapter._errors import ValidationError
 from aidial_adapter_anthropic.dial.tools import (
@@ -82,9 +82,6 @@ class ModelParameters(BaseModel):
             configuration=configuration,
         )
 
-    def add_stop_sequences(self, stop: List[str]) -> "ModelParameters":
-        return self.copy(update={"stop": [*self.stop, *stop]})
-
     @property
     def tools_mode(self) -> ToolsMode | None:
         if self.tool_config is not None:
@@ -93,7 +90,7 @@ class ModelParameters(BaseModel):
 
     def parse_configuration(self, cls: Type[_Model]) -> _Model:
         try:
-            return cls.parse_obj(self.configuration or {})
+            return cls.model_validate(self.configuration or {})
         except PydanticValidationError as e:
             if self.configuration is None:
                 msg = "The configuration at path 'custom_fields.configuration' is missing."

aidial_adapter_anthropic/dial/resource.py

@@ -4,7 +4,7 @@ from abc import ABC, abstractmethod
 from typing import List
 
 from aidial_sdk.chat_completion import Attachment
-from pydantic import BaseModel, Field, root_validator, validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 
 from aidial_adapter_anthropic._utils.resource import Resource
 from aidial_adapter_anthropic._utils.text import truncate_string
@@ -34,7 +34,7 @@ class UnsupportedContentType(ValidationError):
 
 
 class DialResource(ABC, BaseModel):
-    entity_name: str = Field(default=None)
+    entity_name: str | None = None
     supported_types: List[str] | None = Field(default=None)
 
     @abstractmethod
@@ -77,10 +77,10 @@ class URLResource(DialResource):
     def to_attachment(self) -> Attachment:
         return Attachment(type=self.content_type, url=self.url)
 
-    @root_validator
-    def validator(cls, values):
-        values["entity_name"] = values.get("entity_name") or "URL"
-        return values
+    @model_validator(mode="after")
+    def default_entity_name(self):
+        self.entity_name = self.entity_name or "URL"
+        return self
 
     async def download(self, storage: FileStorage | None) -> Resource:
         type = await self.get_content_type()
@@ -114,10 +114,10 @@ class AttachmentResource(DialResource):
     def to_attachment(self) -> Attachment:
         return self.attachment
 
-    @validator("attachment", pre=True)
+    @field_validator("attachment", mode="before")
     def parse_attachment(cls, value):
         if isinstance(value, dict):
-            attachment = Attachment.parse_obj(value)
+            attachment = Attachment.model_validate(value)
             # Working around the issue of defaulting missing type to a markdown:
             # https://github.com/epam/ai-dial-sdk/blob/2835107e950c89645a2b619fecba2518fa2d7bb1/aidial_sdk/chat_completion/request.py#L22
             if "type" not in value:
@@ -125,10 +125,10 @@ class AttachmentResource(DialResource):
             return attachment
         return value
 
-    @root_validator(pre=True)
-    def validator(cls, values):
-        values["entity_name"] = values.get("entity_name") or "attachment"
-        return values
+    @model_validator(mode="after")
+    def default_entity_name(self):
+        self.entity_name = self.entity_name or "attachment"
+        return self
 
     async def download(self, storage: FileStorage | None) -> Resource:
         type = await self.get_content_type()

aidial_adapter_anthropic/dial/storage.py

@@ -3,11 +3,12 @@ import hashlib
 import io
 import logging
 import mimetypes
-from typing import Mapping, Optional, TypedDict
+from typing import Mapping, Optional
 from urllib.parse import unquote, urljoin
 
 import aiohttp
 from pydantic import BaseModel
+from typing_extensions import TypedDict
 
 _log = logging.getLogger(__name__)
 

{aidial_adapter_anthropic-0.1.0.dist-info → aidial_adapter_anthropic-0.2.0rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: aidial-adapter-anthropic
-Version: 0.1.0
+Version: 0.2.0rc0
 Summary: Package implementing adapter from DIAL Chat Completions API to Anthropic API
 License: Apache-2.0
 Keywords: ai
@@ -12,7 +12,7 @@ Requires-Dist: aidial-sdk (>=0.28.0,<1)
 Requires-Dist: aiohttp (>=3.13.3,<4)
 Requires-Dist: anthropic (>=0.75.0,<1)
 Requires-Dist: pillow (>=10.4.0,<11)
-Requires-Dist: pydantic (>=1.10.17,<3)
+Requires-Dist: pydantic (>=2.8.2,<3)
 Project-URL: Documentation, https://epam-rail.com/dial_api
 Project-URL: Homepage, https://epam-rail.com
 Project-URL: Repository, https://github.com/epam/ai-dial-adapter-anthropic/

{aidial_adapter_anthropic-0.1.0.dist-info → aidial_adapter_anthropic-0.2.0rc0.dist-info}/RECORD

@@ -1,39 +1,39 @@
-aidial_adapter_anthropic/_utils/json.py,sha256=IknyukIXEzQkZ81zo5ukUexvtvnlfRrlxvvYCekoAkc,2941
+aidial_adapter_anthropic/__init__.py,sha256=gWNLIVwPavaMsawwgowSX3-3hM6dxctGgQ3HSw2a7l8,166
+aidial_adapter_anthropic/_utils/json.py,sha256=suurdZDIu3tWbaWVscL4WU-mF-1yJPZ0bNOIKN_IBxs,2947
 aidial_adapter_anthropic/_utils/list.py,sha256=SHC9KD2ckQCvtzVq31KuvQfl-RMfAfyVkPqDviOLuIY,1996
-aidial_adapter_anthropic/_utils/pydantic.py,sha256=7h0cThzw0nHOL5H05yMzbkwilZ8tegCjb0iRdjOrtl4,111
 aidial_adapter_anthropic/_utils/resource.py,sha256=69i3zt7OP4_uceM4_5xqhBCW_pQOfrgEFhZoAdAS9yk,1563
 aidial_adapter_anthropic/_utils/text.py,sha256=d9LJdO9jom3B7_dS3SaqjlNvngOFrgzsZHdQye92Kyo,106
 aidial_adapter_anthropic/adapter/__init__.py,sha256=jgJHHpSRJqObAPx3TBcsxZqNL3jw7g9bDnbcfqml8HY,222
-aidial_adapter_anthropic/adapter/_base.py,sha256=MEpRlxgZsaZRedMtQyqJaeFDHhUz5Af9CGJnnqu3pDA,2818
-aidial_adapter_anthropic/adapter/_claude/adapter.py,sha256=oT0BIXX7rbLGphe9RdiFZrNywk5hopn1fNquJTpfDVw,20484
+aidial_adapter_anthropic/adapter/_base.py,sha256=ETtsTm3RhU0clGEt7bhNmOc5N80RH_-oKM6eKd-UJ3k,2749
+aidial_adapter_anthropic/adapter/_claude/adapter.py,sha256=Kr4TPBqPooMazfy7AmJ4H2NMVSaI7-eQRoUeL0gCAWA,20495
 aidial_adapter_anthropic/adapter/_claude/blocks.py,sha256=wXP7UDILn5Az06BAptLo24VFZkXf16ofhlCJsJCclgU,3907
 aidial_adapter_anthropic/adapter/_claude/citations.py,sha256=NBxl_LxP8dUaG0_Xej0Eo5Yst5X8farQdomggdJm9iE,2175
-aidial_adapter_anthropic/adapter/_claude/config.py,sha256=pKfoUUoUOEyYbYR0KL88F3lvCTPNxzGF6L6dgLCn2fI,1314
+aidial_adapter_anthropic/adapter/_claude/config.py,sha256=Z7pu73LT08TLcv7gbn1iQFRNTUvoh-WHeQ5Ya-u_cvk,1349
 aidial_adapter_anthropic/adapter/_claude/converters.py,sha256=IanS08NHoT1jgcV6_Hf5MzuiE5HdKP9K29A70fVXNr4,10353
 aidial_adapter_anthropic/adapter/_claude/params.py,sha256=ymFwYEjgq1EjuOM2GsojlcQpUkDvy_DLR8euCHrqsTI,981
-aidial_adapter_anthropic/adapter/_claude/state.py,sha256=hhixpQKGn-F-eDPMLwsFCV1wWcxOOz3uXIM3pgvJmDU,1499
+aidial_adapter_anthropic/adapter/_claude/state.py,sha256=eC-Qeby0f68pFu10lwfRJT6ANowFTiIUM1QEE68D60o,1510
 aidial_adapter_anthropic/adapter/_claude/tokenizer/__init__.py,sha256=zA6TwQu4qMH5XVBOKetsQuPs4Prx1-uNsFopiTBpx_A,256
 aidial_adapter_anthropic/adapter/_claude/tokenizer/anthropic.py,sha256=EGw7N-6V-huH3FOwrIeHUb5jYpWfinYhZEvGd1la_fs,1738
 aidial_adapter_anthropic/adapter/_claude/tokenizer/approximate.py,sha256=MfuH2E-I7vqJgYkp0hKbmDSNC1HopWr592n7gKUa8p4,10427
 aidial_adapter_anthropic/adapter/_claude/tokenizer/base.py,sha256=2wivuP51FXaieqdRgMpBoPARIgkcB9ypyOlTYVykcng,816
 aidial_adapter_anthropic/adapter/_claude/tools.py,sha256=TrsM7fTbm22FhSKLi2aeCE7VGFWgVoKDA_pnZihiKp8,3053
-aidial_adapter_anthropic/adapter/_decorator/base.py,sha256=fRIFpWaYKpo-nEbLapjWX3owu2Uu3MOLTmRZZD73Xv8,1730
-aidial_adapter_anthropic/adapter/_decorator/preprocess.py,sha256=VmyS6rG5sJwCVBzLmwy5qU0_9T2o-vbylFevGxnSsro,2204
-aidial_adapter_anthropic/adapter/_decorator/replicator.py,sha256=JgQ5qAJJJzOJG-byKHRedqyzC9ETn3CkpkfHUCBJ-rI,959
+aidial_adapter_anthropic/adapter/_decorator/base.py,sha256=rBl_BorYjejRR6Q2syyjq3zDEEpb_DNfdvYWY525C_c,1775
+aidial_adapter_anthropic/adapter/_decorator/preprocess.py,sha256=ALKfa5g9pb5i8YJxkJULCiNt_kBdaCbYJTmfhuyhJ2Q,2249
+aidial_adapter_anthropic/adapter/_decorator/replicator.py,sha256=jQSDEe-uwFbZ6NF6CSdYXP4SvHEOci6UCOOTcHzK0v8,965
 aidial_adapter_anthropic/adapter/_errors.py,sha256=pBm-JmTv4Y9m24vKqGm-Kh2vm6fv-jbM_I2_lMZkpCk,3058
 aidial_adapter_anthropic/adapter/_tokenize.py,sha256=jv43xTtTbo1mZahXVpL24Bf9ON8xNluctH9na2_9PJg,566
-aidial_adapter_anthropic/adapter/_truncate_prompt.py,sha256=TBUpbGiaRmquzwiqLJnR8oah4AJjrMAib26SksCd9C0,4738
+aidial_adapter_anthropic/adapter/_truncate_prompt.py,sha256=TxtHSVpSQNqxn1gGLfQgycF2arnn__l5C5D9BNNbwMo,4763
 aidial_adapter_anthropic/adapter/claude.py,sha256=eF3K4yV1tODNF_CPYmhnZQIs6LD4e6MqE-aJz8onD90,500
-aidial_adapter_anthropic/dial/_attachments.py,sha256=-laRPBBDGxA3u1lso19zL-sPNbelPrA0Lc5OWhd3GTM,8218
+aidial_adapter_anthropic/dial/_attachments.py,sha256=dDngPo0BJhYclPl3942uSgtXay2hf8WTeUkUBoZqVJw,8175
 aidial_adapter_anthropic/dial/_lazy_stage.py,sha256=I6ADVm9HonhUr9TIrbO8yV3nkK1Pwb662gzUFZW_6Ys,1000
 aidial_adapter_anthropic/dial/_message.py,sha256=X1t6M56rGPhhnlI7qutNZc-qlwtzdAZ2OwNo6yP9JAk,9895
-aidial_adapter_anthropic/dial/consumer.py,sha256=XWKzt-srYJbWm2S4smBTupIFlZfYdkhGckE8sLCRkCY,7304
-aidial_adapter_anthropic/dial/request.py,sha256=Xi_pROYLDuYyTbKpDwwPqszaFWvpNmgsCpFiuP9CIh0,5015
-aidial_adapter_anthropic/dial/resource.py,sha256=Zvb-j28LFi_HIozGHbADdGbOVqgPZVyF5StQ4vqdcWo,5927
-aidial_adapter_anthropic/dial/storage.py,sha256=i0TezPbfbvWIPu3RoxxDSq9aOuQXXJY33J6yARYiuUU,4365
+aidial_adapter_anthropic/dial/consumer.py,sha256=y3zvrWUShS_j8D7mUR9d-biMCCbZMlZA9tGNl1jG-3Y,7310
+aidial_adapter_anthropic/dial/request.py,sha256=PmJ10rVHGghRFMd8RowY0hNsMe_L2ofP_a8F6bMX0Mk,4881
+aidial_adapter_anthropic/dial/resource.py,sha256=ekYrSOaPExHbJc7CokqqUNBWFw3rx4ydsK-6ZUoTL6k,5936
+aidial_adapter_anthropic/dial/storage.py,sha256=ptSn-p5D7iuC19xHRPNacUOp5TmjtHNnA_hditiqlLY,4394
 aidial_adapter_anthropic/dial/token_usage.py,sha256=YICZT1HyukT3eLGamhqnDm5ygWqJ-_ZT8jUo27jW5iE,638
 aidial_adapter_anthropic/dial/tools.py,sha256=QUiMl4dQwv_rdxBUIMqh0Oh65YOsu4KsSufmbj75W8w,5610
-aidial_adapter_anthropic-0.1.0.dist-info/LICENSE,sha256=6Uw_PHOiAbebWhTu8GKOQaW-26oTFnPuBZXQptqkqbw,11348
-aidial_adapter_anthropic-0.1.0.dist-info/METADATA,sha256=WE90tZqlL0ZQ7kEDaiFFgBuUjYugJ2sFhJ7RdbvTI6A,2586
-aidial_adapter_anthropic-0.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-aidial_adapter_anthropic-0.1.0.dist-info/RECORD,,
+aidial_adapter_anthropic-0.2.0rc0.dist-info/LICENSE,sha256=6Uw_PHOiAbebWhTu8GKOQaW-26oTFnPuBZXQptqkqbw,11348
+aidial_adapter_anthropic-0.2.0rc0.dist-info/METADATA,sha256=wNIEHK43H19eGguXnDefqG2pLgzzpVsefMtnpN0PWK4,2587
+aidial_adapter_anthropic-0.2.0rc0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+aidial_adapter_anthropic-0.2.0rc0.dist-info/RECORD,,

aidial_adapter_anthropic/_utils/pydantic.py

@@ -1,6 +0,0 @@
-from pydantic import BaseModel
-
-
-class ExtraForbidModel(BaseModel):
-    class Config:
-        extra = "forbid"