Files
wehub-resource-sync 97e91a83f3
Ruff / Ruff (push) Has been cancelled
Test / Core Tests (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.10) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.11) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.12) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.13) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.9) (push) Has been cancelled
Test / Full Coverage (Python 3.11) (push) Has been cancelled
Test / Core Provider Tests (OpenAI) (push) Has been cancelled
Test / Core Provider Tests (Anthropic) (push) Has been cancelled
Test / Core Provider Tests (Google) (push) Has been cancelled
Test / Core Provider Tests (Other) (push) Has been cancelled
Test / Anthropic Tests (push) Has been cancelled
Test / Gemini Tests (push) Has been cancelled
Test / Google GenAI Tests (push) Has been cancelled
Test / Vertex AI Tests (push) Has been cancelled
Test / OpenAI Tests (push) Has been cancelled
Test / Writer Tests (push) Has been cancelled
Test / Auto Client Tests (push) Has been cancelled
ty / type-check (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:36:38 +08:00

210 lines
6.1 KiB
Markdown

---
title: Using Pydantic Models for Structured Outputs
description: Learn how to define LLM output schemas with Pydantic models.
---
# Response Model
Define LLM output schemas using `pydantic.BaseModel`. For more details, see the [Pydantic documentation](https://docs.pydantic.dev/latest/concepts/models/).
After defining a Pydantic model, use it as the `response_model` in your client `create` calls. The `response_model` parameter:
- Defines the schema and prompts for the language model
- Validates the response from the API
- Returns a Pydantic model instance
## Prompting
Use docstrings and field annotations to define the prompt for generating responses.
```python
from pydantic import BaseModel, Field
import instructor
class User(BaseModel):
"""
This is the prompt that will be used to generate the response.
Any instructions here will be passed to the language model.
"""
name: str = Field(description="The name of the user.")
age: int = Field(description="The age of the user.")
client = instructor.from_provider("openai/gpt-4o-mini")
user = client.create(
response_model=User,
messages=[{"role": "user", "content": "Extract: John is 30 years old"}],
)
```
Docstrings, types, and field annotations are used to generate the prompt. The `create` method uses this prompt to generate the response.
## Optional Values
Use `Optional` and `default` to make fields optional when sent to the language model.
```python
from pydantic import BaseModel, Field
from typing import Optional
import instructor
class User(BaseModel):
name: str = Field(description="The name of the user.")
age: int = Field(description="The age of the user.")
email: Optional[str] = Field(description="The email of the user.", default=None)
client = instructor.from_provider("openai/gpt-4o-mini")
user = client.create(
response_model=User,
messages=[{"role": "user", "content": "Extract: John is 30 years old"}],
)
```
Fields can also be omitted from the schema sent to the language model using Pydantic's `SkipJsonSchema` annotation. See [Fields](fields.md#omitting-fields-from-schema-sent-to-the-language-model) for details.
## Dynamic Model Creation
Create models at runtime using Pydantic's `create_model` function:
```python
from pydantic import BaseModel, create_model
class FooModel(BaseModel):
foo: str
bar: int = 123
BarModel = create_model(
'BarModel',
apple=(str, 'russet'),
banana=(str, 'yellow'),
__base__=FooModel,
)
print(BarModel)
#> <class '__main__.BarModel'>
print(BarModel.model_fields.keys())
#> dict_keys(['foo', 'bar', 'apple', 'banana'])
```
??? notes "When would I use this?"
Consider a situation where the model is dynamically defined, based on some configuration or database. For example, we could have a database table that stores the properties of a model for
some model name or id. We could then query the database for the properties of the model and use that to create the model.
```sql
SELECT property_name, property_type, description
FROM prompt
WHERE model_name = {model_name}
```
We can then use this information to create the model.
```python
from pydantic import BaseModel, create_model, Field
from typing import List
types = {
'string': str,
'integer': int,
'boolean': bool,
'number': float,
'List[str]': List[str],
}
# Mocked cursor.fetchall()
cursor = [
('name', 'string', 'The name of the user.'),
('age', 'integer', 'The age of the user.'),
('email', 'string', 'The email of the user.'),
]
BarModel = create_model(
'User',
**{
property_name: (types[property_type], Field(description=description))
for property_name, property_type, description in cursor
},
__base__=BaseModel,
)
print(BarModel.model_json_schema())
"""
{
'properties': {
'name': {
'description': 'The name of the user.',
'title': 'Name',
'type': 'string',
},
'age': {
'description': 'The age of the user.',
'title': 'Age',
'type': 'integer',
},
'email': {
'description': 'The email of the user.',
'title': 'Email',
'type': 'string',
},
},
'required': ['name', 'age', 'email'],
'title': 'User',
'type': 'object',
}
"""
```
This would be useful when different users have different descriptions for the same model. We can use the same model but have different prompts for each user.
## Adding Behavior
Add methods to Pydantic models like any Python class. This lets you add custom logic to your models.
```python
from pydantic import BaseModel
from typing import Literal
import instructor
client = instructor.from_provider("openai/gpt-4.1-mini")
class SearchQuery(BaseModel):
query: str
query_type: Literal["web", "image", "video"]
def execute(self):
print(f"Searching for {self.query} of type {self.query_type}")
#> Searching for cat of type image
return "Results for cat"
query = client.create(
model="gpt-4.1-mini",
messages=[{"role": "user", "content": "Search for a picture of a cat"}],
response_model=SearchQuery,
)
results = query.execute()
print(results)
#> Results for cat
```
Now we can call `execute` on our model instance after extracting it from a language model. If you want to see more examples of this checkout our post on [RAG is more than embeddings](../blog/posts/rag-and-beyond.md)
## See Also
- [Response Models Tutorial](../learning/getting_started/response_models.md) - Step-by-step guide to creating response models
- [Simple Object Extraction](../learning/patterns/simple_object.md) - Basic extraction patterns
- [Nested Structures](../learning/patterns/nested_structure.md) - Complex hierarchical models
- [Optional Fields](../learning/patterns/optional_fields.md) - Working with optional data
- [Types](./types.md) - Working with different data types
- [Fields](./fields.md) - Advanced field configuration