mirror of
https://github.com/langchain-ai/ambient-agent-101.git
synced 2026-07-01 21:14:02 -04:00
catherine-langchain
23 lines
631 B
Python
23 lines
631 B
Python
#!/usr/bin/env python
|
|
|
|
import pytest
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
project_root = str(Path(__file__).parent.parent)
|
|
if project_root not in sys.path:
|
|
sys.path.append(project_root)
|
|
|
|
def pytest_addoption(parser):
|
|
"""Add command-line options to pytest."""
|
|
parser.addoption(
|
|
"--agent-module",
|
|
action="store",
|
|
default="email_assistant_hitl_memory",
|
|
help="Specify which email assistant module to test"
|
|
)
|
|
|
|
@pytest.fixture(scope="session")
|
|
def agent_module_name(request):
|
|
"""Return the agent module name from command line."""
|
|
return request.config.getoption("--agent-module") |