Update main (1).py
Browse files- main (1).py +2 -2
main (1).py
CHANGED
|
@@ -15,7 +15,7 @@ genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
|
|
| 15 |
generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
|
| 16 |
|
| 17 |
# 使用 Gemini-1.5-flash 模型
|
| 18 |
-
model = genai.GenerativeModel('gemini-1.5-flash', system_instruction="
|
| 19 |
|
| 20 |
# 設定 Line Bot 的 API 金鑰和秘密金鑰
|
| 21 |
line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
|
|
@@ -88,7 +88,7 @@ def handle_message(event):
|
|
| 88 |
# 取得使用者輸入的文字
|
| 89 |
prompt = event.message.text
|
| 90 |
# 使用 Gemini 模型生成文字
|
| 91 |
-
completion = model.
|
| 92 |
# 檢查生成結果是否為空
|
| 93 |
if (completion.parts[0].text != None):
|
| 94 |
# 取得生成結果
|
|
|
|
| 15 |
generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
|
| 16 |
|
| 17 |
# 使用 Gemini-1.5-flash 模型
|
| 18 |
+
model = genai.GenerativeModel('gemini-1.5-flash', system_instruction="j請用正常人的語氣") # 或是使用 "你是博通古今的萬應機器人!"
|
| 19 |
|
| 20 |
# 設定 Line Bot 的 API 金鑰和秘密金鑰
|
| 21 |
line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
|
|
|
|
| 88 |
# 取得使用者輸入的文字
|
| 89 |
prompt = event.message.text
|
| 90 |
# 使用 Gemini 模型生成文字
|
| 91 |
+
completion = model.start_chat(prompt, generation_config=generation_config)
|
| 92 |
# 檢查生成結果是否為空
|
| 93 |
if (completion.parts[0].text != None):
|
| 94 |
# 取得生成結果
|