o "i^@sndZddlmZmZddlmZddlmZddlm Z ddl m Z m Z Gdddee e e fZ e e Zd S) z* CRUD operations for KnowledgeBase model. )OptionalList)Session)CRUDBase) KnowledgeBase)KnowledgeBaseCreateKnowledgeBaseUpdatec@sReZdZdZddddededeedeedeef d d Z ded edeefd d Z dddededeedeefddZ dddddededeededede ef ddZ dddddededeededede ef ddZdddddededeededede ef ddZdd d!ded"ededeed#edeef d$d%ZdS)&CRUDKnowledgeBasez2 CRUD operations for KnowledgeBase model. N) created_by tenant_iddbnamer r returncCsJ|ttj|k}|r|tj|k}|dur!|tj|k}|S)aC Get knowledge base by name. Args: db: Database session name: Knowledge base name created_by: Optional user ID to filter by creator tenant_id: Optional tenant ID for multi-tenant isolation Returns: KnowledgeBase instance or None N)queryrfilterr r r first)selfr r r r rr?/lsinfo/ai/hellotax_ai/base_platform/app/crud/knowledge_base.py get_by_names zCRUDKnowledgeBase.get_by_nameidcCs|j||dS)z5Alias for delete() method for backward compatibility.)r)delete)rr rrrrremove/szCRUDKnowledgeBase.remove)r codecCs6|ttj|k}|dur|tj|k}|S)a Get knowledge base by code. Args: db: Database session code: Knowledge base code tenant_id: Optional tenant ID for multi-tenant isolation Returns: KnowledgeBase instance or None N)rrrrr r)rr rr rrrr get_by_code3szCRUDKnowledgeBase.get_by_coderd)r skiplimitkb_typerrcCB|ttj|k}|dur|tj|k}|||S)an Get knowledge bases by type. Args: db: Database session kb_type: Knowledge base type tenant_id: Optional tenant ID for multi-tenant isolation skip: Number of records to skip limit: Maximum number of records to return Returns: List of knowledge base instances N)rrrtyper offsetrall)rr rr rrrrrr get_by_typeLzCRUDKnowledgeBase.get_by_typestatuscCr)aq Get knowledge bases by status. Args: db: Database session status: Knowledge base status tenant_id: Optional tenant ID for multi-tenant isolation skip: Number of records to skip limit: Maximum number of records to return Returns: List of knowledge base instances N)rrrr%r r!rr")rr r%r rrrrrr get_by_statusir$zCRUDKnowledgeBase.get_by_status search_termcCsL|ttjd|d}|dur|tj|k}|||S)av Search knowledge bases by name. Args: db: Database session search_term: Search term tenant_id: Optional tenant ID for multi-tenant isolation skip: Number of records to skip limit: Maximum number of records to return Returns: List of matching knowledge base instances %N) rrrr liker r!rr")rr r'r rrrrrrsearch_by_names z CRUDKnowledgeBase.search_by_nameT)r commitkb_idr+cCsh|ttj|k}|dur|tj|k}|}|sdS||_|||r2|| ||S)aq Update knowledge base status. Args: db: Database session kb_id: Knowledge base ID status: New status tenant_id: Optional tenant ID for multi-tenant isolation commit: Whether to commit the transaction Returns: Updated knowledge base instance or None if not found N) rrrrr rr%addr+refresh)rr r,r%r r+rkbrrr update_statuss  zCRUDKnowledgeBase.update_status)__name__ __module__ __qualname____doc__rstrrintrrrrrr#r&r*boolr0rrrrr s     " " %r N)r4typingrrsqlalchemy.ormr app.crud.baserapp.models.knowledge_baserapp.schemas.knowledge_baserrr knowledge_baserrrrs    E