a
    f	g3                     @   s   d dl Z d dlZd dlZd dlZd dlZd dlmZ dd Zdd Zdd Zd	d
 Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd ZdS )    N)errorc                  C   s^   t jjttddddd} | jdks0J d| v s<J t jj| j	d} | j
dksZJ d S )	Nztest file dataZtada)prompt
completionz	fine-tune)filepurposeid)r   Zuploaded)openaiFilecreateioStringIOjsondumpsr   retriever   statusresult r   ^/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/openai/tests/test_endpoints.pytest_file_upload   s    r   c                  C   s.   t jjddddgd} t| jdks*J d S )Ngpt-3.5-turbouserHello!rolecontent)modelmessages   r   ZChatCompletionr
   lenchoicesr   r   r   r   test_chat_completions   s    r"   c                  C   s0   t jjddddgdd} t| jdks,J d S )Nr   r   r   r      )r   r   nr   r   r   r   r   test_chat_completions_multiple"   s    r%   c                  C   s>   d } t jjddddgdd}|D ]} t| jdks"J q"d S )Nr   r   r   r   T)r   r   streamr   r   )r   eventsr   r   r   test_chat_completions_streaming)   s    
r(   c                  C   s(   t jjdddd} t| jdks$J d S )NThis was a testr#   adar   r$   Zenginer   
Completionr
   r    r!   r   r   r   r   test_completions5   s    r.   c                  C   s,   t jjddgddd} t| jdks(J d S )Nr)   zThis was another testr#   r*   r+   
   r,   r   r   r   r   !test_completions_multiple_prompts:   s    
r0   c                  C   s8   t jjdddd} t| jdks$J | jds4J d S )Nr)   r#   r*   )r   r$   r   )r   r-   r
   r    r!   r   
startswithr   r   r   r   test_completions_modelA   s    r2   c                   C   sF   t tj& tjjdddddd W d    n1 s80    Y  d S )NA  testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttestr/   r*   d   g{Gz?)r   r$   r   Z
max_tokensrequest_timeout)pytestZraisesr   Timeoutr   r-   r
   r   r   r   r   test_timeout_raises_errorG   s    r7   c                   C   s   t jjdddd d S )Ntestr*   r/   )r   r   r4   )r   r-   r
   r   r   r   r   test_timeout_does_not_errorS   s
    r9   c                  C   sJ   t  .} | t_tjjddd}|s(J W d    n1 s<0    Y  d S )Nhello worldr*   r   r   )requestsSessionr   requestssessionr-   r
   )sessionr   r   r   r   test_user_session\   s    
r@   c                  C   s*   dd } | t _t jjddd}|s&J d S )Nc                  S   s"   t  } | dt jjdd | S )Nzhttps://   )max_retries)r<   r=   mountadaptersHTTPAdapter)r?   r   r   r   factoryh   s    z*test_user_session_factory.<locals>.factoryr:   r*   r;   )r   r>   r-   r
   )rF   r   r   r   r   test_user_session_factoryg   s    rG   )r   r   r5   r<   r   r   r   r"   r%   r(   r.   r0   r2   r7   r9   r@   rG   r   r   r   r   <module>   s    	