Python教程：用 Python 玩转 Google Gemini 对话

今天以 Mac 为例，介绍下如何用 Python 实现 Google Gemini 对话。

免费申请 Gemini API
安装 Gemini Python 库
在 Mac 终端运行下方代码，即可下载 google-generativeai Python 库。

pip3 install google-generativeai
不过通过这行代码下载速度会很慢，因此推荐通过清华源下载：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple google-generativeai
接下来，开始玩转。

Python 玩转 Gemini
列出模型
通过下面的代码，可以获取你的 API 所支持的模型，比如 Gemini Pro、Gemini Pro Vision。

# setup
import google.generativeai as genai
 
genai.configure(api_key='这里填你的 key')  # 填入自己的 api_key
 
# 查询模型
for m in genai.list_models():
    print(m.name)
    print(m.supported_generation_methods)
内容生成
通过下方代码，就可以向 Gemini 提问了。

# setup
import google.generativeai as genai
 
genai.configure(api_key='这里填你的 key')  # 填入自己的 api_key
 
model = genai.GenerativeModel('gemini-pro')
response = model.generate_content("讲一个故事")
print(response.text)
流式输出
为了实现更好的交互体验，推荐大家使用流式输出模式下的 Gemini，代码如下。

# setup
import google.generativeai as genai
 
genai.configure(api_key='这里填你的 key')  # 填入自己的 api_key
 
model = genai.GenerativeModel('gemini-pro')
response = model.generate_content("说一个笑话听听", stream=True)
 
for chunk in response:
    print(chunk.text)

    print("_"*80)

以上就是“Python教程：用 Python 玩转 Google Gemini 对话”的详细内容，想要了解更多IT圈内资讯欢迎持续关注编程学习网。