fix schema models

ITISFoundation · Dec 6, 2024 · c8b30ca · c8b30ca
1 parent c530793
commit c8b30ca
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/packages/models-library/src/models_library/api_schemas_webserver/auth.py b/packages/models-library/src/models_library/api_schemas_webserver/auth.py
@@ -2,11 +2,11 @@
 from typing import Annotated, Any
 
 from models_library.basic_types import IDStr
-from pydantic import AliasGenerator, BaseModel, ConfigDict, Field, HttpUrl, SecretStr
+from pydantic import AliasGenerator, ConfigDict, Field, HttpUrl, SecretStr
 from pydantic.alias_generators import to_camel
 
 from ..emails import LowerCaseEmailStr
-from ._base import InputSchema
+from ._base import InputSchema, OutputSchema
 
 
 class AccountRequestInfo(InputSchema):
@@ -53,7 +53,7 @@ class UnregisterCheck(InputSchema):
 #
 
 
-class ApiKeyCreateRequest(BaseModel):
+class ApiKeyCreateRequest(OutputSchema):
     display_name: Annotated[str, Field(..., min_length=3)]
     expiration: timedelta | None = Field(
         None,
@@ -64,7 +64,6 @@ class ApiKeyCreateRequest(BaseModel):
         alias_generator=AliasGenerator(
             validation_alias=to_camel,
         ),
-        from_attributes=True,
         json_schema_extra={
             "examples": [
                 {
@@ -93,7 +92,6 @@ class ApiKeyCreateResponse(ApiKeyCreateRequest):
         alias_generator=AliasGenerator(
             serialization_alias=to_camel,
         ),
-        from_attributes=True,
         json_schema_extra={
             "examples": [
                 {
@@ -124,15 +122,14 @@ class ApiKeyCreateResponse(ApiKeyCreateRequest):
     )
 
 
-class ApiKeyGet(BaseModel):
+class ApiKeyGet(OutputSchema):
     id: IDStr
     display_name: Annotated[str, Field(..., min_length=3)]
 
     model_config = ConfigDict(
         alias_generator=AliasGenerator(
             serialization_alias=to_camel,
         ),
-        from_attributes=True,
         json_schema_extra={
             "examples": [
                 {