lQj`ddlmZmZmZmZddlmZddlmZddl m Z m Z m Z m Z ddlZddlZddlZddl mZerddlZGddeeZy) )EmbeddingFunctionSpace Embeddings Documentsvalidate_config_schema)decode_embedding)ListDictAnyOptionalN) TYPE_CHECKINGc eZdZdZ ddeedededeefdZded e fd Z e d efd Z d e fd Zd ee fd Ze deeefd dfdZd eeeffdZdeeefdeeefd dfdZe deeefd dfdZy)PerplexityEmbeddingFunctionza This class is used to generate embeddings for a list of texts using the Perplexity API. Napi_key model_nameapi_key_env_var dimensionsc ddl}|tjdt t jdd|_n||_|xst j|j|_ |jstd|jd||_ ||_ |j|j|_ y#t$r tdwxYw) a Initialize the PerplexityEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the Perplexity API. Defaults to "PERPLEXITY_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "pplx-embed-v1-0.6b". api_key (str, optional): API key for the Perplexity API. If not provided, will look for it in the environment variable. dimensions (int, optional): Perplexity embeddings support Matryoshka representation learning, allowing you to reduce embedding dimensions while maintaining quality. rNzcThe perplexityai python package is not installed. Please install it with `pip install perplexityai`zDirect api_key configuration will not be persisted. Please use environment variables via api_key_env_var for persistent storage.PERPLEXITY_API_KEYzThe z! environment variable is not set.)r) perplexity ImportError ValueErrorwarningswarnDeprecationWarningosgetenvrrrr Perplexity_client)selfrrrrrs /home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/perplexity_embedding_function.py__init__z$PerplexityEmbeddingFunction.__init__s&     MM_"  99) * 6#7D #2D A"))D,@,@"A ||t++,,MN %$,z,,T\\B 3 u  s CCinputreturnc|jjj||j|j}|j Dcgc]}t |jc}Scc}w)z Generate embeddings for the given documents. Args: input: Documents to generate embeddings for. Returns: Embeddings for the documents. )r$modelr)r embeddingscreaterrdatar embedding)r!r$responseembs r"__call__z$PerplexityEmbeddingFunction.__call__CsY<<**11//2  rUs=NNME,,  z5"3I">z5r1