JmsC commited on
Commit
03b9804
·
verified ·
1 Parent(s): 95abd55

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -3
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 = "你是一位全知全能的智者,每次回答問題之餘,每個小時還會講一個100字以內的充滿智慧啓發的笑話! "
14
- thinking_config = genai.types.ThinkingConfig(thinking_budget=0) # thinking_budget = 0, turn off thinking mode
15
- generation_config = genai.types.GenerateContentConfig(max_output_tokens=512, temperature=0.5, top_p=0.5,
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