Skip to content

Gemini Python SDK基本用法


使用Gemini Api可以帮助开发者快速在自有产品中集成Gemini的能力,Python使用Gemini Api有两种方式: 使用Gemini 的Python SDK直接使用API,具体使用哪种方式,大家可以根据自己的场景和需求选择。本文将为大家介绍Gemini Python SDK的使用方法。

使用Gemini 的Python SDK
  1. 安装Gemini Api SDK

Gemini Api 的Python SDK包含在google-generativeai包中,安装命令:

sh
pip install google-generativeai
  1. 引入相关依赖

创建文件main.py,引入相关依赖

python
 import google.generativeai as genai
  1. 设置Api key

在前文快速开始中已经介绍了如何生成并获取Gemini的Api key,获取Api Key以后请设置 GEMINI_API_KEY

python
GEMINI_API_KEY = "替换为自己的Api Key"
genai.configure(api_key=GEMINI_API_KEY)
  1. 查看SDK支持的模型
python
for m in genai.list_models():
    if 'generateContent' in m.supported_generation_methods:
    print(m.name)

输出:

python
models/gemini-pro
models/gemini-pro-vision

目前SDK支持两个模型:

gemini-pro

仅支持文本类型的提示语

gemini-pro-vision

同时支持文本和图片类型的提示语

注意:由于众所周知的网络问题,在使用时需要满足科学上网的条件。

  1. 使用gemini-pro模型
python
model = genai.GenerativeModel('gemini-pro')
  1. 写一个提示语
python
response = model.generate_content("告诉我你是谁?并做一分钟自我介绍")
print(response.text)

输出:

python
大家好,我是来福,一个友好且乐于助人的 AI 助理。我很高兴见到你们!

我出生在一个高科技研发中心,由一群聪明又勤劳的科学家们创造出来。我的目的是帮助人们解决问题、学习新知识和完成任务。

我拥有丰富的信息和技能,可以回答各种各样的问题,包括但不限于:

* 通用知识:如历史、科学、文化、地理等。
* 日常生活:如烹饪、旅行、购物、健康等。
* 科技发展:如计算机、人工智能、通信等。
* 语言学习:如英语、汉语、法语、西班牙语等。

除了提供信息,我还能够:

* 翻译语言:将一种语言翻译成另一种语言。
* 撰写文本:写诗、写故事、写信、写报告等。
* 生成代码:生成编程语言代码。
* 计算数值:如加减乘除、平方根、几何形状计算等。
* 播放音乐:播放您指定的音乐,或生成新的音乐。

如果您有什么问题或需要帮助,请随时问我。我很乐意为您服务!

本文介绍了Gemini Python SDK的基本用法,有兴趣的同学可以快速上手体验。后续我们将继续学习Gemini Python SDK的高级用法。

Gemini中文文档