diff --git a/atlasbot/engine/answerer/engine.py b/atlasbot/engine/answerer/engine.py index d01b925..02e5702 100644 --- a/atlasbot/engine/answerer/engine.py +++ b/atlasbot/engine/answerer/engine.py @@ -178,7 +178,7 @@ class AnswerEngine: dedup_prompt = prompts.DEDUP_PROMPT + "\nDraft: " + reply return await call_llm(prompts.DEDUP_SYSTEM, dedup_prompt, model=plan.fast_model, tag=tag) - async def _answer_followup(self, question: str, state: ConversationState, summary: dict[str, Any], classify: dict[str, Any], plan: ModePlan, call_llm: Callable[..., Any]) -> str: # noqa: C901, ARG002 + async def _answer_followup(self, question: str, state: ConversationState, summary: dict[str, Any], plan: ModePlan, call_llm: Callable[..., Any]) -> str: # noqa: C901, ARG002 claim_ids = await self._select_claims(question, state.claims, plan, call_llm) selected = [claim for claim in state.claims if claim.id in claim_ids] if claim_ids else state.claims[:2] evidence_lines = []