Merge pull request #5199 from zby/calendar_example

Mismatch in example fixed
This commit is contained in:
Krish Dholakia 2024-08-14 08:37:07 -07:00 committed by GitHub
commit 5b46c88eb6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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))