用 Python 玩转 Google Gemini 对话

Google Gemini

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

免费申请 Gemini API

前往这个网页^[1]免费申请 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)

总结

近日，我还将 Gemini 集成到了 Zotero，后面有时间会介绍下。