uPj3dZddlmZmZddlmZmZmZddlm Z ddl m Z m Z m Z ddlmZerddlmZddlmZmZGd d Zy ) a  Transformation handler for LiteLLM database-backed skills. This module provides the SDK-level transformation layer that converts API requests to database operations via LiteLLMSkillsHandler. Pattern follows litellm/llms/litellm_proxy/responses/transformation.py ) CoroutineSequence) TYPE_CHECKINGFinalOptional) JsonValue)DeleteSkillResponseListSkillsResponseSkill) LlmProviders)Logging)LiteLLM_SkillsTableUserAPIKeyAuthc!eZdZdZedefdZ d"dedzdedzdedzdeedzd e dzd edzd edzd e ee fdzd edzde dde de ddedzdeeeeefzfdZ d#dedzdedzdedzd e dzd edzd edzd e ee fdzd edzde ddefdZ d$dedede de ddedzde ddeeeeefzfdZ d%dedede ddefdZ d&dede de ddedzde ddeeeeefzf dZ d'dede ddefdZ d&dede de ddedzde ddeeeeefzf dZ d'dede ddefdZdd defd!Zy)("LiteLLMSkillsTransformationHandlerz Transformation handler for skills API requests to LiteLLM database operations. This is used when custom_llm_provider="litellm_proxy" to store/retrieve skills from the LiteLLM proxy database instead of calling an external API. returnc6tjjS)z%Return the provider name for logging.)r LITELLM_PROXYvalue)selfs u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/litellm_proxy/skills/transformation.pycustom_llm_providerz6LiteLLMSkillsTransformationHandler.custom_llm_provider#s))///N display_title description instructionsfiles file_content file_name file_typemetadatauser_iduser_api_key_dictr _is_async logging_objLiteLLMLoggingObjlitellm_call_idc | r$| j|dd|id| i|j|rb|s`t|trPt |dkDrB|d}t|t r-t |dk\r|d}|d}t |dkDr|dnd}| r|j |||||||| | Sddl}|jj|j |||||||| | S) a Create a skill in LiteLLM database. Args: display_title: Display title for the skill description: Description of the skill instructions: Instructions/prompt for the skill files: Files to upload - list of tuples (filename, content, content_type) file_content: Binary content of skill files (alternative to files) file_name: Original filename (alternative to files) file_type: MIME type (alternative to files) metadata: Additional metadata user_id: User ID for tracking _is_async: Whether to return a coroutine Returns: Skill object or coroutine that returns Skill Nrr'kwargsmodeloptional_paramslitellm_paramsrrzapplication/zip) rrrrrr r!r"r#) update_from_kwargsr isinstancelistlentuple_async_create_skillasyncioget_event_looprun_until_complete)rrrrrrrr r!r"r#r$r%r'r* first_filer6s rcreate_skill_handlerz7LiteLLMSkillsTransformationHandler.create_skill_handler(s!H   * *!0- @ 1?C$($<$< +  %&3u:>$)!H j%0S_5I *1 I#-a=L14Z11D 1 J[I +++'))##!"3,   %%'::  $ $+'))##!"3 %   rc Kddlm} ddlm} | |||||||} | j | || d{} |j | S7w)z%Async implementation of create_skill.rLiteLLMSkillsHandler)NewSkillRequest)rrrrrr r!)datar"r#N))litellm.llms.litellm_proxy.skills.handlerr=litellm.proxy._typesr> create_skill_db_skill_to_response)rrrrrrr r!r"r#r=r> skill_requestdb_skills rr5z6LiteLLMSkillsTransformationHandler._async_create_skill|sl S8.'#%% !5 A A/!B!   ))(33  s3A A A limitoffsetc |r%|j|d||dd|i|j|r|j|||Sddl}|j j |j|||S)a List skills from LiteLLM database. Args: limit: Maximum number of skills to return offset: Number of skills to skip _is_async: Whether to return a coroutine logging_obj: LiteLLM logging object litellm_call_id: Call ID for logging Returns: ListSkillsResponse or coroutine that returns ListSkillsResponse N)rFrGr'r)rFrGr#r)r0r_async_list_skillsr6r7r8) rrFrGr$r%r'r#r*r6s rlist_skills_handlerz6LiteLLMSkillsTransformationHandler.list_skills_handlers0   * **/6 B 1?C$($<$< +  **"3+  %%'::  # #"3 $   rcKddlm}|j|||d{}|Dcgc]}|j|}}t |t ||k\dS7 >/!?!  ))(33  s 757c |r$|j|dd|id|i|j|r|j||Sddl}|j j |j||S)ai Delete a skill from LiteLLM database. Args: skill_id: The skill ID to delete _is_async: Whether to return a coroutine logging_obj: LiteLLM logging object litellm_call_id: Call ID for logging Returns: DeleteSkillResponse or coroutine that returns DeleteSkillResponse NrSr'r)rUr)r0r_async_delete_skillr6r7r8rWs rdelete_skill_handlerz7LiteLLMSkillsTransformationHandler.delete_skill_handler%s,   * *!+X 6 1?C$($<$< +  ++!"3,  %%'::  $ $!"3 %   rcKddlm}|j||d{}t|d|j ddS7$w) z%Async implementation of delete_skill.rr<rUNidtype skill_deleted)r_r`)r@r= delete_skillr get)rrSr#r=results rr\z6LiteLLMSkillsTransformationHandler._async_delete_skillSsV S2??/@  #d|FO4   sAA%ArErc |jr|jjnd}|jr|jjnd}t|j|||j |j |jxsddS)z Convert a database skill record to Anthropic-compatible Skill response. Args: db_skill: LiteLLM_SkillsTable record Returns: Skill object customskill)r_ created_at updated_atrlatest_versionsourcer`)ri isoformatrjr rSrrkrl)rrErirjs rrCz8LiteLLMSkillsTransformationHandler._db_skill_to_responsees@H?R?RH//99;XZ ?G?R?RH//99;XZ   !!"00#22??.h  r) NNNNNNNNNNFNN) NNNNNNNNN)rFNNN)rnrN)FNNN)N)__name__ __module__ __qualname____doc__propertystrrrobjectbytesdictrrboolr rr:r5intr rKrJrXrVr r]r\rCrrrrs0S00 %)"&#')-%) $ $04"8<59&*R TzR 4ZR Dj R $& R dl R :R :R sI~&-R tR $$45R R 12R tR 66501 1!R l%)"&#'%) $ $04"8<!4Tz!44Z!4Dj !4 dl !4 : !4:!4sI~&-!4t!4$$45!4 !4J59&*8<0 0 0  0 12 0 t 0 $$450  i8J(JK K0 h8<    $$45    4 59&*8< , , , 12 , t , $$45 ,  66501 1, b9= 4 4$$45 4  4$ 59&*8< , , , 12 , t , $$45 ,  y9L)LM M, b9=  $$45    $ .C  rrN)rrcollections.abcrrtypingrrrpydanticr#litellm.types.llms.anthropic_skillsr r r litellm.types.utilsr *litellm.litellm_core_utils.litellm_loggingr r&rArrrrzrrrs<011 -WH_ _ r