alanchen1115 commited on
Commit
14e4374
·
verified ·
1 Parent(s): 94e0787

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -16,7 +16,7 @@ generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, te
16
  # Retrieve the PDF
17
  files = glob.glob('docs/*.pdf')
18
  print(files)
19
- files_content = [genai.types.Part.from_bytes(data=filepath.read_bytes(), mime_type='application/pdf') for filepath in files]
20
  print(files_content)
21
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
22
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 
16
  # Retrieve the PDF
17
  files = glob.glob('docs/*.pdf')
18
  print(files)
19
+ files_content = [genai.types.Part.from_bytes(data=pathlib.Path(filepath).read_bytes(), mime_type='application/pdf') for filepath in files]
20
  print(files_content)
21
  # 設定 Line Bot 的 API 金鑰和秘密金鑰
22
  line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])