添加内存中的多轮对话支持

app.py

@@ -3,13 +3,14 @@ import os
 import openai
 
 # 请记得要把 api 的 key 放到 settings 下面的 Repository Secrets 里。
-# 目前有个特别奇怪的问题： duplicate 的 key 如果和原来的 key 重名，build 就会失败。不知是否是今天正在 migrating 的原因。
-# 作为 workaround，请对 key 使用一个不同的名字，并且记得修改下面这行代码中的 key 的名字。
-openai.api_key = os.getenv("openai_key")
+openai.api_key = os.getenv("key")
+
 
 # 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
 prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
 
+history = {}
+
 # 修改本函数，来实现你自己的 chatbot
 # p: 对机器人说话的内容  
 # qid: 当前消息的唯一标识。例如 `'bxqid-cManAtRMszw...'`。由平台生成并传递给机器人，以便机器人区分单个问题（写日志、追踪调试、异步回调等）。同步调用可忽略。  
@@ -17,14 +18,30 @@ prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气
 # 返回值：[type, content]
 # 详见 https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md
 def chat(p, qid, uid):
-    return ["text", callapi(p)]
+    global history
+    if uid in history:
+        msgs = history[uid]
+    else:
+        msgs = []
+    response = callapi(p, msgs)
+    history[uid] = msgs + [[p, response]]
+    return ["text", response]
+
 
-def callapi(p):
+def callapi(p, msgs):
+    if (len(msgs) > 8):  #hard-code 8 回合对话。如果需要更精准的，应该计算 token 数
+        msgs = msgs[-8:]
+        
+    data = [{"role":"system", "content":prompt}]
+    for m in msgs:
+        data = data + [
+            {"role":"user", "content":m[0]},
+            {"role":"assistant", "content":m[1]}
+        ]
+    data = data + [{"role":"user", "content":p}]
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        messages= [{"role":"system", "content":prompt},
-                   {"role":"user", "content":p}
-                   ]
+        messages= data
     )
     print(response)
     response = response["choices"][0]["message"]["content"]