Update main.py
Browse files
main.py
CHANGED
|
@@ -10,9 +10,9 @@ from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSend
|
|
| 10 |
client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 11 |
|
| 12 |
# 設定生成文字的參數 + 角色扮演
|
| 13 |
-
system_instruction = "
|
| 14 |
-
thinking_config = genai.types.ThinkingConfig(thinking_budget=
|
| 15 |
-
generation_config = genai.types.GenerateContentConfig(max_output_tokens=512, temperature=0.
|
| 16 |
thinking_config=thinking_config,
|
| 17 |
system_instruction=system_instruction)
|
| 18 |
|
|
@@ -86,7 +86,9 @@ def handle_message(event):
|
|
| 86 |
elif working_status:
|
| 87 |
try:
|
| 88 |
user_id = event.source.user_id
|
|
|
|
| 89 |
chat = chat_sessions.get(user_id) or client.chats.create(model="gemini-2.5-flash", config=generation_config)
|
|
|
|
| 90 |
chat_sessions[user_id] = chat
|
| 91 |
# 取得使用者輸入的文字
|
| 92 |
user_input = event.message.text
|
|
|
|
| 10 |
client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 11 |
|
| 12 |
# 設定生成文字的參數 + 角色扮演
|
| 13 |
+
system_instruction = "你是一位全知全能的智者,每次回答問題之餘,每30分鐘還會講一個100字以內的充滿智慧啓發的笑話! "
|
| 14 |
+
thinking_config = genai.types.ThinkingConfig(thinking_budget=1) # thinking_budget = 0, turn off thinking mode
|
| 15 |
+
generation_config = genai.types.GenerateContentConfig(max_output_tokens=512, temperature=0.7, top_p=0.5,
|
| 16 |
thinking_config=thinking_config,
|
| 17 |
system_instruction=system_instruction)
|
| 18 |
|
|
|
|
| 86 |
elif working_status:
|
| 87 |
try:
|
| 88 |
user_id = event.source.user_id
|
| 89 |
+
# jms:每個user_id 開一個 chat
|
| 90 |
chat = chat_sessions.get(user_id) or client.chats.create(model="gemini-2.5-flash", config=generation_config)
|
| 91 |
+
# jms:不同的人有不同的記憶
|
| 92 |
chat_sessions[user_id] = chat
|
| 93 |
# 取得使用者輸入的文字
|
| 94 |
user_input = event.message.text
|