Formatting

Author: pamelafox

structured_outputs_basic.py

@@ -34,7 +34,6 @@
     MODEL_NAME = os.environ["OPENAI_MODEL"]
 
 
-
 class CalendarEvent(BaseModel):
     name: str
     date: str
@@ -52,7 +51,7 @@ class CalendarEvent(BaseModel):
 
 
 message = completion.choices[0].message
-if (message.refusal):
+if message.refusal:
     rich.print(message.refusal)
 else:
     event = message.parsed

structured_outputs_description.py

@@ -34,7 +34,6 @@
     MODEL_NAME = os.environ["OPENAI_MODEL"]
 
 
-
 class CalendarEvent(BaseModel):
     name: str
     date: str = Field(..., description="A date in the format YYYY-MM-DD")
@@ -44,16 +43,19 @@ class CalendarEvent(BaseModel):
 completion = client.beta.chat.completions.parse(
     model=MODEL_NAME,
     messages=[
-        {"role": "system", "content": "Extract the event information. If no year is specified, assume the current year (2025)."},
+        {
+            "role": "system",
+            "content": "Extract the event information. If no year is specified, assume the current year (2025).",
+        },
         {"role": "user", "content": "Alice and Bob are going to a science fair on the 1st of april."},
     ],
     response_format=CalendarEvent,
 )
-CalendarEvent(name='Science Fair', date='2025-04-01', participants=['Alice', 'Bob'])
+CalendarEvent(name="Science Fair", date="2025-04-01", participants=["Alice", "Bob"])
 
 message = completion.choices[0].message
-if (message.refusal):
+if message.refusal:
     rich.print(message.refusal)
 else:
     event = message.parsed
-    rich.print(event)
+    rich.print(event)

structured_outputs_enum.py

@@ -35,20 +35,21 @@
     MODEL_NAME = os.environ["OPENAI_MODEL"]
 
 
-
 class DayOfWeek(str, Enum):
-  SUNDAY = "Sunday"
-  MONDAY = "Monday"
-  TUESDAY = "Tuesday"
-  WEDNESDAY = "Wednesday"
-  THURSDAY = "Thursday"
-  FRIDAY = "Friday"
-  SATURDAY = "Saturday"
+    SUNDAY = "Sunday"
+    MONDAY = "Monday"
+    TUESDAY = "Tuesday"
+    WEDNESDAY = "Wednesday"
+    THURSDAY = "Thursday"
+    FRIDAY = "Friday"
+    SATURDAY = "Saturday"
+
 
 class CalendarEvent(BaseModel):
-  name: str
-  date: DayOfWeek
-  participants: list[str]
+    name: str
+    date: DayOfWeek
+    participants: list[str]
+
 
 completion = client.beta.chat.completions.parse(
     model=MODEL_NAME,
@@ -61,7 +62,7 @@ class CalendarEvent(BaseModel):
 
 
 message = completion.choices[0].message
-if (message.refusal):
+if message.refusal:
     rich.print(message.refusal)
 else:
     event = message.parsed

structured_outputs_function_calling.py

@@ -35,15 +35,16 @@
 
 
 class GetDeliveryDate(BaseModel):
-  order_id: str
+    order_id: str
+
 
 response = client.chat.completions.create(
-   model=MODEL_NAME,
-   messages=[
-     {"role": "system",
-      "content": "You're a customer support bot. Use the tools to assist the user."},
-     {"role": "user",
-      "content": "Hi, can you tell me the delivery date for my order #12345?"}],
-   tools=[openai.pydantic_function_tool(GetDeliveryDate)])
-
-rich.print(response.choices[0].message.tool_calls[0].function)
+    model=MODEL_NAME,
+    messages=[
+        {"role": "system", "content": "You're a customer support bot. Use the tools to assist the user."},
+        {"role": "user", "content": "Hi, can you tell me the delivery date for my order #12345?"},
+    ],
+    tools=[openai.pydantic_function_tool(GetDeliveryDate)],
+)
+
+rich.print(response.choices[0].message.tool_calls[0].function)

structured_outputs_nested.py

@@ -34,15 +34,16 @@
     MODEL_NAME = os.environ["OPENAI_MODEL"]
 
 
-
 class Participant(BaseModel):
-  name: str
-  job_title: str
+    name: str
+    job_title: str
+
 
 class CalendarEvent(BaseModel):
-  name: str
-  date: str
-  participants: list[Participant]
+    name: str
+    date: str
+    participants: list[Participant]
+
 
 completion = client.beta.chat.completions.parse(
     model=MODEL_NAME,
@@ -55,7 +56,7 @@ class CalendarEvent(BaseModel):
 
 
 message = completion.choices[0].message
-if (message.refusal):
+if message.refusal:
     rich.print(message.refusal)
 else:
     event = message.parsed