Feb 16 (edited) • 💡 Help
Collecting required fields using Pydantic model
I am developing a chatbot that must collect all required fields before proceeding to the next step. For this example, here is my Pydantic model:
class PersonalInfo(BaseModel):
first_name: str = Field(
...,
description="User's first name"
)
last_name: str = Field(
...,
description="User's last name"
)
The chatbot prompts the user to enter their first and last name. Once the input is received, I want to check for missing fields. If a field is missing (e.g., the user doesn’t enter their last name), the bot should respond with a specific prompt, such as:"Please enter your last name."
This is my current approach, and I’d like feedback on whether there’s a more efficient way to handle it.
from pydantic import BaseModel, Field
from typing import Optional
from openai import OpenAI
import json
class PersonalInfo(BaseModel):
first_name: str = Field(
...,
description="User's first name"
)
last_name: str = Field(
...,
description="User's last name"
)
def collect_personal_info():
client = OpenAI()
personal_info = {}
print("Hello, please enter your first and last name:")
user_input = input().strip()
# Try to parse first and last name from initial input
msg = [
{
"role": "system",
"content": """You are a helpful assistant collecting user's personal information.
Extract first name and last name from user input. Return them in a structured format."""
},
{
"role": "user",
"content": user_input
}
]
completion = do_completion(client, msg)
personal_info = completion.choices[0].message.parsed
# Check each field in the Pydantic model
while True:
missing_fields = False
for field_name, field in PersonalInfo.model_fields.items():
if not getattr(personal_info, field_name):
missing_fields = True
print(f"\nI am missing your {field_name.replace('_', ' ')}. Please enter it:")
user_input = input().strip()
msg.append({"role": "user", "content": f"My {field_name.replace('_', ' ')} is {user_input}"})
completion = do_completion(client, msg)
personal_info = completion.choices[0].message.parsed
if not missing_fields:
break
print("\nThank you! I have all your information. Let's move to the next step!")
return personal_info
def do_completion(client: OpenAI, messages):
completion = client.beta.chat.completions.parse(
model="gpt-4o-2024-08-06",
messages=messages,
response_format=PersonalInfo
)
return completion
def main():
personal_info = collect_personal_info()
print(f"\nCollected Information:")
print(f"First Name: {personal_info.first_name}")
print(f"Last Name: {personal_info.last_name}")
if __name__ == "__main__":
main()
3
2 comments
Collecting required fields using Pydantic model
skool.com/data-alchemy
Your Community to Master the Fundamentals of Working with Data and AI — by Datalumina®
Leaderboard (30-day)
1
+203
2
+75
3
+33
4
+32
5
+29
Powered by