tPjUdZddlmZmZmZddlZddlmZddlZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZd d lmZGd deZGdde ZeZeed<GddeZy)z/ Humanloop integration https://humanloop.com/ )AnyFinalcastN) TypedDict) DualCache)_get_httpx_client)get_secret_str)AllMessageValues) PromptSpec)StandardCallbackDynamicParams) CustomLoggercReZdZUeed<eeed<edzed<eeefdzed<y)PromptManagementClient prompt_idprompt_templateNmodeloptional_params) __name__ __module__ __qualname__str__annotations__listr dictrc/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/humanloop.pyrrs0N*++ :#s(^d**rrceZdZedZdededzfdZdee de ee fdee fdZ ded edefd Z ded edefd Zdee de dzdee fd Zd ededefdZy)HumanLoopPromptManagercy)N humanloopr)selfs rintegration_namez'HumanLoopPromptManager.integration_namesrhumanloop_prompt_idreturnNcHttdz|j|S)N)key)rr get_cache)r#r%s r_get_prompt_from_id_cachez0HumanLoopPromptManager._get_prompt_from_id_cache#s!*T14>>FY>3Z[[rrprompt_variablescg}|D]o}|jd}|rIt|tr9|jddjdd}|jdi|}||d<|j |q|S)ac Helper function to compile the prompt by substituting variables in the template. Args: prompt_template: List[AllMessageValues] prompt_variables (dict): A dictionary of variables to substitute into the prompt template. Returns: list: A list of dictionaries with variables substituted. contentz{{{z}}}r)get isinstancerreplaceformatappend)r#rr+compiled_promptstemplatetcformatted_templatecompiled_contents r_compile_prompt_helperz-HumanLoopPromptManager._compile_prompt_helper&s;=' .Hi(BjS)%'ZZc%:%B%B4%M"#<#5#<#<#P?O#P &6#  # #H -  . rhumanloop_api_keycDt}d|}|j||dd} |j|j}|d}t|tr|g} n*t|tr|} ntdt||d} i} |jD]\} } | tj vs| | | <t#|t%tt&| | | S#tj$r'}t d|j jd}~wwxYw) Nz%https://api.humanloop.com/v5/prompts/zapplication/json)z X-Api-Keyz Content-Type)urlheadersz%Error getting prompt from Humanloop: r6zInvalid template message type: r)rrrr)rr0raise_for_statushttpxHTTPStatusError Exceptionresponsetextjsonr1rr ValueErrortypeitemslitellmOPENAI_CHAT_COMPLETION_PARAMSrrr )r#r%r;clientbase_urlrCe json_responsetemplate_messagetemplate_messagestemplate_modelrkvs r_get_prompt_from_id_apiz.HumanLoopPromptManager._get_prompt_from_id_api?sD)+ABUAVW **. 2%  W  % % ' (}} "/ "; & -!1 2  ($ / 0 >tDT?U>VWX X -g 6!#!'') 'DAqG999%&" '&) &6!79JK +   !$$ WCAJJOOCTUV V WsC%%D8"DDc|j|}|4|j||}|j||tj|S)N)r(valuettl)r*rT set_cacherI"HUMANLOOP_PROMPT_CACHE_TTL_SECONDS)r#r%r;prompts r_get_prompt_from_idz*HumanLoopPromptManager._get_prompt_from_idesS//0CD >112EGXYF NN'>>    rc8d}|i}|j||}|S)Nrr+)r:)r#rr+compiled_prompts rcompile_promptz%HumanLoopPromptManager.compile_promptps8 .2  #! 55+-6  rprompt_management_clientrcT|d|dS|j|jddS)Nr/)r2r$)r#r`rs r_get_model_from_promptz-HumanLoopPromptManager._get_model_from_prompts7 #G , 8+G4 4==D$9$9#:!!-#@#  2AAR1S1YWY'-  - -- " 55fk5l&(BBBr)NNNFF)rrrrrr rr r intbooltuplert __classcell__)rxs@rrhrhs*.#'%)38=B4C4C'(4C! 4C : 4C + 4C"?4C $&4CDj4Cd 4C&*D[4C04d{4C      4C4Crrh)__doc__typingrrrr@typing_extensionsrrIlitellm.cachingr&litellm.llms.custom_httpx.http_handlerrlitellm.secret_managers.mainr litellm.types.llms.openair "litellm.types.prompts.init_promptsr litellm.types.utilsr custom_loggerrrr rfrrhrrrrsj $# '%D769='+Y+gBYgBT/005Cl5Cr