25 lines
742 B
Python
25 lines
742 B
Python
from negotiation_agent.environment import NegotiationEnv
|
|
|
|
def initialize_environment_usecase(scenario: int = 0,
|
|
target_price: float = 100,
|
|
threshold_price: float = 120) -> NegotiationEnv:
|
|
"""
|
|
환경을 초기화하고 설정하는 유스케이스
|
|
|
|
Args:
|
|
scenario (int): 초기 시나리오 번호 (0-3)
|
|
target_price (float): 목표 가격
|
|
threshold_price (float): 임계 가격
|
|
|
|
Returns:
|
|
NegotiationEnv: 초기화된 협상 환경
|
|
"""
|
|
env = NegotiationEnv(
|
|
scenario=scenario,
|
|
target_price=target_price,
|
|
threshold_price=threshold_price
|
|
)
|
|
initial_state, _ = env.reset()
|
|
|
|
return env
|