o
    RTh3                     @   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sJ d| v sJ t jj| j	d} | j
dks-J d S )	Nztest file datatada)prompt
completionz	fine-tune)filepurposeid)r   uploaded)openaiFilecreateioStringIOjsondumpsr   retriever   statusresult r   ^/home/air/segue/gemini/backup/venv/lib/python3.10/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
   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'   enginer
   
Completionr   r#   r$   r   r   r   r   test_completions5   s   r2   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.   
   r0   r   r   r   r   !test_completions_multiple_prompts:   s   
r4   c                  C   s8   t jjdddd} t| jdksJ | jdsJ d S )Nr,   r&   r-   )r   r'   r   )r
   r1   r   r#   r$   r   
startswithr   r   r   r   test_completions_modelA   s   r6   c                   C   sH   t tj tjjdddddd W d    d S 1 sw   Y  d S )N  testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttestr3   r-   d   g{Gz?)r   r'   r   
max_tokensrequest_timeout)pytestraisesr   Timeoutr
   r1   r   r   r   r   r   test_timeout_raises_errorG   s   "r>   c                   C   s   t jjdddd d S )Ntestr-   r3   )r   r   r:   )r
   r1   r   r   r   r   r   test_timeout_does_not_errorS   s
   
r@   c                  C   sL   t  } | t_tjjddd}|sJ W d    d S 1 sw   Y  d S )Nhello worldr-   r   r   )requestsSessionr
   requestssessionr1   r   )sessionr   r   r   r   test_user_session\   s   

"rG   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)rC   rD   mountadaptersHTTPAdapter)rF   r   r   r   factoryh   s   z*test_user_session_factory.<locals>.factoryrA   r-   rB   )r
   rE   r1   r   )rM   r   r   r   r   test_user_session_factoryg   s   rN   )r   r   r;   rC   r
   r   r   r%   r(   r+   r2   r4   r6   r>   r@   rG   rN   r   r   r   r   <module>   s"    	