o
    HEDij                     @   s  d dl mZ d dlZd dlmZmZ d dlmZ ejj	d0ddZ
ejj	d0dd	Zejj	d0d
dZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0ddZejj	d0d d!Zejj	d0d"d#Zejj	d0d$d%Zejj	d0d&d'Zejj	d0d(d)Zejj	d0d*d+Zejj	d0d,d-Zejj	d0d.d/ZdS )1    )api_versionN)
CompletionEngine)ApiTypereturnc                  C      t ddd} | dksJ d S )Ntest_engineazure2021-11-01-previewJ/openai/deployments/test_engine/completions?api-version=2021-11-01-previewr   	class_urlurl r   p/var/www/www-root/data/www/ovozai.pdev.uz/venv/lib/python3.10/site-packages/openai/tests/test_url_composition.py&test_completions_url_composition_azure	      r   c                  C   r   )Nr   azure_adr
   r   r   r   r   r   r   )test_completions_url_composition_azure_ad   r   r   c                  C   s   t d} | dksJ d S )Nr    /engines/test_engine/completionsr   r   r   r   r   (test_completions_url_composition_default   s   
r   c                  C   s   t dd} | dksJ d S )Nr   open_air   r   r   r   r   r   (test_completions_url_composition_open_ai!   s   r   c                  C   s<   t t tdd} W d    d S 1 sw   Y  d S )Nr   invalid)pytestraises	Exceptionr   r   r   r   r   r   -test_completions_url_composition_invalid_type'   s   "r   c                  C   (   t ddddd} |  }|dksJ d S )Ntest_idr   r	   r
   idengineapi_typer   R/openai/deployments/test_engine/completions/test_id?api-version=2021-11-01-previewr   instance_url
completionr   r   r   r   3test_completions_url_composition_instance_url_azure-      r*   c                  C   r   )Nr    r   r   r
   r!   r%   r&   r(   r   r   r   6test_completions_url_composition_instance_url_azure_ad<   r+   r,   c                  C   sH   t dddd d} tt |   W d    d S 1 sw   Y  d S )Nr    r   r	   r!   r   r   r   r   r'   )r)   r   r   r   >test_completions_url_composition_instance_url_azure_no_versionK   s   
"r.   c                  C   s$   t ddd} |  }|dksJ d S )Nr    r   )r"   r#   (/engines/test_engine/completions/test_idr&   r(   r   r   r   5test_completions_url_composition_instance_url_defaultT   s   r0   c                  C   r   )Nr    r   r   r
   r!   r/   r&   r(   r   r   r   5test_completions_url_composition_instance_url_open_ai[   s   r1   c                  C   sF   t dddd} tt |  }W d    d S 1 sw   Y  d S )Nr    r   r   r"   r#   r$   r-   r(   r   r   r   5test_completions_url_composition_instance_url_invalidg   s   
"r3   c                  C   s0   t ddddd} d| d< |  }|dksJ d S )	Nr    r   r	   r
   r!      timeoutz]/openai/deployments/test_engine/completions/test_id?api-version=2021-11-01-preview&timeout=12r&   r(   r   r   r   ;test_completions_url_composition_instance_url_timeout_azuren   s   r6   c                  C   s.   t dddd} d| d< |  }|dksJ d S )Nr    r   r   r2   r4   r5   z3/engines/test_engine/completions/test_id?timeout=12r&   r(   r   r   r   <test_completions_url_composition_instance_url_timeout_openai~   s   r7   c                  C   F   t dddd} | jdksJ | jtjksJ | d}|dks!J d S )Nr    r	   r
   r"   r$   r   test_operationI/openai/deployments/test_id/test_operation?api-version=2021-11-01-previewr   r$   typed_api_typer   AZUREr'   r#   r   r   r   r   (test_engine_search_url_composition_azure      
r@   c                  C   r8   )Nr    r   r
   r9   r:   r;   )r   r$   r=   r   AZURE_ADr'   r?   r   r   r   +test_engine_search_url_composition_azure_ad   rA   rC   c                  C   sf   t ddd d} | jdksJ | jtjksJ tt | d W d    d S 1 s,w   Y  d S )Nr    r	   r9   r:   )	r   r$   r=   r   r>   r   r   r   r'   r#   r   r   r   3test_engine_search_url_composition_azure_no_version   s   "rE   c                  C   s@   t dddd} | jdksJ | jtjksJ |  dksJ d S )Nr    r	   r
   r9   z6/openai/engines/test_id?api-version=2021-11-01-previewr<   rD   r   r   r   5test_engine_search_url_composition_azure_no_operation   s   rF   c                  C   s@   t dd} | jd ksJ | jtjksJ |  }|dksJ d S )Nr    )r"   /engines/test_idr   r$   r=   r   OPEN_AIr'   r?   r   r   r   *test_engine_search_url_composition_default   s
   
rJ   c                  C   sB   t ddd} | jdksJ | jtjksJ |  }|dksJ d S )Nr    r   r"   r$   rG   rH   r?   r   r   r   *test_engine_search_url_composition_open_ai   s
   rL   c                  C   sZ   t ddd} | jdksJ tt | jtjksJ W d    d S 1 s&w   Y  d S Nr    r   rK   )r   r$   r   r   r   r=   r   rI   rD   r   r   r   /test_engine_search_url_composition_invalid_type   s
   "rN   c                  C   sR   t ddd} | jdksJ tt |   W d    d S 1 s"w   Y  d S rM   )r   r$   r   r   r   searchrD   r   r   r   1test_engine_search_url_composition_invalid_search   s
   
"rP   )r   N)sysr   r   openair   r   openai.utilr   markr   r   r   r   r   r   r*   r,   r.   r0   r1   r3   r6   r7   r@   rC   rE   rF   rJ   rL   rN   rP   r   r   r   r   <module>   s\    
