Python 官方 SDK 现代化调用范例:从环境安装到多轮对话历史管理
用 Python 写一个带记忆的聊天机器人!基于 google-generativeai 官方库编写极简实战代码,图解 chat = model.start_chat() 历史管理技巧。
一、环境安装与初始化
在终端中执行:pip install google-generativeai
二、带完整历史记忆的精简 Python 代码
import os
import google.generativeai as genai
genai.configure(api_key=os.environ["GEMINI_API_KEY"])
model = genai.GenerativeModel("gemini-1.5-flash")
chat = model.start_chat(history=[])
while True:
user_input = input("User: ")
if user_input.lower() == "exit":
break
response = chat.send_message(user_input)
print(f"Gemini: {response.text}")
通过 start_chat 对象,所有往来提问历史会在后台自动维护,开发者无需手动拼接复杂的 JSON 消息数组!