File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -562,7 +562,7 @@ responses = await asyncio.gather(*[
562562
563563# ## Unit Tests
564564
565- Test agent selection with PydanticAI's built-in test model :
565+ Test agent selection with PydanticAI's built-in test model (your catalog should contain the choice provided below, i.e. the `TotalSegmentator` tool) :
566566
567567` ` ` python
568568from pydantic_ai import Agent
@@ -598,12 +598,12 @@ Test with real VLM (expensive, slow):
598598def test_real_agent():
599599 from ai_agent.agent.agent import run_agent
600600
601- with open("tests/data/sample.png ", "rb") as f:
601+ with open("tests/data/sample.tif ", "rb") as f:
602602 image_bytes = f.read()
603603
604604 result = run_agent(
605- task="segment the lungs",
606- image_paths=["tests/data/sample.dcm "],
605+ task="I want to segment the lungs of this CT scan ",
606+ image_paths=["tests/data/sample.tif "],
607607 image_bytes=image_bytes,
608608 )
609609
You can’t perform that action at this time.
0 commit comments