Google Gen AI Python SDK 使用教程
1. 项目介绍
Google Gen AI Python SDK 是由 Google 开发的一个开源 Python SDK,它为开发者提供了一个接口,用于将 Google 的生成模型集成到 Python 应用程序中。该 SDK 支持使用 Gemini Developer API 和 Vertex AI API。
2. 项目快速启动
安装
首先,需要安装 Google Gen AI Python SDK。可以通过 pip 命令进行安装:
pip install google-genai
创建客户端
接下来,创建一个客户端实例来使用 SDK。根据需要选择以下两种方式之一:
使用 Gemini Developer API
from google import genai
client = genai.Client(api_key='GEMINI_API_KEY')
使用 Vertex AI API
from google import genai
from google.genai import types
client = genai.Client(
vertexai=True,
project='your-project-id',
location='us-central1',
http_options=types.HttpOptions(api_version='v1')
)
确保替换 'GEMINI_API_KEY'
、'your-project-id'
和 'us-central1'
为实际的 API 密钥、项目 ID 和位置。
3. 应用案例和最佳实践
以下是一个使用 SDK 生成内容的示例:
生成文本内容
response = client.models.generate_content(
model='gemini-2.0-flash-001',
contents='为什么天空是蓝色的?'
)
print(response.text)
使用上传的文件
首先,下载文件到本地:
wget -q https://storage.googleapis.com/generativeai-downloads/data/a11.txt
然后,使用 SDK 上传文件并生成内容:
file = client.files.upload(file='a11.txt')
response = client.models.generate_content(
model='gemini-2.0-flash-001',
contents=['你能总结这个文件吗?', file]
)
print(response.text)
4. 典型生态项目
目前,Google Gen AI Python SDK 的生态系统包括多种模型和服务,开发者可以根据需要选择适合自己项目的模型和服务。例如,使用 Vertex AI API 可以访问到 Google Cloud 提供的先进的人工智能服务,而使用 Gemini Developer API 则可以访问到专门为生成任务设计的模型。
以上就是 Google Gen AI Python SDK 的基本使用教程。开发者可以根据具体的应用场景和需求,进一步探索和利用 SDK 提供的功能。