o ZhF/@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_documentZdocumentZsemantic_similarityZ similarityZclassificationZ clusteringZquestion_answeringZqaZfact_verificationZ verificationz1dict[EmbeddingTaskTypeOptions, EmbeddingTaskType]_EMBEDDING_TASK_TYPExEmbeddingTaskTypeOptionsreturnEmbeddingTaskTypecCst|tr |}t|SN) isinstancestrlowerr)rr#T/var/www/html/lang_env/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    r1model model_types.BaseModelNameOptionscontentcontent_types.ContentType task_typeEmbeddingTaskTypeOptions | Nonetitle str | Noneoutput_dimensionality int | Noneclient"glm.GenerativeServiceClient | Nonerequest_options&helper_types.RequestOptionsType | Nonetext_types.EmbeddingDictcCdSrr#r2r4r6r8r:r<r>r#r#r$ embed_contentd rC#Iterable[content_types.ContentType]text_types.BatchEmbeddingDictcCrArr#rBr#r#r$rCprD?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|dVqdSr2r4r6r8r:Nr EmbedContentRequestr to_content.0cr2r:r6r8r#r$  z embed_content..r2requestscs|]}|dVqdSvaluesNr#rVer#r#r$rY embeddingsrQr_)r make_model_namer r%rRETRIEVAL_DOCUMENTr,r rr!rr1EMBEDDING_MAX_BATCH_SIZEr BatchEmbedContentsRequestbatch_embed_contentstypeto_dictextendrSrrTrC r2r4r6r8r:r<r>resultr\r/Zembedding_requestZembedding_responseZembedding_dictr#rXr$rC|sZ '   'glm.GenerativeServiceAsyncClient | NonecdSrr#rBr#r#r$embed_content_async rpcrorr#rBr#r#r$rprq 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.NrJrKrLrrMrNc3rOrPrRrUrXr#r$rYrZz&embed_content_async..r[csr]r^r#r`r#r#r$rY(rbrcrQr_)r rdr r%rrer,r rr!rr1rfr rgrhrirjrkrSrrTrCrlr#rXr$rps\      )rrrr)r'r(r)r*rr+)NNNNN)r2r3r4r5r6r7r8r9r:r;r<r=r>r?rr@)r2r3r4rEr6r7r8r9r:r;r<r=r>r?rrF)r2r3r4rGr6r7r8r9r:r;r<rHr>r?rrI)r2r3r4r5r6r7r8r9r:r;r<rnr>r?rr@)r2r3r4rEr6r7r8r9r:r;r<rnr>r?rrF)r2r3r4rGr6r7r8r9r:r;r<rrr>r?rrI)0 __future__r itertoolstypingrrrrrrZgoogle.ai.generativelanguageZaiZgenerativelanguageZglmZgoogle.generativeair Zgoogle.generativeai.clientr r Zgoogle.generativeai.typesr r rrZDEFAULT_EMB_MODELrfZTaskTyperr*r!rZTASK_TYPE_UNSPECIFIEDZRETRIEVAL_QUERYreZSEMANTIC_SIMILARITYZCLASSIFICATIONZ CLUSTERINGZQUESTION_ANSWERINGZFACT_VERIFICATIONr__annotations__r%Zbatchedr1AttributeErrorr&rCrpr#r#r#r$s              "    b