Update main.py

main.py

@@ -21,9 +21,12 @@ for filename in files:
         pdf_content+= page.extract_text()
         
 # 設定生成文字的參數 + 角色扮演
+system_instruction = pdf_content+"\n"+
+                     "您是一位問答助手。請僅限使用以上提供的內容來回答問題。如果您不知道答案，請說您不知道，不要捏造答案。"
+thinking_config = genai.types.ThinkingConfig(thinking_budget=0) # thinking_budget = 0,  turn off thinking mode
 generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, temperature=0.1, top_p=0.2,
-                    system_instruction=pdf_content+"\n"+
-                    "您是一位問答助手。請僅限使用以上提供的內容來回答問題。如果您不知道答案，請說您不知道，不要捏造答案。")
+                                                      thinking_config=thinking_config,
+                                                      system_instruction=system_instruction)
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])