_jnddlZddlmZddlmZddlmZmZmZm Z ddl m Z ddl Z GddeZ dS)N)OpenAIEmbeddingFunctionvalidate_config_schema)DictAnyOptionalList)SpaceceZdZ ddeededefdZedefdZdefdZ de efd Z de ee ffd Zed e ee fddfd Zed e ee fdd fdZd S)BasetenEmbeddingFunctionCHROMA_BASETEN_API_KEYapi_keyapi_baseapi_key_env_varc ddl}n#t$rtdwxYw|tjdt t jdd|_n||_|pt j|j}|std|jd||_ |std||_ d |_ d|_ | |j |j |_dS) a^ Initialize the BasetenEmbeddingFunction. Args: api_key (str, optional): The API key for your Baseten account api_base (str, required): The Baseten URL of the deployment api_key_env_var (str, optional): The environment variable to use for the API key. Defaults to "CHROMA_BASETEN_API_KEY". rNzWThe openai python package is not installed. Please install it with `pip install openai`zDirect api_key configuration will not be persisted. Please use environment variables via api_key_env_var for persistent storage.BASETEN_API_KEYzAPI key not provided and z! environment variable is not set.z'The api_base argument must be provided.zbaseten-embedding-model)rbase_url)openai ImportError ValueErrorwarningswarnDeprecationWarningosgetenvrrr model_name dimensionsOpenAIclient)selfrrrrresolved_api_keys z/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/utils/embedding_functions/baseten_embedding_function.py__init__z!BasetenEmbeddingFunction.__init__ s'  MMMM   i     M_"    9& ' ' 3#4D #2D #Ebi0D&E&E cD,@ccc (  HFGG G  3mmDL4=mQQ s!returncdS)Nbasetenr'r"namezBasetenEmbeddingFunction.name;syr(cdS)Ncosiner'r s r" default_spacez&BasetenEmbeddingFunction.default_space?sxr(c gdS)N)r+l2ipr'r,s r"supported_spacesz)BasetenEmbeddingFunction.supported_spacesBs%%%%r(c |j|jdS)Nrrr3r,s r" get_configz#BasetenEmbeddingFunction.get_configEs Md>RSSSr(configc|d}|d}||td|tdtd||S)a~ Build the BasetenEmbeddingFunction from a configuration dictionary. Args: config (Dict[str, Any]): A dictionary containing the configuration parameters. Expected keys: 'api_key', 'api_base', 'api_key_env_var'. Returns: BasetenEmbeddingFunction: An instance of BasetenEmbeddingFunction. rrNzVMissing 'api_key_env_var' or 'api_base' in configuration for BasetenEmbeddingFunction.zAMissing 'api_base' in configuration for BasetenEmbeddingFunction.)rrr)getrr )r5rrs r"build_from_configz*BasetenEmbeddingFunction.build_from_configHs!**%677::j))  "h&6h   S (+    r(Nc&t|ddS)z Validate the configuration using the JSON schema. Args: config: Configuration to validate Raises: ValidationError: If the configuration does not match the schema r&Nr)r5s r"validate_configz(BasetenEmbeddingFunction.validate_configis vy11111r()r )__name__ __module__ __qualname__rstrr# staticmethodr)r r-r r1rrr4r8r:r'r(r"r r sR 8 -R-R#-R-R -R-R-R-R^#\u&$u+&&&&TDcNTTTT $sCx. 5O   \ @ 2S#X 24 2 2 2\ 2 2 2r(r )rrDs NMMMMM,,,,,,,,,,,,$$$$$$i2i2i2i2i26i2i2i2i2i2r(