引言

运用AI模型在获取我们想要的返回结果时,可能很多时候返回的都是非结构化的内容,还需要人为进行分类总结等。那么有没有一种方法可以让AI的返回能自动化转化成某种结构化的形式?答案是肯定的。

什么是Pydantic类?

Pydantic 是一个用于数据验证和设置的 Python 库。它允许您定义数据模型,并在运行时自动验证数据的类型和格式。Pydantic 常用于 FastAPI 等框架中,以确保传入的数据符合预期的结构和约束。那么在运用AI模型的时候,能不能运用Pydantic的思想对AI的输出进行改造呢?

实战演练:对AI输出进行改造

首先我们要设定我们要完成的内容,在此次的实战演练中,我们要完成"对一个流程进行改造优化"的任务。下面是演示代码:

from typing import Optional

from typing import Optional
from langchain_groq.chat_models import ChatGroq
from langchain_core.pydantic_v1 import BaseModel,Field



#Pydantic:
class Process(BaseModel):
    """优化改造现有流程"""

    setup: str = Field(description="请完成一个流程改造")
    critical_point: str = Field(description="流程改造的难点在哪里?")
    rating: Optional[int] = Field(default=None, description="对于这样的改造,你的满意度是多少,请打分(1到10分)")



# 使用LangChain模型+ Pydantic来约束和改造LLM的输出
llm = ChatGroq(api_key='gsk_xxx',
               model="mixtral-8x7b-32768",
               temperature=0,
               max_tokens=None,
               timeout=None,
               max_retries=2,
               )

structured_llm = llm.with_structured_output(Process)
ai_res = structured_llm.invoke("请改造下现有的软件开发流程,从现在的瀑布模型改成敏捷模型")
print(ai_res)

代码解析:

  • 第3行,要引入Pydantic的BaseModel类和Field类。
  • 第8到-13行,定义Process(基于BaseModel)类,在此类中可以自定义你想达到的目的。比如setup字段,critical_point字段(说出流程改造的难点在哪里?)
  • 第18-24行,定义langchain模型(还是用的ChatGrog大模型)。
  • 第26-28行,告诉AI要实现什么流程的优化以及显示AI输出结果。

可以看到脚本执行结果如下图所示:

AI全知道-如何从AI模型中获取结构化的返回数据?_软件测试

希望这篇文章能为您提供有价值的信息,帮助您更好地理解和利用Langchain。感谢您的阅读!


如果您对Langchain的使用有任何疑问或需要进一步的交流,欢迎在评论区留言,让我们共同探讨与进步。