lQjfddlZddlmZddlmZddlmZmZmZm Z ddl m Z ddl Z GddeZ y)N)OpenAIEmbeddingFunctionvalidate_config_schema)DictAnyOptionalList)SpaceceZdZ ddeededefdZedefdZdefdZ de efdZ de ee ffd Zed e ee fddfd Zed e ee fdd fd Zy )BasetenEmbeddingFunctionapi_keyapi_baseapi_key_env_varc ddl}|tjdt t jdd|_n||_|xst j|j}|std|jd||_ |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.BASETEN_API_KEYzAPI 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 ValueErrorwarningswarnDeprecationWarningosgetenvrr r model_name dimensionsOpenAIclient)selfr rrrresolved_api_keys /home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/baseten_embedding_function.py__init__z!BasetenEmbeddingFunction.__init__ s     MM_"  99& ' 3#4D #2D #Ebii0D0D&E+D,@,@+AAbc ( FG G  3mmDLL4==mQ = i  s CC/returncy)Nbasetenr&r!namezBasetenEmbeddingFunction.name;sr'cy)Ncosiner&rs r! default_spacez&BasetenEmbeddingFunction.default_space?sr'c gdS)N)r*l2ipr&r+s r!supported_spacesz)BasetenEmbeddingFunction.supported_spacesBs%%r'c4|j|jdS)Nrrr2r+s r! get_configz#BasetenEmbeddingFunction.get_configEs 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. rrNzVMissing 'api_key_env_var' or 'api_base' in configuration for BasetenEmbeddingFunction.zAMissing 'api_base' in configuration for BasetenEmbeddingFunction.)r rr)getrr )r4rrs r!build_from_configz*BasetenEmbeddingFunction.build_from_configHsl!**%67::j)  "h&6h   S (+  r'Nct|dy)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)r4s r!validate_configz(BasetenEmbeddingFunction.validate_configis vy1r')CHROMA_BASETEN_API_KEY)__name__ __module__ __qualname__rstrr" staticmethodr(r r,r r0rrr3r7r9r&r'r!r r s 8 -R#-R-R -R^#u&$u+&TDcNT $sCx. 5O  @ 2S#X 24 2 2r'r )rrDs. N,,$i26i2r'