o "i$@srdZddlmZmZddlmZddlmZddlm Z m Z ddl m Z m Z Gdddee e e fZee Zd S) z" CRUD operations for Agent model. )OptionalList)Session)CRUDBase)Agent ChatMessage) AgentCreate AgentUpdatec@seZdZdZdededeefddZdddded e d eed eedeef d d Z ddddeded ededede ef ddZ ddddded ed eededede ef ddZ dddeded ede dedeef ddZd dd!deded"ededeef d#d$ZdS)% CRUDAgentz* CRUD operations for Agent model. dbidreturncCs|j||dS)z5Alias for delete() method for backward compatibility.)r )delete)selfr r r6/lsinfo/ai/hellotax_ai/base_platform/app/crud/agent.pyremoveszCRUDAgent.removeN) created_by tenant_idnamerrcCsT|ttj|k}|r|tj|k}|dur&ttdr&|tj|k}|S)a Get agent by name. Args: db: Database session name: Agent name created_by: Optional user ID to filter by creator tenant_id: Optional tenant ID for filtering Returns: Agent instance or None Nr)queryrfilterrrhasattrrfirst)rr rrrrrrr get_by_names zCRUDAgent.get_by_namerd)skiplimitmodel_idrrcCs.|ttj|ktj|k||S)aK Get agents using a specific model. Args: db: Database session model_id: Model ID tenant_id: Tenant ID for filtering skip: Number of records to skip limit: Maximum number of records to return Returns: List of agent instances )rrrrroffsetrall)rr rrrrrrr get_by_model3s zCRUDAgent.get_by_model)rrrcCsF|ttjdktj|k}|r|tj|k}|||S)aa Get active agents. Args: db: Database session tenant_id: Tenant ID for filtering created_by: Optional user ID to filter by creator skip: Number of records to skip limit: Maximum number of records to return Returns: List of active agent instances active) rrrstatusrrrrr )rr rrrrrrrrget_active_agentsNs zCRUDAgent.get_active_agentsT)commitagent_idr#r%cCsT|ttj|ktj|k}|sdS||_|||r(|| ||S)a\ Update agent status. Args: db: Database session agent_id: Agent ID tenant_id: Tenant ID for filtering status: New status (active, inactive, etc.) commit: Whether to commit the transaction Returns: Updated agent instance or None if not found N) rrrr rrr#addr%refresh)rr r&rr#r%agentrrr update_statusns   zCRUDAgent.update_statusF)softr%r+cCsz|ttj|k}|sdS|ttj|kjdd|r0ttdr0d|_ | |n|||r;| |S)am Delete an agent and all associated chat messages. Args: db: Database session id: Agent ID to delete soft: Whether to perform soft delete (set is_deleted=True) commit: Whether to commit the transaction (default: True) Returns: Deleted agent instance or None if not found NF)synchronize_session is_deletedT) rrrr rrr&rrr-r'r%)rr r r+r%r)rrrrs  zCRUDAgent.delete)__name__ __module__ __qualname____doc__rintrrrstrrrr!r$boolr*rrrrrr s  $   ' ,r N)r1typingrrsqlalchemy.ormr app.crud.baserapp.models.agentrrapp.schemas.agentrr r r)rrrrs   5