update routes

http-api/routes/courses/__init__.py

@@ -1,11 +1,11 @@
 from http import HTTPStatus
 
-from aws_lambda_powertools.event_handler import Response, content_types
 from aws_lambda_powertools.event_handler.api_gateway import Router
 from aws_lambda_powertools.event_handler.exceptions import NotFoundError
 from layercake.dynamodb import DynamoDBCollection, DynamoDBPersistenceLayer, KeyPair
 from meilisearch import Client as Meilisearch
 
+from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from course import create_course, update_course
 from middlewares import AuditLogMiddleware, Tenant, TenantMiddleware
@@ -61,13 +61,13 @@ def get_courses():
 )
 def post_course(payload: Course):
     tenant: Tenant = router.context['tenant']
-    org = Org(id=tenant.id, name=tenant.name)
-
-    create_course(payload, org, persistence_layer=course_layer)
-
-    return Response(
+    create_course(
+        payload,
+        Org(id=tenant.id, name=tenant.name),
+        persistence_layer=course_layer,
+    )
+    return JSONResponse(
         body=payload,
-        content_type=content_types.APPLICATION_JSON,
         status_code=HTTPStatus.CREATED,
     )
 
@@ -91,9 +91,7 @@ def get_course(id: str):
 )
 def put_course(id: str, payload: Course):
     update_course(id, payload, persistence_layer=course_layer)
-
-    return Response(
+    return JSONResponse(
         body=payload,
-        content_type=content_types.APPLICATION_JSON,
         status_code=HTTPStatus.OK,
     )