5P@i VUddlmZddlZddlmZmZmZddlZddlmZm Z m Z m Z mZddl m cmZddlmZddlmZddlmZddlmZd Zd Zej0Ze eeefZiej:ej:dej:d ej:d ej:ej<ej<d ej<dej<dej<ej>ej>dej>dej>dej>ej@ej@dej@dej@dej@ejBejBdejBdejBejDejDdejDdejDiZ#de$d<d"dZ% ejLZ'e d$ d%dZ*e d$ d&d Z* d$ d'd!Z*y#e($re dZ)d#dZ'YWwxYw)() annotationsN)IterableSequenceMapping)roverloadTypeVarUnionr)get_default_generative_client) text_types) model_types) content_typeszmodels/embedding-001dtask_type_unspecified unspecifiedretrieval_queryqueryretrieval_documentdocumentsemantic_similarity similarityclassification clusteringz1dict[EmbeddingTaskTypeOptions, EmbeddingTaskType]_EMBEDDING_TASK_TYPEcTt|tr|j}t|SN) isinstancestrlowerr)xs g/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/embedding.py to_task_typer&?s"!S GGI  ""Tc#K|dkrtd|g}|D](}|j|t||k(s#|g}*|r|yyw)Nrz Batch size `n` must be >0, got: ) ValueErrorappendlen)iterablenbatchitems r%_batchedr1Ksd q5?sCD D D LL 5zQ    K s :A A cyr modelcontent task_typetitleclients r% embed_contentr:Yr'cyr r3r4s r%r:r:dr;r'ctj| t}r&ttj ur t dr tt|trt|ttfsdgi}fd|D}t|tD]b}tj|}|j|} t!| j#| } |dj%d| dDd|Stj&t)j*|}|j-|} t!| j#| } | dd| d<| S) a Calls the API to create embeddings for content passed in. Args: model: Which model 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`. Return: Dictionary containing the embedding (list of float values) for the input content. zIIf a title is specified, the task must be a retrieval document type task. embeddingc3vK|]0}tjtj|2yw)r5r6r7r8N)glmEmbedContentRequestr to_content).0cr5r7r8s r% z embed_content..s@   # #]%=%=a%@I]b   s69)r5requestsc3&K|] }|d yw)valuesNr3)rDes r%rFz embed_content..s&Yqq{&Ys embeddingsr@rI)r make_model_namer r&EmbeddingTaskTypeRETRIEVAL_DOCUMENTr*r!rr"rr1EMBEDDING_MAX_BATCH_SIZErABatchEmbedContentsRequestbatch_embed_contentstypeto_dictextendrBr rCr:) r5r6r7r8r9resultrGr/embedding_requestembedding_responseembedding_dicts ` `` r%r:r:osq,  ' ' .E ~.0 i(0A0T0TT W   + '8$Z#w-Pr"    h(@A ZE # = =ETY Z !'!PQN ;  & &&YN<rtsK#77>>**D013*LL c+-T-TK..K.DDK$:: K %%'8'H'H K (( K(88K  . .K((*;*N*NK++K+>>K!44K))+<+P+PK,,K,@@K #77!K"$$&7&F&F#K$'''66  "3">">###..-KG4#   H$ 2615  + &/   /     2615  + 0/   /  #  26*. 8 +8 L8/8  8 ( 8 > 8O A s HH('H(