o :hF/@sUddlmZddlZddlmZmZmZmZmZm Z ddl m m Z ddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZd Zd ZejZeeeefZiejejdejd ejdejej ej dej dej dej ej!ej!dej!dej!dej!ej"ej"dej"dej"dej"ej#ej#dej#dej#ej$ej$dej$dej$dej%dej%dej%ej%ej%dej&d ej&d!ej&ej&ej&i Z'd"e(d#<dPd(d)Z)zej*Z+Wne,yed*Z-dQd0d1Z+Ynwe     dRdSdAdBZ.e     dRdTdEdBZ.     dRdUdIdBZ.e     dRdVdKdLZ/e     dRdWdMdLZ/     dRdXdOdLZ/dS)Y) annotationsN)AnyIterableoverloadTypeVarUnionMapping)protos)get_default_generative_client)#get_default_generative_async_client) helper_types) model_types) text_types) content_typeszmodels/embedding-001dZtask_type_unspecified unspecifiedZretrieval_queryqueryZretrieval_documentdocumentZsemantic_similarityZ similarityZclassificationZ clusteringZquestion_answeringZqaZfact_verificationZ verificationz1dict[EmbeddingTaskTypeOptions, EmbeddingTaskType]_EMBEDDING_TASK_TYPExEmbeddingTaskTypeOptionsreturnEmbeddingTaskTypecCst|tr |}t|SN) isinstancestrlowerr)rr$e/var/www/html/figdemos/bartoux_crm/venv/lib/python3.10/site-packages/google/generativeai/embedding.py to_task_typeHs r&Titerable Iterable[T]nintIterable[list[T]]ccsZ|dkr td|dg}|D]}||t||kr#|Vg}q|r+|VdSdS)NrzKInvalid input: The batch size 'n' must be a positive integer. You entered: z'. Please enter a number greater than 0.) ValueErrorappendlen)r(r*batchitemr$r$r%_batchedTs    r2model model_types.BaseModelNameOptionscontentcontent_types.ContentType task_typeEmbeddingTaskTypeOptions | Nonetitle str | Noneoutput_dimensionality int | Noneclient"glm.GenerativeServiceClient | Nonerequest_options&helper_types.RequestOptionsType | Nonetext_types.EmbeddingDictcCdSr r$r3r5r7r9r;r=r?r$r$r% embed_contentd rD#Iterable[content_types.ContentType]text_types.BatchEmbeddingDictcCrBr r$rCr$r$r%rDprE?content_types.ContentType | Iterable[content_types.ContentType]glm.GenerativeServiceClient8text_types.EmbeddingDict | text_types.BatchEmbeddingDictc sXt|dur i}|durt}r&ttjur&tdddr4dkr4tddr:tt|trt|t t fsdgi}fdd |D}t |t D]'} t j| d } |j| fi|} t| | } |dd d | d Dq[|St jt|d } |j| fi|} t| | } | dd| d<| S)aCalls the API to create embeddings for content passed in. Args: model: Which [model](https://ai.google.dev/models/gemini#embedding) to call, as a string or a `types.Model`. content: Content to embed. task_type: Optional task type for which the embeddings will be used. Can only be set for `models/embedding-001`. title: An optional title for the text. Only applicable when task_type is `RETRIEVAL_DOCUMENT`. output_dimensionality: Optional reduced dimensionality for the output embeddings. If set, excessive values from the output embeddings will be truncated from the end. request_options: Options for the request. Return: Dictionary containing the embedding (list of float values) for the input content. NsInvalid task type: When a title is specified, the task must be of a 'retrieval document' type. Received task type: and title: .rQInvalid value: `output_dimensionality` must be a non-negative integer. Received: embeddingc3*|]}tjt|dVqdSr3r5r7r9r;Nr EmbedContentRequestr to_content.0cr3r;r7r9r$r%  z embed_content..r3requestscs|]}|dVqdSvaluesNr$rWer$r$r%rZ embeddingsrRr`)r make_model_namer r&rRETRIEVAL_DOCUMENTr-r!rr"rr2EMBEDDING_MAX_BATCH_SIZEr BatchEmbedContentsRequestbatch_embed_contentstypeto_dictextendrTrrUrD r3r5r7r9r;r=r?resultr]r0Zembedding_requestZembedding_responseZembedding_dictr$rYr%rD|sZ '   'glm.GenerativeServiceAsyncClient | NonecdSr r$rCr$r$r%embed_content_async rqcrpr r$rCr$r$r%rqrr glm.GenerativeServiceAsyncClientc sft|dur i}|durt}r'ttjur'tdddr5dkr5tddr;tt|trt|t t fsdgi}fdd |D}t |t D]*} t j| d } |j| fi|IdH} t| | } |dd d | d Dq\|St jt|d } |j| fi|IdH} t| | } | dd| d<| S)z?Calls the API to create async embeddings for content passed in.NrKrLrMrrNrOc3rPrQrSrVrYr$r%rZr[z&embed_content_async..r\csr^r_r$rar$r$r%rZ(rcrdrRr`)r rer r&rrfr-r!rr"rr2rgr rhrirjrkrlrTrrUrDrmr$rYr%rqs\      )rrrr)r(r)r*r+rr,)NNNNN)r3r4r5r6r7r8r9r:r;r<r=r>r?r@rrA)r3r4r5rFr7r8r9r:r;r<r=r>r?r@rrG)r3r4r5rHr7r8r9r:r;r<r=rIr?r@rrJ)r3r4r5r6r7r8r9r:r;r<r=ror?r@rrA)r3r4r5rFr7r8r9r:r;r<r=ror?r@rrG)r3r4r5rHr7r8r9r:r;r<r=rsr?r@rrJ)0 __future__r itertoolstypingrrrrrrgoogle.ai.generativelanguageaigenerativelanguageglmgoogle.generativeair google.generativeai.clientr r google.generativeai.typesr r rrZDEFAULT_EMB_MODELrgTaskTyperr+r"rTASK_TYPE_UNSPECIFIEDRETRIEVAL_QUERYrfSEMANTIC_SIMILARITYCLASSIFICATION CLUSTERINGQUESTION_ANSWERINGFACT_VERIFICATIONr__annotations__r&Zbatchedr2AttributeErrorr'rDrqr$r$r$r%s              "    b