o
    wi                     @   sb   d dl mZmZ d dlZd dlZd dlmZ d dlmZ ej	deeeef fddZ
dd	 ZdS )
    )Any	GeneratorN)create_server_process)AdapterConfigreturnc                 c   s2    t dddddd}t|\}}|V  |  d S )Nz)http://localhost:3300/v1/chat/completionsT</think>   )api_urluse_reasoningend_reasoning_tokenmax_logged_responsesmax_logged_requests)r   r   	terminate)fake_openai_endpointadapter_configp r   j/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/evaluation/adapters/test_adapter_server.pyadapter_server   s   	r   c                 C   s   d| j  }dddd}tj||d}|jdksJ d| v s"J t| d d	ks.J tj||d}|jdks<J d| v sDJ t| d d	ksPJ d
| d d	 d d vs`J d S )Nzhttp://localhost:zThis is a test promptd   g      ?)prompt
max_tokenstemperature)json   choicesr   r   messagecontent)
local_portrequestspoststatus_coder   len)r   capfdurldataresponser   r   r    test_adapter_server_post_request,   s   $r'   )typingr   r   pytestr   /nemo.collections.llm.evaluation.adapters.serverr   #nemo.collections.llm.evaluation.apir   fixturer   r'   r   r   r   r   <module>   s   