o
    *j                     @   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   d/var/www/html/Deteccion_Ine/venv/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durationZleaveatZarrivebyZ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'   r3   )r-   r1   r   r.   r2   r   )r,   r(   r*   r3   r   r1   r2   )r'   r&   r(   r%   r0   r1   r2   r*   r)   r3   r   r-   r.   r   r/   r,   r   )r0   r1   r2   r3   r-   r.   r/      r$   r#   r+   r.   r-   )addrZfeepostrefticketZdepartdest)informrequestnooffer	recommendselect	offerbookofferbookednobook)r:   r;   r<   r=   r>   )r:   r;   r<   r?   r@   r>   r:   r;   )byegreetreqmorewelcome)r   r   r   r   r   r   r   Zgeneralchoiceopen)r   r2   r0   r+   r1   r3   rF   )r:   r;   r<   r=   r>   rA   r?   r@   rD   rE   rB   rC   userr   Z
user_delexrespr   Zresp_genZpv_respZbspnr   Zbspn_genZpv_bspnZbsdxZbsdx_genZpv_bsdxZqspnr   Zqspn_genZpv_qspnZaspnr   Zaspn_genZpv_aspnr   )ZdspnZdspn_genZ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)Zprompt_num_for_understandZunderstand_tokensir   r   r   get_understand_tokens      rP   c                 C   rJ   )Nz<policy_rK   rL   )Zprompt_num_for_policyZpolicy_tokensrO   r   r   r   get_policy_tokens   rQ   rR   c                 C   s   g dt  |  }|S )Nr   )	db_tokens)Zother_tokensZspecial_tokensr   r   r   get_special_tokens   s   rT   N)Zall_domainsZall_domains_with_bracketZ
db_domainsZplaceholder_tokensZnormlize_slot_namesZrequestable_slotsZall_reqslotZinformable_slotsZall_infslotZ	all_slotsZget_slotsZda_abbr_to_slot_nameZdialog_actsZall_actsvaluesZactsZactrN   Zdialog_act_paramsZdialog_act_all_slotsZslot_name_to_slot_tokenZ
eos_tokensZ
sos_tokensrS   rP   rR   rT   r   r   r   r   <module>   s4  





	
	
