o ՂiH@sddZddlmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZGdddeZGdddeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdd d eZGd!d"d"eZGd#d$d$eZ Gd%d&d&eZ!d'S)(z Knowledge Base models ) ColumnStringTextBooleanInteger ForeignKeyJSONUniqueConstraintFloatDate BigInteger) relationship) BaseModelc@seZdZdZdZeedddZeedddZee ddZ eeddd Z eedd d Z eed d d Z eedd Zeed d Zeed d Zeed d ZeeeddddZeeeddddZeeddd ZedddZdS) KnowledgeBasezU Knowledge Base model Represents a collection of documents and knowledge knowledge_basesdFnullableT2u📚defaultgeneralenabledrpublic.tenants.idrindexpublic.users.id not_builtKnowledgeCategoryknowledge_baseback_populatesN)__name__ __module__ __qualname____doc__ __tablename__rrnamecoder descriptionicontypestatusr is_publicr doc_countqa_count entity_countr tenant_id created_by graph_statusr categoriesr6r6A/lsinfo/ai/hellotax_ai/base_platform/app/models/knowledge_base.pyr s"     rc@seZdZdZdZeeedddZee dddZ ee ddZ eeedddZ ee d ddZee d ddZeed d Zed ddZedddZeddde gdZdS)rzU Knowledge Category model Represents a category for organizing documents knowledge_categoriesknowledge_bases.idTrrFknowledge_categories.idrrrrrr5r!KnowledgeDocumentcategoryzKnowledgeCategory.idchildren remote_sidebackref foreign_keysN)r#r$r%r&r'rrrknowledge_base_idrr(rr* parent_idr+color sort_orderr r documentsparentr6r6r6r7r$s    rc@seZdZdZdZeeeddddZee dddZ ee d ddd Z ee ddZ eeddZeed d Zee d ddZee d ddZedddZdS) TagCategoryzV Tag Category model - represents taxonomy dimensions (Part 1, Part 2, Part 3) tag_categoriesrFTrrrrruniquerrr KnowledgeTagr<r!N)r#r$r%r&r'rrrr2rr(r)rr* part_numberrEr+rDr tagsr6r6r6r7rH9s   rHc@s*eZdZdZdZeeeddddZee ddddZ ee dd Z eeed dddZ eeed dddZee ddddZeeddd Zeedd Zeedd Zeedd Zeedd Zee ddddZee ddddZeedd Zee ddd ZedddZeddde gdZeddddZdS)rLz Knowledge Tag model - supports both hierarchical user tags and predefined taxonomy tags Represents a tag for categorizing documents (tenant-level global) knowledge_tagsrFTrrrJrknowledge_tags.idztag_categories.id)rrrg?ru🏷️)rrrz#6366f1rrHrNr!zKnowledgeTag.idr=r>r; document_tags) secondaryr"N) r#r$r%r&r'rrrr2rr(rr*rCtag_category_idtag_coder is_predefined is_systemrkeywordsr search_weightsearch_priorityr+rDrEr-r r<rGrFr6r6r6r7rLOs2        rLc@seZdZdZdZeeeddddZeeeddddZ ee ddd Z ee dd Z eed d Zeedd Zeed d Zed ddZdS) TagAutoRulez] Tag Auto Rule model - defines rules for automatic tagging based on document content tag_auto_rulesrFTrrPrrg333333?rrrL auto_rulesr@N)r#r$r%r&r'rrrr2tag_idr rule_typer rule_configr confidence_thresholdr is_activepriorityr tagr6r6r6r7rZws    rZc@seZdZdZdZeeddddZeeddZ eeddZ ee e ddddZ ee e d dddZee e d dddZeedddZeed ddZeed ddZeed ddZeed ddZeeddZee ddZeeddZeedddZeed ddZee ddZeeddZeed ddZee ddZeed ddZee ddZ ee ddZ!eed ddZ"ee dddZ#eeddZ$eeddZ%ee ddZ&ee ddZ'ee ddZ(eed ddZ)eedddZ*eedddZ+ee ddZ,ee ddZ-ee.ddZ/ee.ddZ0ee.ddZ1eed ddZ2ee3ddZ4ee e dddddZ5ee3ddZ6eeddZ7ee3ddZ8ee9ddZ:eeddddZ;ee ddZe=d!egd"Z?e=d#d$d%Z@e=d&d'd(d)ZAe=d*d'd(d)ZBd+S),r;zP Knowledge Document model Represents a document in a knowledge base knowledge_documentsFTrrr:rrrrdraftrrrpending automatici recursive)rrknowledge_documents.idzSET NULLondelete@rrFr!UserrArLrQ)rRDocumentVectordocumentzall, delete-orphan)r"cascadeDocumentVersionN)Cr#r$r%r&r'rrtitlercontentsummaryrr category_idr2 author_idsource reference_url file_path file_typer-rr. view_count is_vectorized vector_modelvectorization_statusvectorization_progressvectorization_errorr4r1segmentation_mode chunk_size chunk_overlap splitter_type window_sizeenable_parent_childenable_reference_extractioncharacter_count recall_countdata_center_doc_iddoc_type doc_numberissuing_authoritydoc_number_yeardoc_number_serialr issue_dateeffective_date expire_date doc_statusrsupersedes_doc_idssuperseded_by_doc_id tax_type_tagshas_attachmentattachment_typesr parse_quality_score content_hashversion_numberr r<authorrNvectorsversionsr6r6r6r7r;sn                            r;c@sLeZdZdZdZddiZeeedddddZ eeed ddddZ d Z d S) DocumentTagz1 Document Tag association (Many-to-Many) rQextend_existingTrnCASCADEro) primary_keyrPN) r#r$r%r&r'__table_args__rrr document_idr^idr6r6r6r7rsrc@sVeZdZdZdZeeedddZeeddZ ee ddZ ee ddZ eed ddZeed ddZeed ddd Zeed ddd Zeed ddZeed ddZeed ddZeedddd Zeed ddZeed ddZee ddZeed ddd Zeed ddZeeddZeed ddd ZeedddZ e!dddZ"dS)ruzP Document Vector model Stores vector embeddings for semantic search document_vectorsrnFrrrTrrfalse)server_defaultrrrrrgr;rr!N)#r#r$r%r&r'rrrr chunk_indexr chunk_textrvectorr milvus_id model_nameparent_chunk_idr is_parent chunk_level prev_chunk_id next_chunk_idrrr referenceschunk_idrr issue_date_int chunk_type media_pathr rvr6r6r6r7rus0     ruc@seZdZdZdZeeedddZeeddZ ee dddZ ee ddZ ee ddZeeedddZed d d Zed egd ZdS)rxzD Document Version model Tracks document version history document_versionsrnFrrfTrr;rr!rsrtN)r#r$r%r&r'rrrrrrryrrzchange_summary editor_idr rveditorr6r6r6r7rxs    rxc@seZdZdZdZeeddZeeddZee e dddZ ee ddZ eeddZee ddZeedddZee d dZee d dZeed d dZee e d ddZed ZedegdZdS) KnowledgeQAzA Knowledge QA model Stores question and answer pairs knowledge_qaFrr:TrrrractiverrrsrtN)r#r$r%r&r'rrquestionanswerrrr|rrelated_documentsrrquestion_vectorrr use_count helpful_countr-r3r r<creatorr6r6r6r7r%s        rc@seZdZdZdZeeeddddZee dddZ ee dddZ ee d ddZ ee ddZee d ddZeedd Zeed d ZeeddZed ddZeddddfZdS)KnowledgeMetadataFieldzb Knowledge Metadata Field model Defines metadata field templates for a knowledge base knowledge_metadata_fieldsr9FTrrrrrgrrrmetadata_fieldsr]rB field_keyuq_kb_field_keyr(N)r#r$r%r&r'rrrrBr field_namer field_typer field_options default_valuer is_requiredrErr*r r r rr6r6r6r7rBs      rc@sleZdZdZdZeeeddddZeeeddddZ ee ddZ e d d d Z e d Zed dddfZdS)DocumentMetadataValuezP Document Metadata Value model Stores metadata values for documents document_metadata_valuesrnFTrzknowledge_metadata_fields.idrr;metadata_valuesr]rrfield_id uq_doc_fieldrN)r#r$r%r&r'rrrrrrvaluer rvfieldr rr6r6r6r7r\s   rc@s\eZdZdZdZeeeddddZee dddZ ee ddZ eeddZ eeddZd S) GraphBuildStatusz'Graph build status for a knowledge basegraph_build_statusr9FTrrrN)r#r$r%r&r'rrrrBrr-r error_message node_count edge_countr6r6r6r7rqs  rc@sTeZdZdZdZeeeddddZee dddZ ee d ddZ ee ddZ d S) GraphEntityz5Graph entity (node) stored in PG as a mirror of Neo4jgraph_entitiesr9FTrrrrgN)r#r$r%r&r'rrrkb_idr entity_type entity_namer propertiesr6r6r6r7r|src@sDeZdZdZdZeeeddddZee dddZ ee ddZ d S) KnowledgeBaseMetadataz/Key-value metadata attached to a knowledge baseknowledge_base_metadatar9FTrrkrN) r#r$r%r&r'rrrrBrkeyrrr6r6r6r7rs rc@peZdZdZdZeeeddddZee dddZ ee dddZ ee d ddZ ee dd Zeed d Zd S) KnowledgeBaseMetadataFieldz9Schema definition for metadata fields of a knowledge baseknowledge_base_metadata_fieldsr9FTrrkrrrrNr#r$r%r&r'rrrrBrr field_labelrrrrEr6r6r6r7r rc@r) MetadataFieldz!Generic metadata field definitionrr9FTrrkrrrrNrr6r6r6r7rrrN)"r& sqlalchemyrrrrrrrr r r r sqlalchemy.ormr app.models.baserrrrHrLrZr;rrurxrrrrrrrrr6r6r6r7s*4  (M,