phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #5079 from BerriAI/litellm_add_pydantic_model_support

Browse source 
feat(utils.py): support passing response_format as pydantic model
This commit is contained in:
Krish Dholakia 2024-08-07 14:43:05 -07:00 • committed by GitHub
commit 2e434d56e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 353 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -31,6 +31,7 @@ from typing import (
Literal,
Mapping,
Optional,
Type,
Union,
)
@ -608,7 +609,7 @@ def completion(
logit_bias: Optional[dict] = None,
user: Optional[str] = None,
# openai v1.0+ new params
response_format: Optional[dict] = None,
response_format: Optional[Union[dict, Type[BaseModel]]] = None,
seed: Optional[int] = None,
tools: Optional[List] = None,
tool_choice: Optional[Union[str, dict]] = None,