gabcares commited on
Commit
3a04dfd
·
verified ·
1 Parent(s): 2eefc8a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -2
main.py CHANGED
@@ -24,6 +24,8 @@ from sqlmodel import SQLModel, select
24
  from sqlmodel.sql.expression import SelectOfScalar
25
  from sqlmodel.ext.asyncio.session import AsyncSession
26
 
 
 
27
  from sqlalchemy.ext.asyncio import create_async_engine
28
  from sqlalchemy import Engine
29
 
@@ -124,13 +126,14 @@ Result = Union[OneResult, BulkResult]
124
  ResultItem = Union[OneResultItem, BulkResultItem]
125
 
126
 
127
- class EndpointResponse(SQLModel):
128
  execution_msg: str
129
  execution_code: int
130
  result: ResultItem
 
131
 
132
 
133
- class ErrorResponse(SQLModel):
134
  execution_msg: str
135
  execution_code: int
136
  error: Optional[str]
 
24
  from sqlmodel.sql.expression import SelectOfScalar
25
  from sqlmodel.ext.asyncio.session import AsyncSession
26
 
27
+ from pydantic import BaseModel
28
+
29
  from sqlalchemy.ext.asyncio import create_async_engine
30
  from sqlalchemy import Engine
31
 
 
126
  ResultItem = Union[OneResultItem, BulkResultItem]
127
 
128
 
129
+ class EndpointResponse(BaseModel):
130
  execution_msg: str
131
  execution_code: int
132
  result: ResultItem
133
+ # result: Any
134
 
135
 
136
+ class ErrorResponse(BaseModel):
137
  execution_msg: str
138
  execution_code: int
139
  error: Optional[str]