steveliuatclass commited on
Commit
ac3c793
·
verified ·
1 Parent(s): 781fca1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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
- #再次加入當前prompt,避免gemini忘記當前prompt
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 輪對話, 使用者與gemini各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沒答案!請換個說法!"