Update api/utils.py

api/utils.py

@@ -7,7 +7,6 @@ from api.logger import logger
 from api.config import MODEL_MAPPING, EDITEA_API_ENDPOINT, EDITEA_HEADERS
 from fastapi import HTTPException
 
-# Editee class is now integrated within utils.py
 class Editee:
     label = "Editee"
     url = "https://editee.com"
@@ -45,7 +44,6 @@ class Editee:
         **kwargs
     ):
         model = cls.get_model(model)
-
         headers = EDITEA_HEADERS
 
         async with ClientSession(headers=headers) as session:
@@ -109,7 +107,7 @@ async def process_response(request: ChatRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 # Function to process streaming response
-def process_response_stream(request: ChatRequest):
+async def process_response_stream(request: ChatRequest):
     try:
         model = MODEL_MAPPING.get(request.model, request.model)
         messages = [
@@ -130,7 +128,8 @@ def process_response_stream(request: ChatRequest):
                 yield "data: [DONE]\n\n"
             except Exception as e:
                 logger.error(f"Error in streaming response: {e}")
-                raise HTTPException(status_code=500, detail=str(e))
+                yield f"data: [ERROR] {str(e)}\n\n"
+
         return event_generator()
     except Exception as e:
         logger.error(f"Error processing streaming response: {e}")