o j6hF/@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-001dtask_type_unspecified unspecifiedretrieval_queryqueryretrieval_documentdocumentsemantic_similarity similarityclassification clusteringquestion_answeringqafact_verification verificationz1dict[EmbeddingTaskTypeOptions, EmbeddingTaskType]_EMBEDDING_TASK_TYPExEmbeddingTaskTypeOptionsreturnEmbeddingTaskTypecCst|tr |}t|SN) isinstancestrlowerr&)r'r/l/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/embedding.py to_task_typeHs r1Titerable 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)r3r5batchitemr/r/r0_batchedTs    r=model model_types.BaseModelNameOptionscontentcontent_types.ContentType task_typeEmbeddingTaskTypeOptions | Nonetitle str | Noneoutput_dimensionality int | Noneclient"glm.GenerativeServiceClient | Nonerequest_options&helper_types.RequestOptionsType | Nonetext_types.EmbeddingDictcCdSr+r/r>r@rBrDrFrHrJr/r/r0 embed_contentd rO#Iterable[content_types.ContentType]text_types.BatchEmbeddingDictcCrMr+r/rNr/r/r0rOprP?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|dVqdSr>r@rBrDrFNr EmbedContentRequestr to_content.0cr>rFrBrDr/r0  z embed_content..r>requestscs|]}|dVqdSvaluesNr/rber/r/r0re embeddingsr]rk)r make_model_namer r1r*RETRIEVAL_DOCUMENTr8r,rr-rr=EMBEDDING_MAX_BATCH_SIZEr BatchEmbedContentsRequestbatch_embed_contentstypeto_dictextendr_rr`rO r>r@rBrDrFrHrJresultrhr;embedding_requestembedding_responseembedding_dictr/rdr0rO|sZ '   'glm.GenerativeServiceAsyncClient | NonecdSr+r/rNr/r/r0embed_content_async rcr~r+r/rNr/r/r0rr 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.NrVrWrXrrYrZc3r[r\r^rardr/r0rerfz&embed_content_async..rgcsrirjr/rlr/r/r0re(rnror]rk)r rpr r1r*rqr8r,rr-rr=rrr rsrtrurvrwr_rr`rOrxr/rdr0rs\      )r'r(r)r*)r3r4r5r6r)r7)NNNNN)r>r?r@rArBrCrDrErFrGrHrIrJrKr)rL)r>r?r@rQrBrCrDrErFrGrHrIrJrKr)rR)r>r?r@rSrBrCrDrErFrGrHrTrJrKr)rU)r>r?r@rArBrCrDrErFrGrHr}rJrKr)rL)r>r?r@rQrBrCrDrErFrGrHr}rJrKr)rR)r>r?r@rSrBrCrDrErFrGrHrrJrKr)rU)0 __future__r itertoolstypingrrrrrrgoogle.ai.generativelanguageaigenerativelanguageglmgoogle.generativeair google.generativeai.clientr r google.generativeai.typesr r rrDEFAULT_EMB_MODELrrTaskTyper*r6r-r(TASK_TYPE_UNSPECIFIEDRETRIEVAL_QUERYrqSEMANTIC_SIMILARITYCLASSIFICATION CLUSTERINGQUESTION_ANSWERINGFACT_VERIFICATIONr&__annotations__r1batchedr=AttributeErrorr2rOrr/r/r/r0s              "    b