mirror of
https://github.com/pydantic/pydantic-ai-harness.git
synced 2026-07-21 02:45:34 +00:00
e1654ccfe2
* feat: add FileSystem and Shell capabilities with exhaustive testing - FileSystemToolset: 8 tools (read, write, edit, list, search, find, mkdir, info) with path-traversal prevention, allow/deny patterns, optimistic concurrency - ShellToolset: 1 tool (run_command) with command validation, timeout handling, and async subprocess execution via anyio --- Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: David Sanchez <64162682+dsfaccini@users.noreply.github.com>
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import inspect
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from pydantic_ai import Agent
|
|
from pydantic_ai.models.test import TestModel
|
|
|
|
import pydantic_ai_harness
|
|
|
|
|
|
def test_import():
|
|
assert pydantic_ai_harness.__doc__ is not None
|
|
assert isinstance(pydantic_ai_harness.__all__, list)
|
|
|
|
|
|
def test_lazy_import_filesystem():
|
|
from pydantic_ai_harness import FileSystem
|
|
|
|
assert inspect.isclass(FileSystem)
|
|
assert hasattr(FileSystem, 'get_toolset')
|
|
|
|
|
|
def test_lazy_import_shell():
|
|
from pydantic_ai_harness import Shell
|
|
|
|
assert inspect.isclass(Shell)
|
|
assert hasattr(Shell, 'get_toolset')
|
|
|
|
|
|
def test_lazy_import_unknown():
|
|
with pytest.raises(AttributeError, match='has no attribute'):
|
|
pydantic_ai_harness.__getattr__('Nonexistent')
|
|
|
|
|
|
def test_test_model_fixture(test_model: TestModel):
|
|
assert isinstance(test_model, TestModel)
|
|
|
|
|
|
def test_test_agent_fixture(test_agent: Agent[None, str]):
|
|
assert test_agent.name == 'test-agent'
|
|
|
|
|
|
def test_tmp_dir_fixture(tmp_dir: Path):
|
|
assert tmp_dir.is_dir()
|
|
|
|
|
|
async def test_allow_model_requests(allow_model_requests: None):
|
|
import pydantic_ai.models
|
|
|
|
assert pydantic_ai.models.ALLOW_MODEL_REQUESTS is True
|