Skip to content

Commit 306758c

Browse files
committed
changed tests example for agent.md docs
1 parent cdc2ca9 commit 306758c

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

docs/architecture/agent.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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
568568
from pydantic_ai import Agent
@@ -598,12 +598,12 @@ Test with real VLM (expensive, slow):
598598
def 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

0 commit comments

Comments
 (0)