Added examples

Author: Diophontine

src/main.py

@@ -6,14 +6,21 @@
 from pydantic import BaseModel, Field
 
 
-class Furniture(BaseModel):
-    type: str = Field(description="the type of furniture")
-    style: str = Field(description="style of the furniture")
-    colour: str = Field(description="colour")
+class Person(BaseModel):
+    first_name: str = Field(description="first name")
+    last_name: str = Field(description="last name")
+    dob: str = Field(description="date of birth")
 
-furniture_request = "I'd like a blue mid century chair"
 
-parser = PydanticOutputParser(pydantic_object=Furniture)
+class PeopleList(BaseModel):
+    people: list[Person] = Field(description="A list of people")
+
+
+model = ChatOpenAI(model="gpt-4")
+people_data = model.predict(
+    "Generate a list of 10 fake peoples information. Only return the list. Each person should have a first name, last name and date of birth.")
+
+parser = PydanticOutputParser(pydantic_object=PeopleList)
 
 prompt = PromptTemplate(
     template="Answer the user query.\n{format_instructions}\n{query}\n",
@@ -22,10 +29,10 @@ class Furniture(BaseModel):
         "format_instructions": parser.get_format_instructions()},
 )
 
-_input = prompt.format_prompt(query=furniture_request)
+_input = prompt.format_prompt(query=people_data)
 
 model = ChatOpenAI()
 output = model.predict(_input.to_string())
 
 parsed = parser.parse(output)
-print(parsed.colour)
+print(parsed)