use session_id

src/talk_to_your_manual/use_aliyun.py

@@ -1,5 +1,5 @@
 from dashscope import Application
-from fastapi import APIRouter
+from fastapi import APIRouter, status
 from fastapi.responses import JSONResponse
 import os
 
@@ -7,6 +7,8 @@ router = APIRouter()
 API_KEY = os.environ.get("API_KEY")
 APP_ID = os.environ.get("APP_ID")
 
+session_id = None
+
 
 @router.get("/aliyun")
 async def call_aliyun(prompt: str):
@@ -14,13 +16,22 @@ async def call_aliyun(prompt: str):
         api_key=API_KEY,
         app_id=APP_ID,
         prompt=prompt,
+        session_id=session_id,
     )
-    return JSONResponse(
-        status_code=response.status_code,
-        content={
-            "request_id": response.request_id,
-            "message": response.message,
-            "output": response.output,
-            "usage": response.usage,
-        },
-    )
+    if response.status_code != status.HTTP_200_OK:
+        session_id = response.output.session_id
+        return JSONResponse(
+            status_code=status.HTTP_200_OK,
+            content={
+                "request_id": response.request_id,
+                "answer": response.output.text,
+            },
+        )
+    else:
+        return JSONResponse(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            content={
+                "code": response.code,
+                "message": response.message,
+            },
+        )