fix: Async Await

apps/apis/chats/__init__.py

@@ -14,7 +14,7 @@ router_base_configs = {
 
 # Message
 @router.post("/", **router_base_configs)
-def message(data: ChatsModel, fw_index=Depends(forward_middleware)):
+async def message(data: ChatsModel, fw_index=Depends(forward_middleware)):
     # Forward request
     fw_data = {
         "json": {"prompt": data.prompt}
@@ -29,7 +29,7 @@ def message(data: ChatsModel, fw_index=Depends(forward_middleware)):
 
 # Message to conversation
 @router.post("/{cid}", **router_base_configs)
-def conversation(cid: str, data: ChatsModel, fw_index=Depends(forward_middleware)):
+async def conversation(cid: str, data: ChatsModel, fw_index=Depends(forward_middleware)):
     is_exist = is_conversation_storage_exist(cid)
     if not is_exist:
         raise HTTPException(status_code=404, detail="Conversation not found")
@@ -52,7 +52,7 @@ def conversation(cid: str, data: ChatsModel, fw_index=Depends(forward_middleware
 
 # Get conversation by id
 @router.get("/{cid}", **router_base_configs)
-def get_conversation(cid: str):
+async def get_conversation(cid: str):
     is_exist = is_conversation_storage_exist(cid)
     if not is_exist:
         raise HTTPException(status_code=404, detail="Conversation not found")
@@ -63,13 +63,13 @@ def get_conversation(cid: str):
 
 # Create conversation
 @router.get("/create", **router_base_configs)
-def create_conversation():
+async def create_conversation():
     cid = create_conversation_storage()
     return {"conversation_id": cid}
 
 
 # Delete conversation
 @router.delete("/delete/{cid}", **router_base_configs)
-def delete_conversation(cid: str):
+async def delete_conversation(cid: str):
     delete_conversation_storage(cid)
     return {"conversation_id": cid}

apps/apis/img2text/__init__.py

@@ -12,7 +12,7 @@ router_base_configs = {
 
 # Image to text
 @router.post("/", **router_base_configs)
-def image_to_text(
+async def image_to_text(
     image: Img2TextModel.image = Img2TextModel.image_default,
     fw_index: Img2TextModel.fw_index = Depends(forward_middleware)
 ):

apps/apis/rembg/__init__.py

@@ -44,7 +44,7 @@ async def remove_background(
     # Read image
     image_bytes = await image.read()
     # Process image
-    processed_image = await rembg_controller(image_bytes)
+    processed_image = rembg_controller(image_bytes)
     # If stream is True, return StreamingResponse
     if RembgModel.stream_parser(stream):
         return StreamingResponse(processed_image, media_type="application/octet-stream", headers={"Content-Disposition": f"attachment;filename={image.filename}"})

apps/apis/rembg/controllers/rembg_controller.py

@@ -3,7 +3,7 @@ import rembg
 import io
 
 
-async def rembg_controller(image: bytes):
+def rembg_controller(image: bytes):
     # Remove background
     image = rembg.remove(image)
     # Return image

apps/apis/vqa/__init__.py

@@ -12,7 +12,7 @@ router_base_configs = {
 
 
 @router.post("/", **router_base_configs)
-def vision_question_answer(
+async def vision_question_answer(
     image: VQAModel.image = VQAModel.image_default,
     question: VQAModel.question = VQAModel.question_default,
     fw_index: VQAModel.fw_index = Depends(forward_middleware)

apps/services/foward/__init__.py

@@ -13,19 +13,19 @@ router_base_configs = {
 
 # Create foward destination
 @router.post("/", **router_base_configs)
-def create_forward_destination(data: CreateFwDestinationModel):
+async def create_forward_destination(data: CreateFwDestinationModel):
     index = append_fw_destination(data.url)
     return {"fw_index": index}
 
 
 # Delete foward destination
 @router.delete("/", **router_base_configs)
-def delete_forward_destination(data: DeleteFwDestinationModel):
+async def delete_forward_destination(data: DeleteFwDestinationModel):
     delete_fw_destination(data.index)
     return {"fw_index": data.index}
 
 
 # Get foward destinations
 @router.get("/", **router_base_configs)
-def get_forward_destinations():
+async def get_forward_destinations():
     return {"fw": get_fw_destinations()}

apps/services/foward/fw_middleware.py

@@ -4,7 +4,7 @@ from fastapi import HTTPException
 from .fw_storage import get_fw_destinations, update_fw_destination
 
 
-async def verify_fw_resources(url: str) -> bool:
+def verify_fw_resources(url: str) -> bool:
     # Verify forward destination resources logger
     print("Verifying resources on", url)
     try:
@@ -35,7 +35,8 @@ async def forward_middleware(fw: str = None):
             # Verify forward destination resources
             fw = None
             for i in range(0, len(fw_destinations)):
-                is_resource_available = await verify_fw_resources(fw_destinations[i]['url'])
+                is_resource_available = verify_fw_resources(
+                    fw_destinations[i]['url'])
                 if is_resource_available:
                     fw = i
                     break