o
    }oi                     @   s<   d dl Z d dlmZ d dlmZ d dlmZmZ dd ZdS )    N)Request)SystemMessageInterceptor)AdapterMetadataAdapterRequestc                  C   s   t dd} dddddddgd	d
d}tjdddit|d}t|t d}| |}|j	 }|d d d dks>J d S )Nzdetailed thinking on)new_system_messagezgpt-4.1systemzyou are a helpful assistant)rolecontentuserz&Are semicolons optional in JavaScript?d   g      ?)modelmessages
max_tokenstemperaturePOSTzContent-Typezapplication/json)methodheadersdata)rmetar   r   r	   )
r   r   from_valuesjsondumpsr   r   intercept_requestr   get_json)system_message_interceptorr   requestadapter_requestadapter_responsejson_output r    a/home/ubuntu/.local/lib/python3.10/site-packages/tests/evaluation/adapters/test_system_message.pytest_new_system_injected   s(   
	

r"   )	r   flaskr   Pnemo.collections.llm.evaluation.adapters.interceptors.system_message_interceptorr   ;nemo.collections.llm.evaluation.adapters.interceptors.typesr   r   r"   r    r    r    r!   <module>   s
   