Update main.py

main.py

@@ -24,6 +24,8 @@ from sqlmodel import SQLModel, select
 from sqlmodel.sql.expression import SelectOfScalar
 from sqlmodel.ext.asyncio.session import AsyncSession
 
+from pydantic import BaseModel
+
 from sqlalchemy.ext.asyncio import create_async_engine
 from sqlalchemy import Engine
 
@@ -124,13 +126,14 @@ Result = Union[OneResult, BulkResult]
 ResultItem = Union[OneResultItem, BulkResultItem]
 
 
-class EndpointResponse(SQLModel):
+class EndpointResponse(BaseModel):
     execution_msg: str
     execution_code: int
     result: ResultItem
+    # result: Any
 
 
-class ErrorResponse(SQLModel):
+class ErrorResponse(BaseModel):
     execution_msg: str
     execution_code: int
     error: Optional[str]