diff --git a/llm/models.py b/llm/models.py index 9aa4801f..8cc27a9c 100644 --- a/llm/models.py +++ b/llm/models.py @@ -1623,6 +1623,7 @@ def responses(self) -> Iterator[Response]: tool_results=tool_results, options=self.prompt.options, attachments=attachments, + schema=current_response.prompt.schema, ), self.model, stream=self.stream, @@ -1679,6 +1680,7 @@ async def responses(self) -> AsyncIterator[AsyncResponse]: tool_results=tool_results, options=self.prompt.options, attachments=attachments, + schema=current_response.prompt.schema, ) current_response = AsyncResponse( prompt,