diff --git a/app/model/KnowledgeBase.py b/app/model/KnowledgeBase.py new file mode 100644 index 0000000..0196cde --- /dev/null +++ b/app/model/KnowledgeBase.py @@ -0,0 +1,16 @@ +from datetime import datetime + +from sqlmodel import Field + +from app.model.BasicModel import BasicModel +from app.utils.create_module_service import create_model_service + + +class KnowledgeBaseModel(BasicModel, table=True): + __tablename__ = "pl_knowledge_base" + + code: str = Field(default=None, description="知识库编码") + name: str = Field(default=None, description="知识库名称") + + +KnowledgeBaseService = create_model_service(KnowledgeBaseModel) diff --git a/app/model/KnowledgeDoc.py b/app/model/KnowledgeDoc.py new file mode 100644 index 0000000..1666918 --- /dev/null +++ b/app/model/KnowledgeDoc.py @@ -0,0 +1,17 @@ +from datetime import datetime + +from sqlmodel import Field + +from app.model.BasicModel import BasicModel +from app.utils.create_module_service import create_model_service + + +class KnowledgeDocModel(BasicModel, table=True): + __tablename__ = "pl_knowledge_doc" + + parent_code: str = Field(default=None, description="所属知识库的code") + name: str = Field(default=None, description="文档名称") + path: str = Field(default=None, description="文档原始文件路径") + + +KnowledgeDocService = create_model_service(KnowledgeDocModel) diff --git a/app/server.py b/app/server.py index 8871471..b4dfff6 100644 --- a/app/server.py +++ b/app/server.py @@ -27,6 +27,8 @@ from app.model.ApproveModel import ApproveService from app.model.ConversationModel import ConversationService from app.model.HotelModel import HotelService from app.model.InvoiceModel import InvoiceService +from app.model.KnowledgeBase import KnowledgeBaseService +from app.model.KnowledgeDoc import KnowledgeDocService from app.model.LgApprove import LgApproveService from app.model.LgChat import LgChatService from app.model.LgMessage import LgMessageService @@ -120,6 +122,8 @@ HotelService.add_route(app=app, path="/hotel") OrderService.add_route(app=app, path="/order") InvoiceService.add_route(app=app, path="/invoice") ConversationService.add_route(app=app, path="/conversation") +KnowledgeBaseService.add_route(app=app, path="/knowledge_base") +KnowledgeDocService.add_route(app=app, path="/knowledge_doc") if __name__ == "__main__": run_uvicorn()