o ưi7@s`dZddlZddlmZmZmZmZddlmZddl m Z m Z de fddZ Gd d d Z dS) 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)AnyDictListOptional)verbose_logger)LiteLLM_SkillsTableNewSkillRequestreturncCs^ddl}|}|ddur(t|dtr ||d|d<nt|dtr( tdi|S)zw Convert a Prisma skill record to LiteLLM_SkillsTable. Handles Base64 decoding of file_content field. rN file_content)base64Z model_dumpget isinstancestr b64decodebytesr)Z prisma_skillr datar r `/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/litellm_proxy/skills/handler.py_prisma_skill_to_litellmsrc @seZdZdZeddZe ddedeede fdd Z e dd e d e de e fddZ edede fddZededeeeffddZededee fddZdS)LiteLLMSkillsHandlerz Handler for LiteLLM database-backed skills operations. This class provides static methods for CRUD operations on skills stored in the LiteLLM proxy database (LiteLLM_SkillsTable). cs"ddlm}|durtd|S)z(Get the prisma client from proxy server.r prisma_clientNzRPrisma client is not initialized. Database connection required for LiteLLM skills.)Zlitellm.proxy.proxy_serverr ValueErrorrr r r_get_prisma_client-s z'LiteLLMSkillsHandler._get_prisma_clientNruser_idr cstIdH}dt}||j|j|jd||d}|jdur.ddlm }||j|d<|j durAddl m }| |j |d <|jdurK|j|d <|jdurU|j|d <td |d |j|jjj|dIdH}t|S)z Create a new skill in the LiteLLM database. Args: data: NewSkillRequest with skill details user_id: Optional user ID for tracking Returns: LiteLLM_SkillsTable record NZlitellm_skill_Zcustom)skill_id display_title description instructionssourceZ created_byZ updated_byr) safe_dumpsmetadata)Base64r file_name file_typez%LiteLLMSkillsHandler: Creating skill z with title=)r)rruuiduuid4rrrr!Z*litellm.litellm_core_utils.safe_json_dumpsr r Z prisma.fieldsr"encoder#r$rdebugdblitellm_skillstablecreater)rrrrZ skill_datar r"Z new_skillr r r create_skill9s4       z!LiteLLMSkillsHandler.create_skillrlimitoffsetcsRtIdH}td|d||jjj||ddidIdH}dd|DS) z List skills from the LiteLLM database. Args: limit: Maximum number of skills to return offset: Number of skills to skip Returns: List of LiteLLM_SkillsTable records Nz0LiteLLMSkillsHandler: Listing skills with limit=z , offset=Z created_atZdesc)ZtakeskipordercSsg|]}t|qSr )r).0sr r r sz4LiteLLMSkillsHandler.list_skills..)rrrr(r)r*Z find_many)r.r/rZskillsr r r list_skillsns z LiteLLMSkillsHandler.list_skillsrcsXtIdH}td||jjjd|idIdH}|dur(td|t|S)z Get a skill by ID from the LiteLLM database. Args: skill_id: The skill ID to retrieve Returns: LiteLLM_SkillsTable record Raises: ValueError: If skill not found Nz$LiteLLMSkillsHandler: Getting skill rwhereSkill not found: ) rrrr(r)r* find_uniquerrrrZskillr r r get_skills zLiteLLMSkillsHandler.get_skillcsttIdH}td||jjjd|idIdH}|dur(td||jjjd|idIdH|ddS)z Delete a skill by ID from the LiteLLM database. Args: skill_id: The skill ID to delete Returns: Dict with id and type of deleted skill Raises: ValueError: If skill not found Nz%LiteLLMSkillsHandler: Deleting skill rr6r8Z skill_deleted)idtype) rrrr(r)r*r9rdeleter:r r r delete_skills  z!LiteLLMSkillsHandler.delete_skillc sdz t|IdHWStyYdSty1}ztd|d|WYd}~dSd}~ww)aO Fetch a skill from the database (used by skills injection hook). This is a convenience method that returns None instead of raising an exception if the skill is not found. Args: skill_id: The skill ID to fetch Returns: LiteLLM_SkillsTable or None if not found Nz+LiteLLMSkillsHandler: Error fetching skill z: )rr;r Exceptionrwarning)rer r rfetch_skill_from_dbs z(LiteLLMSkillsHandler.fetch_skill_from_db)N)r-r)__name__ __module__ __qualname____doc__ staticmethodrrrrrr,intrr5r;rr?rCr r r rr%s:   4 r)rGr%typingrrrrZlitellm._loggingrZlitellm.proxy._typesrrrrr r r rs