Skip to content

Commit c3b5237

Browse files
committed
formatting
1 parent 617dc0a commit c3b5237

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

stagehand/llm/inference.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,14 @@ async def extract(
172172
response_format = {"type": "json_object"}
173173
if schema:
174174
if isinstance(schema, dict):
175-
response_format = {"type": "json_schema", "json_schema": {"name": "extraction_schema", "strict": False, "schema": schema}}
175+
response_format = {
176+
"type": "json_schema",
177+
"json_schema": {
178+
"name": "extraction_schema",
179+
"strict": False,
180+
"schema": schema,
181+
},
182+
}
176183
else:
177184
response_format = schema
178185

0 commit comments

Comments
 (0)