首页 前端知识 【langchain学习】使用JsonOutputParser让大模型生成结构化JSON数据

【langchain学习】使用JsonOutputParser让大模型生成结构化JSON数据

2024-09-06 00:09:45 前端知识 前端哥 883 797 我要收藏

使用Langchain处理结构化数据,以JsonOutputParser为例。以下是具体步骤和代码示例:

  1. 导入所需库

    from config import llm
    from langchain_core.output_parsers import JsonOutputParser
    from langchain_core.prompts import PromptTemplate
    from langchain_core.pydantic_v1 import BaseModel, Field
    
  2. 定义数据模型

    class People(BaseModel):
        name: str = Field(description="人物的名称")
        age: str = Field(description="人物的年龄")
        country: str = Field(description="人物的国籍")
    
  3. 设置JsonOutputParser

    parser = JsonOutputParser(pydantic_object=People)
    
  4. 设置提示模板

    prompt = PromptTemplate(
        template="Answer the user query.\n{format_instructions}\n{query}\n",
        input_variables=["query"],
        partial_variables={"format_instructions": parser.get_format_instructions()},
    )
    
  5. 定义链操作

    chain = prompt | llm | parser
    
  6. 调用链操作并处理查询

    query = "姚明"
    result = chain.invoke({"query": query})
    print(result)
    
  7. 示例输出

{'name': '姚明', 'age': '40', 'country': '中国'}

Process finished with exit code 0
转载请注明出处或者链接地址:https://www.qianduange.cn//article/17765.html
标签
langchain
评论
发布的文章

jquery video视频轮播播放

2024-09-27 09:09:58

大家推荐的文章
会员中心 联系我 留言建议 回顶部
复制成功!