o
    ॵi                     @   s  g d Z dd e D Zg dZg dZddddd	d
dZddgg dg dg dg dg dg ddZg dZg dg dgg dg dg dg ddZg dZeg d Z	i Z
e	D ]Zde
e< q[dddd dd!d"d#Zg d$g d$g d%g d&d'd(gd'd(gd'd(gg d)d*Zg Ze D ]ZeD ]Zeevree qqe	d+d,g ed+dg e	d+g ed+d,g e	d+g g d-e	d+g e	d+g g g g g d.Ze	d+d,g Zi Zi d/d0d1d0d2d3d4d3d5d3d6d7d8d7d9d7d:d7d;d7d<d7d=d>d?d>d@d>dAdBdCdBdDdBdEdEdEdFZi d/dGd1dGd2dHd4dHd5dHd6dId8dId9dId:dId;dId<dId=dJd?dJd@dJdAdKdCdKdDdKdLdLdLdFZg dMZdNdO ZdPdQ ZdRdS ZdTS )U)
restauranthotel
attractiontraintaxipolicehospitalc                 C   s   g | ]}d  |qS )z[{}])format).0item r   W/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/utils/nlp/space/ontology.py
<listcomp>   s    r   )r   r   r   r   z<go_r>z<go_b>z<go_a>z<go_d><eos_u><eos_r><eos_b><eos_a><eos_d><eos_q><sos_u><sos_r><sos_b><sos_a><sos_d><sos_q>carpricetimeleavearriveid)zcar typezentrance feedurationleaveatarrivebytrainidphone)postcodeaddressr%   )r'   r%   r&   )
r'   r&   internetr%   parkingtype
pricerangestarsarea	reference)r   r*   r'   r&   r%   r-   r.   )r   r   r   r   r    r.   )r%   r&   r'   r+   foodr-   r.   )r   r   r   r   r   r   r   )r   r'   r&   r%   r(   r)   r*   r+   r/   r,   r-   r.   r   r   r   r   r    )r   destination	departurer   
department)
r*   r)   r+   r(   staydaypeopler-   r,   name)r-   r*   r6   )r0   r4   r   r1   r5   r   )r/   r+   r-   r6   r   r4   r5   )r*   r)   r+   r(   r3   r4   r5   r-   r,   r6   r   r0   r1   r   r2   r/   r   )r3   r4   r5   r6   r0   r1   r2      r'   r&   r.   r1   r0   )addrfeepostrefticketdepartdest)informrequestnooffer	recommendselect	offerbookofferbookednobook)r?   r@   rA   rB   rC   )r?   r@   rA   rD   rE   rC   r?   r@   )byegreetreqmorewelcome)r   r   r   r   r   r   r   generalchoiceopen)r   r5   r3   r.   r4   r6   rL   )r?   r@   rA   rB   rC   rF   rD   rE   rI   rJ   rG   rH   userr   
user_delexrespr   resp_genpv_respbspnr   bspn_genpv_bspnbsdxbsdx_genpv_bsdxqspnr   qspn_genpv_qspnaspnr   aspn_genpv_aspnr   )dspndspn_genpv_dspnr   r   r   r   r   r   )
z<sos_db>z<eos_db>z[book_nores]z[book_fail]z[book_success]z
[db_nores]z[db_0]z[db_1]z[db_2]z[db_3]c                 C   (   g }t | D ]}|d| d q|S )Nz<understand_>rangeappend)prompt_num_for_understandunderstand_tokensir   r   r   get_understand_tokens      rj   c                 C   rb   )Nz<policy_rc   rd   )prompt_num_for_policypolicy_tokensri   r   r   r   get_policy_tokens   rk   rn   c                 C   s   g dt  |  }|S )Nr   )	db_tokens)other_tokensspecial_tokensr   r   r   get_special_tokens   s   rr   N)all_domainsall_domains_with_bracket
db_domainsplaceholder_tokensnormlize_slot_namesrequestable_slotsall_reqslotinformable_slotsall_infslot	all_slotsget_slotsda_abbr_to_slot_namedialog_actsall_actsvaluesactsactrf   dialog_act_paramsdialog_act_all_slotsslot_name_to_slot_token
eos_tokens
sos_tokensro   rj   rn   rr   r   r   r   r   <module>   s4  





	
	
