Working on document repository

src/file-processor/app/models/user.py

@@ -13,34 +13,7 @@ from pydantic import BaseModel, Field, field_validator, EmailStr
 from pydantic_core import core_schema
 
 from app.models.auth import UserRole
-
-
-class PyObjectId(ObjectId):
-  """Custom ObjectId type for Pydantic v2 compatibility."""
-  
-  @classmethod
-  def __get_pydantic_core_schema__(
-      cls, source_type: Any, handler
-  ) -> core_schema.CoreSchema:
-    return core_schema.json_or_python_schema(
-      json_schema=core_schema.str_schema(),
-      python_schema=core_schema.union_schema([
-          core_schema.is_instance_schema(ObjectId),
-          core_schema.chain_schema([
-              core_schema.str_schema(),
-              core_schema.no_info_plain_validator_function(cls.validate),
-          ])
-      ]),
-      serialization=core_schema.plain_serializer_function_ser_schema(
-        lambda x: str(x)
-      ),
-    )
-  
-  @classmethod
-  def validate(cls, v):
-    if not ObjectId.is_valid(v):
-      raise ValueError("Invalid ObjectId")
-    return ObjectId(v)
+from app.models.types import PyObjectId
 
 
 def validate_password_strength(password: str) -> str: