AI AI Club
← 返回文章列表

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 消息数组!

联系我们

会员开通、订单问题或合作咨询,欢迎联系。

QQ1192214187
邮箱winky20000401@gmail.com