feat: 新增模块knowledge_base

knowledge_doc

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)

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)

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()