
    3i                     >    S SK r S SKrS SKJr  SS jrSS jrSS jrg)    N)OpenAIEmbeddingFunctionc                     [         R                  R                  S5      c  [        R                  " S5        [        [         R                  S   SSS9n U " S/5      nUc   e[        U5      S:X  d   e[        US   5      S:X  d   eg )	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     Y/home/james-whalen/.local/lib/python3.13/site-packages/chromadb/test/ef/test_openai_ef.pytest_with_embedding_dimensionsr   
   s    	zz~~&'/,-	 

+,+
B
 ]O$J!!!z?az!}###    c                     [         R                  R                  S5      c  [        R                  " S5        [        [         R                  S   SS9n [        R                  " [        SS9   U " S/5        S S S 5        g ! , (       d  f       g = f)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   +
A>>
Bc                      [         R                  " SSS9  [        SSS9n [         R                  " [        SS9   U " S	/5        S S S 5        g ! , (       d  f       g = f)
Nopenaizopenai not installed)reasonincorrect_api_keyr   r   zIncorrect API key providedr   r   )r   importorskipr   r   r   r   s    r   test_with_incorrect_api_keyr%   "   sF    
)?@	 )<	LB	y(D	E
M? 
F	E	Es   
A
A)returnN)r   r   <chromadb.utils.embedding_functions.openai_embedding_functionr   r   r   r%    r   r   <module>r)      s    	 
$r   