uPjUddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZd Zeed <d Zeed<eddGddZGddee Zy))CallableMapping) dataclass)Final)encode_url_path_segment) GoogleAIStudioInteractionsConfig)validate_vertex_location) VertexBase)VERTEX_CREDENTIALS_TYPES)GenericLiteLLMParams) LlmProvidersv1beta1VERTEX_INTERACTIONS_API_VERSIONglobal$VERTEX_INTERACTIONS_DEFAULT_LOCATIONT)frozenslotscReZdZUeed<eed<eed<edefdZdedefdZy) VertexInteractionsTargetbase_url project_idlocationreturncb|jdtd|jd|jdS)N/z /projects/z /locations/z /interactions)rrrrselfs w/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/vertex_ai/interactions/transformation.pycollection_urlz'VertexInteractionsTarget.collection_urls8}}oQ>?)T]]O= R interaction_idc>t|d}|jd|S)Nr!) field_namer)rr)rr!encoded_interaction_ids rinteraction_urlz(VertexInteractionsTarget.interaction_urls*(?[k(l%%&a(>'?@@r N)__name__ __module__ __qualname__str__annotations__propertyrr%r rrrs@MOM    AcAcAr rc eZdZ ddeedzedzgeeeffdzddffd ZdedzdedzdeeeffdZe de fdZ e defd Z defd Z d edeeeffd Zd edzd edefdZdeeefded edzdefdZ dd edzdedzdedzd eeefdzdedzdef dZ dded ed ededeeeff dZded ed edeeefdeeeff dZded ed edeeefdeeeff dZded ed edeeefdeeeff dZxZS)VertexAIInteractionsConfigNmint_access_tokenrcLt||xs |j|_yN)super__init__#_mint_access_token_with_vertex_base_mint_access_token)rr/ __class__s rr3z#VertexAIInteractionsConfig.__init__$s&   I!I!I r credentialsrc*|j||dS)N vertex_ai)r7rcustom_llm_provider)_ensure_access_token)rr7rs rr4z>VertexAIInteractionsConfig._mint_access_token_with_vertex_base-s$ ((# P[)  r c"tjSr1)r VERTEX_AIrs rr:z.VertexAIInteractionsConfig.custom_llm_provider6s%%%r ctSr1)rrs r api_versionz&VertexAIInteractionsConfig.api_version:s..r ctSr1)rrs rget_default_vertex_locationz6VertexAIInteractionsConfig.get_default_vertex_location>s33r litellm_paramsc|j}|j|j||j|Sr1) model_dumpr5safe_get_vertex_ai_credentialssafe_get_vertex_ai_project)rrB raw_paramss r_mintz VertexAIInteractionsConfig._mintAs?*557 &&  / / ;  + +J 7  r api_basec|j|\}}|s tdt|j|j xst }t |j|xsd|||S)Nz^Vertex AI project is required. Set vertex_project, litellm.vertex_project, or VERTEXAI_PROJECT)rrr)rH ValueErrorr explicit_vertex_ai_locationrDrr get_api_base)rrIrB_rrs r_targetz"VertexAIInteractionsConfig._targetHsz >2 :p 3  , ,^-F-F-H I qMq (&&x'74B!  r headersmodelcV|j|xs t\}}dd|d|S)Nzapplication/jsonzBearer )z Content-Type Authorization)rHr )rrPrQrB access_tokenrNs rvalidate_environmentz/VertexAIInteractionsConfig.validate_environmentWs@ **^%M7K7MN a.&|n5   r agentstreamc|rtj|n t}|j||j}|r|dS|S)Nz?alt=sse)r model_validaterOr)rrIrQrVrBrWparamsrs rget_complete_urlz+VertexAIInteractionsConfig.get_complete_urldsODR / / ?WkWm !% Xv > M M.4.!*H.Hr r! url_suffixc\|j|xsd|}|j||ifSr1)rOr%)rr!rIrBr\targets r_interaction_by_id_requestz5VertexAIInteractionsConfig._interaction_by_id_requestrs9 X%5~F((89*FJJr c(|j|||Sr1r_rr!rIrBrPs r!transform_get_interaction_requestzrysr--!H\H=A5,)22.6$e6 $d#AA$A"ro-Mror