ugG|ddlZddlmZmZmZddlZddlmZmZm Z eje Z GddeeZ y)N)ListcastUnion) DocumentsEmbeddingFunction Embeddingsc2eZdZdZd dedefdZdedefdZy) JinaEmbeddingFunctionz This class is used to get embeddings for a list of texts using the Jina AI API. It requires an API key and a model name. The default model name is "jina-embeddings-v2-base-en". api_key model_namec||_d|_tj|_|jj j d|ddy)a Initialize the JinaEmbeddingFunction. Args: api_key (str): Your API key for the Jina AI API. model_name (str, optional): The name of the model to use for text embeddings. Defaults to "jina-embeddings-v2-base-en". z!https://api.jina.ai/v1/embeddingszBearer identity) AuthorizationzAccept-EncodingN) _model_name_api_urlhttpxClient_sessionheadersupdate)selfr r s o/var/www/openai/venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/jina_embedding_function.py__init__zJinaEmbeddingFunction.__init__sG&;    $$ 'y1j Q inputreturnc|jj|j||jdj }d|vrt |d|d}t |d}tt|Dcgc]}|d c}Scc}w)a Get the embeddings for a list of texts. Args: texts (Documents): A list of texts to get embeddings for. Returns: Embeddings: The embeddings for the texts. Example: >>> jina_ai_fn = JinaEmbeddingFunction(api_key="your_api_key") >>> input = ["Hello, world!", "How are you?"] >>> embeddings = jina_ai_fn(input) )rmodel)jsondatadetailc |dS)Nindex)es rz0JinaEmbeddingFunction.__call__..9sQwZr)key embedding) rpostrrr RuntimeErrorsortedrr)rrresp embeddingssorted_embeddingsresults r__call__zJinaEmbeddingFunction.__call__ s }}!! MM%$:J:J K" $&   tH~. .?CF| #:3GHJCT UCT !4CT UVV Us6 B N)zjina-embeddings-v2-base-en) __name__ __module__ __qualname____doc__strrrrr0r$rrr r s0      WiWJWrr )loggingtypingrrrrchromadb.api.typesrrr getLoggerr1loggerr r$rrr;s=$$ GG   8 $1W-i81Wr