
    ugo                     4    d dl mZ d dlmZ ddZddZddZy)    )embedding_functions)EmbeddingFunctionNc                  >    h d} | t        j                         k(  sJ y)a#  
    Ensure that `get_builtins` is consistent after the ef migration.

    This test is intended to be temporary until the ef migration is complete as
    these expected builtins are likely to grow as long as users add new
    embedding functions.

    REMOVE ME ON THE NEXT EF ADDITION
    >   ONNXMiniLM_L6_V2JinaEmbeddingFunctionCohereEmbeddingFunctionOllamaEmbeddingFunctionOpenAIEmbeddingFunctionOpenCLIPEmbeddingFunctionRoboflowEmbeddingFunctionText2VecEmbeddingFunctionVoyageAIEmbeddingFunctionHuggingFaceEmbeddingServerGooglePalmEmbeddingFunctionInstructorEmbeddingFunctionHuggingFaceEmbeddingFunctionGoogleVertexEmbeddingFunctionAmazonBedrockEmbeddingFunction ChromaLangchainEmbeddingFunction#GoogleGenerativeAiEmbeddingFunction$SentenceTransformerEmbeddingFunctionN)r   get_builtins)expected_builtinss    M/var/www/openai/venv/lib/python3.12/site-packages/chromadb/test/ef/test_ef.pytest_get_builtins_holdsr      s$    *  3 @ @ BBBB    c                  |    t        t        d      sJ t        j                         } | J t        | t              sJ y )NDefaultEmbeddingFunction)hasattrr   r   
isinstancer   )
default_efs    r   test_default_ef_existsr"   '   s?    &(BCCC$==?J!!!j"3444r   c                      t        j                         D ]Z  } | dk(  r	t        t         |       sJ t        t	        t         |       t
              sJ t        t	        t         |       t              rZJ  y )Nr   )r   r   r   r    getattrtype
issubclassr   )efs    r   test_ef_importsr(   /   sc    !..033*B///'"5r:DAAA'"5r:<MNNN 1r   )returnN)chromadb.utilsr   chromadb.api.typesr   r   r"   r(    r   r   <module>r-      s    . 0CD5Or   