diff --git a/src/paperqa/docs.py b/src/paperqa/docs.py index 4c54b6712..4b2e03434 100644 --- a/src/paperqa/docs.py +++ b/src/paperqa/docs.py @@ -780,6 +780,7 @@ async def aquery( summary_llm_model: LLMModel | None = None, embedding_model: EmbeddingModel | None = None, partitioning_fn: Callable[[Embeddable], int] | None = None, + response_schema: BaseModel | None = None, ) -> PQASession: query_settings = get_settings(settings) answer_config = query_settings.answer @@ -862,6 +863,7 @@ async def aquery( messages=messages, callbacks=callbacks, name="answer", + response_format=response_schema, ) answer_text = cast("str", answer_result.text) answer_reasoning = answer_result.reasoning_content