o g@s4ddlZddlZddlmZdddZdddZdS) N)OpenAIEmbeddingFunctionreturncCsltjddur tdttjdddd}|dg}|dus"Jt|dks*Jt|ddks4JdS) NOPENAI_API_KEYOPENAI_API_KEY not setztext-embedding-3-small@)api_key model_name dimensions hello worldr)osenvirongetpytestskiprlen)ef embeddingsrV/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/ef/test_openai_ef.pytest_with_embedding_dimensions s   rcCsjtjddur tdttjddd}tjtdd|dgWddS1s.wYdS)Nrrr)rr z1This model does not support specifying dimensions)matchr )r r rrrrraises Exception)rrrr9test_with_embedding_dimensions_not_working_with_old_models  "r)rN)r rs