Advertisement
Cabaca

Untitled

Aug 6th, 2024
175
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.77 KB | None | 0 0
  1. from langchain_core.output_parsers import JsonOutputParser
  2. from langchain_core.prompts import PromptTemplate
  3. from langchain_core.pydantic_v1 import BaseModel, Field
  4. from langchain_openai import ChatOpenAI
  5.  
  6. model = ChatOpenAI(temperature=0)
  7.  
  8. class Person(BaseModel):
  9.     name: str = Field(description="The name of the person")
  10.     lastname: str = Field(description="The lastname of the person")
  11.  
  12. prompt = PromptTemplate(
  13.     template="Extract information about a person.\n{format_instructions}\n{query}\n",
  14.     input_variables=["query"],
  15.     partial_variables={"format_instructions": parser.get_format_instructions()},
  16. )
  17.  
  18. chain = prompt | model | JsonOutputParser(pydantic_object=Person)
  19.  
  20. chain.invoke({"query": "There is a person called John Doe that lives in the city."})
  21.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement