update pydantic

airbytehq · Jul 25, 2024 · 50ab89c · 50ab89c
1 parent 57477d8
commit 50ab89c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/airbyte-integrations/connectors/source-azure-blob-storage/source_azure_blob_storage/spec.py b/airbyte-integrations/connectors/source-azure-blob-storage/source_azure_blob_storage/spec.py
@@ -6,9 +6,8 @@
 from typing import Any, Dict, Literal, Optional, Union
 
 import dpath.util
-from airbyte_cdk import OneOfOptionConfig
 from airbyte_cdk.sources.file_based.config.abstract_file_based_spec import AbstractFileBasedSpec
-from pydantic import AnyUrl, BaseModel, Field
+from pydantic.v1 import AnyUrl, BaseModel, ConfigDict, Field
 
 
 class Oauth2(BaseModel):
@@ -30,7 +29,7 @@ class Oauth2(BaseModel):
         airbyte_secret=True,
     )
 
-    model_config = {"title": "Authenticate via Oauth2", "discriminator": "auth_type"}
+    model_config = ConfigDict(title="Authenticate via Oauth2", discriminator="auth_type")
 
 
 class StorageAccountKey(BaseModel):
@@ -44,7 +43,7 @@ class StorageAccountKey(BaseModel):
         order=3,
     )
 
-    model_config = {"title": "Authenticate via Storage Account Key", "discriminator": "auth_type"}
+    model_config = ConfigDict(title="Authenticate via Storage Account Key", discriminator="auth_type")
 
 
 class SourceAzureBlobStorageSpec(AbstractFileBasedSpec):