[如何在开发中高效使用Google Generative AI和Vertex AI]

stjklkjhgffxw

于 2024-10-04 07:11:43 发布

阅读量197

点赞数 3

文章标签：人工智能 easyui 前端 python

本文链接：https://blog.csdn.net/stjklkjhgffxw/article/details/142699189

版权

引言

随着人工智能技术的快速发展，Google提供了多种AI服务，使开发者能够更轻松地集成AI功能。本篇文章将探讨如何使用Google Generative AI和Vertex AI，并提供具体的代码示例，帮助你快速上手。

主要内容

什么是Google Generative AI和Vertex AI？

Google Generative AI提供了自然语言处理和生成模型，如Gemini模型，可用于文本和图像生成。Vertex AI是Google Cloud的一个平台，提供了许多机器学习模型的托管和管理功能，更适合企业级应用。

如何安装和配置

安装所需的Python包

您需要安装以下Python包来访问这些服务：

pip install -U langchain-google-genai
pip install langchain-google-vertexai

配置API密钥

确保将API密钥导出到环境变量：

export GOOGLE_API_KEY=your-api-key

使用Google Generative AI

以下是一个使用Gemini模型生成文本的示例：

from langchain_google_genai import ChatGoogleGenerativeAI

llm = ChatGoogleGenerativeAI(model="gemini-pro")
response = llm.invoke("Sing a ballad of LangChain.")
print(response)

使用Vertex AI

使用Vertex AI时，您可以选择使用更强大的商业支持和更高的速率限制。

from langchain_google_vertexai import ChatVertexAI

llm = ChatVertexAI(model="chat-bison")
response = llm.invoke("Explain LangChain in simple terms.")
print(response)

代码示例

以下是一个完整的Google Generative AI使用示例，包含图像输入功能：

from langchain_core.messages import HumanMessage
from langchain_google_genai import ChatGoogleGenerativeAI

llm = ChatGoogleGenerativeAI(model="gemini-pro-vision")

message = HumanMessage(
    content=[
        {"type": "text", "text": "What's in this image?"},
        {"type": "image_url", "image_url": "https://picsum.photos/seed/picsum/200/300"},
    ]
)
response = llm.invoke([message])
print(response)