liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
# Copyright Sierra
|
|
|
|
from typing import Optional, Union
|
|
from tau_bench.envs.base import Env
|
|
from tau_bench.envs.user import UserStrategy
|
|
|
|
|
|
def get_env(
|
|
env_name: str,
|
|
user_strategy: Union[str, UserStrategy],
|
|
user_model: str,
|
|
task_split: str,
|
|
user_provider: Optional[str] = None,
|
|
task_index: Optional[int] = None,
|
|
user_seed: Optional[int] = None,
|
|
) -> Env:
|
|
if env_name == "retail":
|
|
from tau_bench.envs.retail import MockRetailDomainEnv
|
|
|
|
return MockRetailDomainEnv(
|
|
user_strategy=user_strategy,
|
|
user_model=user_model,
|
|
task_split=task_split,
|
|
user_provider=user_provider,
|
|
task_index=task_index,
|
|
user_seed=user_seed,
|
|
)
|
|
elif env_name == "airline":
|
|
from tau_bench.envs.airline import MockAirlineDomainEnv
|
|
|
|
return MockAirlineDomainEnv(
|
|
user_strategy=user_strategy,
|
|
user_model=user_model,
|
|
task_split=task_split,
|
|
user_provider=user_provider,
|
|
task_index=task_index,
|
|
user_seed=user_seed,
|
|
)
|
|
else:
|
|
raise ValueError(f"Unknown environment: {env_name}")
|