Bvh8ddlZddlZddlmZddZddZddZy)N)OpenAIEmbeddingFunctionc tjjdtjdt tjddd}|dg}|Jt |dk(sJt |ddk(sJy) NOPENAI_API_KEYOPENAI_API_KEY not setztext-embedding-3-small@)api_key model_name dimensions hello worldr)osenvirongetpytestskiprlen)ef embeddingss I/RAG/venv/lib/python3.12/site-packages/chromadb/test/ef/test_openai_ef.pytest_with_embedding_dimensionsr s zz~~&'/ ,-  +,+ B ]O$J  !! ! z?a   z!}  ## #ctjjdtjdt tjdd}tj td5|dgdddy#1swYyxYw)Nrrrrr z1This model does not support specifying dimensionsmatchr )r rrrrrraises Exceptionrs r9test_with_embedding_dimensions_not_working_with_old_modelrsh zz~~&'/ ,- 4D)ERT UB L  M?   s . BB ctjddtdd}tjtd5|d gdddy#1swYyxYw) Nopenaizopenai not installed)reasonincorrect_api_keyrrzIncorrect API key providedrr )r importorskiprrrrs rtest_with_incorrect_api_keyr%"sF )?@ )< LB y(D E M? F E Es AA)returnN)r rr)s  $r