diff --git a/docs/my-website/docs/completion/json_mode.md b/docs/my-website/docs/completion/json_mode.md index bf159cd07e7..1d12a22ba0e 100644 --- a/docs/my-website/docs/completion/json_mode.md +++ b/docs/my-website/docs/completion/json_mode.md @@ -84,17 +84,20 @@ from litellm import completion # add to env var os.environ["OPENAI_API_KEY"] = "" -messages = [{"role": "user", "content": "List 5 cookie recipes"}] +messages = [{"role": "user", "content": "List 5 important events in the XIX century"}] class CalendarEvent(BaseModel): name: str date: str participants: list[str] +class EventsList(BaseModel): + events: list[CalendarEvent] + resp = completion( model="gpt-4o-2024-08-06", messages=messages, - response_format=CalendarEvent + response_format=EventsList ) print("Received={}".format(resp))