Skip to content

Commit 6865988

Browse files
committed
Removed pasa.pdf from test_agent_sharing_state as a perf boost
1 parent d543259 commit 6865988

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/test_agents.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -566,6 +566,13 @@ async def gen_answer(self, state) -> str: # noqa: ARG001, RUF029
566566
async def test_agent_sharing_state(
567567
agent_test_settings: Settings, subtests: SubTests, callback_type: str | None
568568
) -> None:
569+
def files_filter(f) -> bool:
570+
# Filter out pasa.pdf just to speed the test and save API costs
571+
return f.name != "pasa.pdf" and IndexSettings.model_fields[
572+
"files_filter"
573+
].default(f)
574+
575+
agent_test_settings.agent.index.files_filter = files_filter
569576
agent_test_settings.agent.search_count = 3 # Keep low for speed
570577
agent_test_settings.answer.evidence_k = 2
571578
agent_test_settings.answer.answer_max_sources = 1

0 commit comments

Comments
 (0)