Jupyter调用Google Gemini AI遇AttributeError问题求助

问题背景 #

在Jupyter中测试Google Gemini AI模型时，运行以下代码：

import google.generativeai as genai
import os

# genai.configure(api_key="API Key")

model = genai.GenerativeModel('gemini-pro')
response = model.generate_content('Please summarise this document: ...')

print(response.text)

出现错误：

AttributeError
Traceback (most recent call last)
Input In [2], in <cell line: 6>()
      2 import os
      4 # genai.configure(api_key=os.environ['API_KEY'])
----> 6 model = genai.GenerativeModel('gemini-pro')
      7 response = model.generate_content('Please summarise this document: ...')
      9 print(response.text)

AttributeError: module 'google.generativeai' has no attribute 'GenerativeModel'

已尝试重新安装google-generativeai包、检查源码，当前环境：Python 3.11.7，google-generativeai 0.5.1。

解决方案 #

方案1：升级到最新版本的google-generativeai #

GenerativeModel类在0.5.1版本中未直接暴露在根模块下，升级到最新版本后即可使用官方推荐的写法。执行以下命令升级：

python3.11 -m pip install --upgrade google-generativeai

升级后，取消代码中genai.configure(api_key="API Key")的注释，填入你的Google Gemini API密钥，原代码即可正常运行。

方案2：适配0.5.1版本的写法 #

若暂时无法升级，可调整导入方式，直接从子模块引入GenerativeModel：

import google.generativeai as genai
from google.generativeai.generative_models import GenerativeModel
import os

genai.configure(api_key="你的API密钥")

model = GenerativeModel('gemini-pro')
response = model.generate_content('Please summarise this document: ...')

print(response.text)

额外检查：确认Jupyter环境一致性 #

有时候Jupyter使用的Python环境与安装包的环境不一致，导致安装的包无法被加载。在Jupyter中执行以下代码查看当前环境路径：

import sys
print(sys.executable)

使用输出的路径执行pip命令，确保包安装在Jupyter使用的环境中，例如：

/usr/bin/python3.11 -m pip install --upgrade google-generativeai

内容的提问来源于stack exchange，提问作者Learner_77