Update main.py

main.py

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
 load_dotenv()
 app = FastAPI()
 BASE_URL = "https://aichatonlineorg.erweima.ai/aichatonline"
-APP_SECRET = os.getenv("APP_SECRET","666")
+APP_SECRET = os.getenv("APP_SECRET","786")
 ACCESS_TOKEN = os.getenv("SD_ACCESS_TOKEN","")
 headers = {
     'accept': '*/*',
@@ -199,31 +199,23 @@ 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):
-                        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"
+    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"
                     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")
@@ -256,4 +248,4 @@ async def generate():
 
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)