结构化输出

有时候我们希望输出的内容不是文本，而是像 json 那样结构化的数据。

  1. from langchain.output_parsers import StructuredOutputParser, ResponseSchema
  2. from langchain.prompts import PromptTemplate
  3. from langchain.llms import OpenAI
  5. llm = OpenAI(model_name="text-davinci-003")
  7. # 告诉他我们生成的内容需要哪些字段，每个字段类型式啥
  8. response_schemas = [
  9.     ResponseSchema(name="bad_string", description="This a poorly formatted user input string"),
  10.     ResponseSchema(name="good_string", description="This is your response, a reformatted response")
  11. ]
  13. # 初始化解析器
  14. output_parser = StructuredOutputParser.from_response_schemas(response_schemas)
  16. # 生成的格式提示符
  17. # {
  18. #    "bad_string": string  // This a poorly formatted user input string
  19. #    "good_string": string  // This is your response, a reformatted response
  20. #}
  21. format_instructions = output_parser.get_format_instructions()
  23. template = """
  24. You will be given a poorly formatted string from a user.
  25. Reformat it and make sure all the words are spelled correctly
  27. {format_instructions}
  29. % USER INPUT:
  30. {user_input}
  32. YOUR RESPONSE:
  33. """
  35. # 将我们的格式描述嵌入到 prompt 中去，告诉 llm 我们需要他输出什么样格式的内容
  36. prompt = PromptTemplate(
  37.     input_variables=["user_input"],
  38.     partial_variables={"format_instructions": format_instructions},
  39.     template=template
  40. )
  42. promptValue = prompt.format(user_input="welcom to califonya!")
  43. llm_output = llm(promptValue)
  45. # 使用解析器进行解析生成的内容
  46. output_parser.parse(llm_output)

image-20230406000017276