Skip to content

Commit 0696d8a

Browse files
committed
rename to answer_iteration_prompt
1 parent 3e5ea7f commit 0696d8a

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

paperqa/docs.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -779,8 +779,8 @@ async def aquery( # noqa: PLR0912
779779
else:
780780
with set_llm_session_ids(session.id):
781781
prior_answer_prompt = ""
782-
if prompt_config.iteration_prompt and session.answer:
783-
prior_answer_prompt = prompt_config.iteration_prompt.format(
782+
if prompt_config.answer_iteration_prompt and session.answer:
783+
prior_answer_prompt = prompt_config.answer_iteration_prompt.format(
784784
prior_answer=session.answer
785785
)
786786
messages = [

paperqa/settings.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,7 @@ class PromptSettings(BaseModel):
266266

267267
summary: str = summary_prompt
268268
qa: str = qa_prompt
269-
iteration_prompt: str | None = Field(
269+
answer_iteration_prompt: str | None = Field(
270270
default=answer_iteration_prompt_template,
271271
description=(
272272
"Prompt to inject existing prior answers into the qa prompt to allow the model to iterate. "

tests/test_paperqa.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,7 @@ def test_location_awareness(docs_fixture) -> None:
539539

540540

541541
def test_query(docs_fixture) -> None:
542-
settings = Settings(prompts={"iteration_prompt": None})
542+
settings = Settings(prompts={"answer_iteration_prompt": None})
543543
docs_fixture.query("Is XAI usable in chemistry?", settings=settings)
544544

545545

@@ -559,7 +559,7 @@ def test_query_with_iteration(docs_fixture) -> None:
559559
), "prior answer not in prompt"
560560
# run without a prior session to check that the flow works correctly
561561
docs_fixture.query(question, llm_model=llm, settings=settings)
562-
assert settings.prompts.iteration_prompt[:10] not in cast( # type: ignore[index]
562+
assert settings.prompts.answer_iteration_prompt[:10] not in cast( # type: ignore[index]
563563
"str", my_results[-1].prompt[1].content # type: ignore[union-attr, index]
564564
), "prior answer prompt should not be inserted"
565565

0 commit comments

Comments
 (0)