_j:ddlZddlZddlmZddZddZddZdS)N)OpenAIEmbeddingFunctionreturnc0tjdtjdt tjddd}|dg}|Jt |dksJt |ddksJdS) NOPENAI_API_KEYOPENAI_API_KEY not setztext-embedding-3-small@)api_key model_name dimensions hello worldr)osenvirongetpytestskiprlen)ef embeddingss \/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/test/ef/test_openai_ef.pytest_with_embedding_dimensionsr s z~~&''/ ,---  +,+   B ]O$$J  ! ! ! z??a     z!}   # # # # # #c.tjdtjdt tjdd}tjtd5|dgddddS#1swxYwYdS)Nrrrr r z1This model does not support specifying dimensionsmatchr )rrrrrrraises Exceptionrs r9test_with_embedding_dimensions_not_working_with_old_modelr s z~~&''/ ,--- 4D)ERT U U UB L    M?s0 B  BBctjddtdd}tjtd5|d gddddS#1swxYwYdS) Nopenaizopenai not installed)reasonincorrect_api_keyrrzIncorrect API key providedrr )r importorskiprrrrs rtest_with_incorrect_api_keyr&"s )?@@@@ )< L L LB y(D E E E M?s AA!$A!)rN)rrr)s{  $ $ $ $r