Bvh JddlZddlmZddlmZmZmZddlZGddeZy)N)OpenAIEmbeddingFunction)DictAnyOptionalczeZdZ d deededefdZedefdZdeee ffdZ edeee fddfd Z y ) BasetenEmbeddingFunctionapi_keyapi_baseapi_key_env_varc ddl}|tjdt ||_|xstj|}|std|d||_ |s td||_ d|_ d|_ |j|j|j |_y#t$r tdwxYw) 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.zAPI key not provided and z! environment variable is not set.z'The api_base argument must be provided.zbaseten-embedding-model)r base_url)openai ImportError ValueErrorwarningswarnDeprecationWarningr osgetenvr r model_name dimensionsOpenAIclient)selfr r r rresolved_api_keys g/RAG/venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/baseten_embedding_function.py__init__z!BasetenEmbeddingFunction.__init__ s     MM_"  /"@bii&@+O+<<]^ ( FG G  3mmDLL4==mQ 5 i  s B))B>returncy)Nbasetenr!rnamezBasetenEmbeddingFunction.name5sr"c4|j|jdS)Nr r r%)rs r get_configz#BasetenEmbeddingFunction.get_config9s MMd>R>RSSr"configc|jd}|jd}|| 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. r r NzVMissing 'api_key_env_var' or 'api_base' in configuration for BasetenEmbeddingFunction.zAMissing 'api_base' in configuration for BasetenEmbeddingFunction.)r r r )getrr)r'r r s rbuild_from_configz*BasetenEmbeddingFunction.build_from_config<sl!**%67::j)  "h&6h   S (+  r"N)CHROMA_BASETEN_API_KEY) __name__ __module__ __qualname__rstrr staticmethodr#rrr&r*r!r"rrr s 8 )R#)R)R )RV#TDcNT $sCx. 5O  r"r) rr3s' '&R 6R r"