Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -78,6 +78,7 @@ def handle_message(event):
|
|
78 |
event.reply_token,
|
79 |
TextSendMessage(text="Event type error:[No message or the message does not contain text]")
|
80 |
)
|
|
|
81 |
|
82 |
# 檢查使用者是否輸入 "再見"
|
83 |
elif event.message.text == "再見":
|
@@ -105,7 +106,7 @@ def handle_message(event):
|
|
105 |
full_prompt = ""
|
106 |
for message in conversation_history[user_id]:
|
107 |
full_prompt += message["role"] + ": " + message["parts"][0] + "\n"
|
108 |
-
|
109 |
full_prompt += "user: " + prompt
|
110 |
# 使用 Gemini 模型生成文字
|
111 |
completion = model.generate_content(full_prompt, generation_config=generation_config)
|
@@ -115,7 +116,7 @@ def handle_message(event):
|
|
115 |
out = completion.parts[0].text
|
116 |
# 將 Gemini 的回覆加入歷史
|
117 |
conversation_history[user_id].append({"role": "model", "parts": [out]})
|
118 |
-
# 只保留最近 7 輪對話, 使用者與
|
119 |
conversation_history[user_id] = conversation_history[user_id][-14:]
|
120 |
else:
|
121 |
# 回覆 "Gemini沒答案!請換個說法!"
|
|
|
78 |
event.reply_token,
|
79 |
TextSendMessage(text="Event type error:[No message or the message does not contain text]")
|
80 |
)
|
81 |
+
return
|
82 |
|
83 |
# 檢查使用者是否輸入 "再見"
|
84 |
elif event.message.text == "再見":
|
|
|
106 |
full_prompt = ""
|
107 |
for message in conversation_history[user_id]:
|
108 |
full_prompt += message["role"] + ": " + message["parts"][0] + "\n"
|
109 |
+
# 再次加入最新 prompt,避免 Gemini 忘記最新 prompt
|
110 |
full_prompt += "user: " + prompt
|
111 |
# 使用 Gemini 模型生成文字
|
112 |
completion = model.generate_content(full_prompt, generation_config=generation_config)
|
|
|
116 |
out = completion.parts[0].text
|
117 |
# 將 Gemini 的回覆加入歷史
|
118 |
conversation_history[user_id].append({"role": "model", "parts": [out]})
|
119 |
+
# 只保留最近 7 輪對話, 使用者與 Gemini 各 7 筆
|
120 |
conversation_history[user_id] = conversation_history[user_id][-14:]
|
121 |
else:
|
122 |
# 回覆 "Gemini沒答案!請換個說法!"
|