polish

src/talk_to_your_manual/use_aliyun.py

@@ -1,5 +1,6 @@
 from dashscope import Application
 from fastapi import APIRouter, status
+from http import HTTPStatus
 from fastapi.responses import JSONResponse
 import os
 
@@ -10,23 +11,28 @@ APP_ID = os.environ.get("APP_ID")
 session_id = None
 
 init_prompt = """
-  You are a helpful assistant. Please answer some questions about the MatchN game.
+  You are a helpful assistant. Please answer some 
+  questions about the MatchN game. You should only
+  user the knowledges from the knowledge base.
 """
 
 
 def init_session():
     print("[init_session] start")
+    global session_id
     response = Application.call(
         api_key=API_KEY,
         app_id=APP_ID,
         prompt=init_prompt,
     )
-    if response.status_code != status.HTTP_200_OK:
+    if response.status_code == HTTPStatus.OK:
         session_id = response.output.session_id
         print(f"[init_session] session_id: {session_id}")
         print(f"[init_session] answer: {response.output.text}")
     else:
+        print(response)
         print("[init_session] init session failed")
+        print(f"[init_session] code: {response.code}")
         print(f"[init_session] detail: {response.message}")
     print("[init_session] end\n\n")
 
@@ -36,18 +42,21 @@ init_session()
 
 @router.get("/aliyun")
 async def call_aliyun(prompt: str):
+    global session_id
     response = Application.call(
         api_key=API_KEY,
         app_id=APP_ID,
         prompt=prompt,
         session_id=session_id,
     )
-    if response.status_code != status.HTTP_200_OK:
+    if response.status_code == HTTPStatus.OK:
+        last_session_id = session_id
         session_id = response.output.session_id
         return JSONResponse(
             status_code=status.HTTP_200_OK,
             content={
-                "request_id": response.request_id,
+                "session_id": response.output.session_id,
+                "last_session_id": last_session_id,
                 "answer": response.output.text,
             },
         )