uPjUdZddlZddlmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZdd lmZd Zeed <ed d Zeed<de fdZGddZy)z Handler for LiteLLM database-backed skills operations. This module contains the actual database operations for skills CRUD. Used by the transformation layer and skills injection hook. N)AnyFinal)verbose_logger) InMemoryCache)LITELLM_SKILL_ID_PREFIX)LiteLLM_SkillsTableNewSkillRequestUserAPIKeyAuth) get_primary_resource_owner_scopeget_resource_owner_scopesis_proxy_adminuser_can_access_resource_owner)SkillsRepository__litellm_skill_not_found___NEGATIVE_SKILL_SENTINELi'<)max_size_in_memory default_ttl _SKILL_CACHEreturncddl}|j}|jd*t|dtr|j |d|d<t di|S)uConvert a Prisma skill record to LiteLLM_SkillsTable. Handles Base64 decoding of file_content field — model_dump() converts Base64 fields to base64-encoded strings. rN file_content)base64 model_dumpget isinstancestr b64decoder) prisma_skillrdatas n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/litellm_proxy/skills/handler.py_prisma_skill_to_litellmr#!s[ ))+D xx+ d>*C 0#)#3#3D4H#ID  & &&c 2eZdZdZedZe ddededzdedzde fdZ e dd e d e dedzde e fd Z ed ededzfd Ze dd ededzde fdZe dd ededzdeeeffdZe dd ededzde dzfdZy)LiteLLMSkillsHandlerz2CRUD for skills stored in ``litellm_skillstable``.c4Kddlm}| td|Sw)Nr prisma_clientzRPrisma client is not initialized. Database connection required for LiteLLM skills.)litellm.proxy.proxy_serverr) ValueErrorr(s r"_get_prisma_clientz'LiteLLMSkillsHandler._get_prisma_client5s!<  qr rsNr!user_iduser_api_key_dictrcKtjd{}ttj}t |xs|}| t d||j|j|jd||d}|jddl m }||j|d<|j$ddlm}|j!|j|d<|j"|j"|d <|j$|j$|d <t'j(d ||jt+|j,j/| d{} t1| S7K7w) Nz?Unable to record skill ownership: caller has no identity scope.custom)skill_id display_title description instructionssource created_by updated_byr) safe_dumpsmetadata)Base64r file_name file_typez5LiteLLMSkillsHandler: Creating skill %s with title=%s)r!)r&r,ruuiduuid4r r+r2r3r4r9*litellm.litellm_core_utils.safe_json_dumpsr8r prisma.fieldsr:encoder;r<rdebugrtablecreater#) r!r-r.r)r1owner skill_datar8r: new_skills r" create_skillz!LiteLLMSkillsHandler.create_skill=sR &:%L%L%NN 45djjl^D78IJUg = ^_ _!!//++ ---  == $ M%/ %>Jz "    ( ,)/t7H7H)IJ~ & >> %&*nnJ{ # >> %&*nnJ{ #TV^`d`r`rs!1-!@!F!F!M!MS]!M!^^ ' 22O OL_s"E*E%D:E*E(E*(E*limitoffsetchKtjd{}tjd||||ddid}|#t |st |}|sgSdd|ii|d<t |jjdi|d{}|Dcgc] }t|c}S77cc}ww) Nz=LiteLLMSkillsHandler: Listing skills with limit=%s, offset=%s created_atdesc)takeskiporderr6inwherer) r&r,rrBr r rrC find_manyr#)rIrJr.r)find_many_kwargs owner_scopesskillsss r" list_skillsz LiteLLMSkillsHandler.list_skillsls &:%L%L%NN \^cekl"F+3   (@Q1R";wul\`?act u0 ;< <'.. Hs/A*A(7A*cKtjd{}tjd|tj |d{}|t t |dd|std|t|jjd|id{tj|t|ddS77{7'w)Nz'LiteLLMSkillsHandler: Deleting skill %sr6rbr1rZ skill_deleted)idtype)r&r,rrBr`rrcr+rrCdeleterr]r)r1r.r)r_s r" delete_skillz!LiteLLMSkillsHandler.delete_skills &:%L%L%NN FQ1==hGG = >wul\`?act u0 ;< <}-33::*hAW:XXXx)AB88 OH Ys4C C2C  C AC #C $$C  C  C cK tj||d{S7#t$rYyt$r!}t j d||Yd}~yd}~wwxYww)zSkills-injection-hook helper: returns None instead of raising on not-found / not-authorized so the hook can silently skip.)r.Nz1LiteLLMSkillsHandler: Error fetching skill %s: %s)r&rdr+ Exceptionrwarning)r1r.es r"fetch_skill_from_dbz(LiteLLMSkillsHandler.fetch_skill_from_dbsZ -77Te7ff ff    " "#VX`bc d s@A%#%A% AAAAAAA)NN)rN)N)__name__ __module__ __qualname____doc__ staticmethodr,r rr rrHintlistrXrr`rddictrjrorr$r"r&r&2s<#37,3,3t,3*D0,3  ,3,3\37===*D0= ! " ==. C C$J  48 / /)D0 /  / /4899)D09 c3h99 48  )D0  t #  r$r&)rtr=typingrrlitellm._loggingrlitellm.caching.in_memory_cacher+litellm.llms.litellm_proxy.skills.constantsrlitellm.proxy._typesrr r -litellm.proxy.common_utils.resource_ownershipr r r r'litellm.repositories.table_repositoriesrr__annotations__rr#r&rr$r"rsi +9OUU E#@%?#u"M eM'.A'"OOr$