alanchen1115 commited on
Commit
13d3707
·
verified ·
1 Parent(s): 17ddbda

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -10,8 +10,11 @@ from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSend
10
  client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
11
 
12
  # 設定生成文字的參數 + 角色扮演
 
 
13
  generation_config = genai.types.GenerateContentConfig(max_output_tokens=512, temperature=0.5, top_p=0.5,
14
- system_instruction="你是一位熱情開朗的機器人,每次回答問題之餘,還會講一個100字以內的笑話!") #角色扮演
 
15
 
16
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
17
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 
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
 
19
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
20
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])