from .mini_bench.agent import ChecklistGenerationAgent
def generate_checklist(**data):
# data: 'intent', 'start_url', 'text_observation'
agent_config = {
'model_name': 'WPRM/qwen-3b-ar-reward-cot-mtl-checklist-enhanced',
'base_url': 'http://165.132.144.84:7701/v1',
'api_key': 'empty',
'temperature': 0.7,
'use_log_probs': True,
'use_checklist': True,
'use_multimodal': False,
'num_generate': 1,
}
checklist_generation_agent = ChecklistGenerationAgent(agent_config)
response_list, cost = checklist_generation_agent.generate_response(data, prompt_type='ours', constraint_str_list=["", "", "", ""])
response = response_list[0]
return response.split("")[-1].split("")[0].strip()