Update main.py

main.py

@@ -144,7 +144,7 @@ async def status_check():
     return {"Status": "API is online..."}
 
 
-async def endpoint_output(endpoint_result: ResultItem, code: int = 0, error: str = None) -> Union[ErrorResponse, EndpointResponse]:
+async def endpoint_output(endpoint_result: Result, code: int = 0, error: str = None) -> Union[ErrorResponse, EndpointResponse]:
     msg = 'Execution failed'
     output = ErrorResponse(**{'execution_msg': msg,
                               'execution_code': code, 'error': error})
@@ -173,7 +173,7 @@ async def endpoint_output(endpoint_result: ResultItem, code: int = 0, error: str
 # Caching Post requests is challenging
 async def sms_posts(instance: Result, idx: str = None, action: str = "add") -> Union[ErrorResponse, EndpointResponse]:
     async with AsyncSession(sms_resource["engine"]) as session:
-        code = 0
+        code = 1
         error = None
         existing = await session.get(instance.__class__, idx)
 
@@ -184,14 +184,13 @@ async def sms_posts(instance: Result, idx: str = None, action: str = "add") -> U
             if checker:
                 if action == "delete":
                     await session.delete(instance)
-                    code = 1
                     await session.commit()
                 else:  # add or update use add
                     session.add(instance) # Not asynchronous
-                    code = 1
                     await session.commit()
                     await session.refresh(instance)                
         except Exception as e:
+            code = 0
             error = e
 
         finally: