Ë
    lQêjÞ  ã                   ó8   — d dl Z d dlZd dlmZ dd„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     úd/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/test/ef/test_openai_ef.pyÚtest_with_embedding_dimensionsr   
   sƒ   € Ü	‡z�z‡~�~Ð&Ó'Ð/Ü�‰Ð,Ô-Ü	 Ü—
‘
Ð+Ñ,Ø+Øô
€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      sm   € Ü	‡z�z‡~�~Ð&Ó'Ð/Ü�‰Ð,Ô-Ü	 ¬¯©Ð4DÑ)EÐRTÔ	U€BÜ	�‰ÜÐLô
ñ ñ 	ˆMˆ?Ô÷÷ ñ ús   Á.
BÂB
c                  ó¾   — t        j                  dd¬«       t        dd¬«      } t        j                  t        d¬«      5   | d	g«       d d d «       y # 1 sw Y   y xY w)
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%   "   sN   € Ü
×Ñ˜Ð)?Õ@Ü	 Ð)<ÈÔ	L€BÜ	�‰”yÐ(DÔ	Eñ Ù
ˆMˆ?Ô÷÷ ñ ús   Á 
AÁA)ÚreturnN)r   r   Ú<chromadb.utils.embedding_functions.openai_embedding_functionr   r   r   r%   © r   r   ú<module>r)      s   ðÛ 	ã õó
$óôr   