Skip to content

Commit de6751f

Browse files
committed
Add response_schema parameter to Docs's aquery response generation
1 parent 04971ba commit de6751f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/paperqa/docs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -780,6 +780,7 @@ async def aquery(
780780
summary_llm_model: LLMModel | None = None,
781781
embedding_model: EmbeddingModel | None = None,
782782
partitioning_fn: Callable[[Embeddable], int] | None = None,
783+
response_schema: BaseModel | None = None,
783784
) -> PQASession:
784785
query_settings = get_settings(settings)
785786
answer_config = query_settings.answer
@@ -862,6 +863,7 @@ async def aquery(
862863
messages=messages,
863864
callbacks=callbacks,
864865
name="answer",
866+
response_format=response_schema,
865867
)
866868
answer_text = cast("str", answer_result.text)
867869
answer_reasoning = answer_result.reasoning_content

0 commit comments

Comments
 (0)