Update main.py

main.py

@@ -199,23 +199,31 @@ async def chat_completions(
 
 
 
-    async def generate():
-        async with httpx.AsyncClient() as client:
-            try:
-                async with client.stream('POST', 'https://sider.ai/api/v2/completion/text', headers=headers, json=json_data, timeout=120.0) as response:
-                    response.raise_for_status()
-                    async for line in response.aiter_lines():
-                        if line and ("[DONE]" not in line):
-                            content = json.loads(line[5:])["data"]
-                            yield f"data: {json.dumps(create_chat_completion_data(content.get('text',''), request.model))}\n\n"
+async def generate():
+    async with httpx.AsyncClient() as client:
+        try:
+            async with client.stream('POST', 'https://sider.ai/api/v2/completion/text', headers=headers, json=json_data, timeout=120.0) as response:
+                response.raise_for_status()
+                async for line in response.aiter_lines():
+                    if line and ("[DONE]" not in line):
+                        data = json.loads(line[5:])
+                        content = data.get("data")
+                        
+                        # Check if content is None or does not contain the text key
+                        if content and 'text' in content:
+                            yield f"data: {json.dumps(create_chat_completion_data(content['text'], request.model))}\n\n"
+                        else:
+                            logger.error(f"Unexpected content format: {data}")
+                            yield f"data: {json.dumps(create_chat_completion_data('Error: Unexpected content format', request.model, 'stop'))}\n\n"
                     yield f"data: {json.dumps(create_chat_completion_data('', request.model, 'stop'))}\n\n"
                     yield "data: [DONE]\n\n"
-            except httpx.HTTPStatusError as e:
-                logger.error(f"HTTP error occurred: {e}")
-                raise HTTPException(status_code=e.response.status_code, detail=str(e))
-            except httpx.RequestError as e:
-                logger.error(f"An error occurred while requesting: {e}")
-                raise HTTPException(status_code=500, detail=str(e))
+        except httpx.HTTPStatusError as e:
+            logger.error(f"HTTP error occurred: {e}")
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except httpx.RequestError as e:
+            logger.error(f"An error occurred while requesting: {e}")
+            raise HTTPException(status_code=500, detail=str(e))
+
 
     if request.stream:
         logger.info("Streaming response")