Move everything to pydantic v1 to avoid warnings

aidial_adapter_vertexai/chat/bison/prompt.py

@@ -2,7 +2,7 @@
 from typing import List, Optional, Set, Tuple
 
 from aidial_sdk.chat_completion import Message, Role
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 from vertexai.preview.language_models import ChatMessage, ChatSession
 
 from aidial_adapter_vertexai.chat.errors import ValidationError

aidial_adapter_vertexai/dial_api/request.py

@@ -8,7 +8,7 @@
     MessageContentTextPart,
     Request,
 )
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from aidial_adapter_vertexai.chat.errors import ValidationError
 

aidial_adapter_vertexai/dial_api/storage.py

@@ -6,7 +6,7 @@
 from urllib.parse import unquote, urljoin
 
 import aiohttp
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 from typing_extensions import TypedDict
 
 from aidial_adapter_vertexai.utils.log_config import app_logger as log

aidial_adapter_vertexai/embedding/multi_modal.py

@@ -4,7 +4,7 @@
 from aidial_sdk.chat_completion.request import Attachment
 from aidial_sdk.embeddings import Response as EmbeddingsResponse
 from aidial_sdk.embeddings.request import EmbeddingsRequest
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 from vertexai.vision_models import (
     Image,
     MultiModalEmbeddingModel,

aidial_adapter_vertexai/embedding/text.py

@@ -3,7 +3,7 @@
 
 from aidial_sdk.embeddings import Response as EmbeddingsResponse
 from aidial_sdk.embeddings.request import EmbeddingsRequest
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 from vertexai.language_models import TextEmbeddingInput
 
 from aidial_adapter_vertexai.chat.errors import ValidationError

aidial_adapter_vertexai/utils/json.py

@@ -10,7 +10,7 @@
 from typing import Any
 
 import proto
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from aidial_adapter_vertexai.utils.protobuf import message_to_dict
 

tests/integration_tests/test_chat_completion.py

@@ -14,7 +14,7 @@
 )
 from openai.types.chat.chat_completion_message import FunctionCall
 from openai.types.chat.completion_create_params import Function
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from aidial_adapter_vertexai.chat.static_tools import StaticToolsConfig
 from aidial_adapter_vertexai.deployments import ChatCompletionDeployment

tests/utils/openai.py

@@ -36,7 +36,7 @@
 )
 from openai.types.chat.completion_create_params import Function
 from openai.types.shared_params.function_definition import FunctionDefinition
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from aidial_adapter_vertexai.chat.static_tools import StaticToolsConfig
 from aidial_adapter_vertexai.utils.resource import Resource