
    ug                     0    d dl Z d dlZd dlmZ ddZddZy)    N)OpenAIEmbeddingFunctionc                  
   t         j                  j                  d      t        j                  d       t        t         j                  d   dd      }  | dg      }|J t        |      dk(  sJ t        |d         dk(  sJ y )	NOPENAI_API_KEYOPENAI_API_KEY not setztext-embedding-3-small@   )api_key
model_name
dimensionshello world   r   )osenvirongetpytestskipr   len)ef
embeddingss     T/var/www/openai/venv/lib/python3.12/site-packages/chromadb/test/ef/test_openai_ef.pytest_with_embedding_dimensionsr   
   s    	zz~~&'/,-	 

+,+
B
 ]O$J!!!z?az!}###    c                     t         j                  j                  d      t        j                  d       t        t         j                  d   d      } t        j                  t        d      5   | dg       d d d        y # 1 sw Y   y xY w)Nr   r   r   )r   r
   z1This model does not support specifying dimensions)matchr   )r   r   r   r   r   r   raises	Exception)r   s    r   9test_with_embedding_dimensions_not_working_with_old_modelr      sh    	zz~~&'/,-	 4D)ERT	UB	L
 	M?
 
 
s   .
BB
)returnN)r   r   <chromadb.utils.embedding_functions.openai_embedding_functionr   r   r    r   r   <module>r       s    	 
$r   