tuoling805 1 年之前
父节点
当前提交
5237ea70a9
共有 1 个文件被更改,包括 11 次插入0 次删除
  1. 11 0
      models/db.go

+ 11 - 0
models/db.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"eta/eta_api/models/aimod"
 	"eta/eta_api/models/company"
 	"eta/eta_api/models/data_manage"
 	"eta/eta_api/models/data_manage/excel"
@@ -166,6 +167,9 @@ func init() {
 	// 初始化EXCEL的表
 	initExcel()
 
+	//初始化AI
+	initAi()
+
 	// 初始化部分数据表变量(直接init会有顺序问题=_=!)
 	data_manage.InitEdbSourceVar()
 }
@@ -479,3 +483,10 @@ func initSmartReport() {
 		new(smart_report.SmartReportSaveLog), // 智能研报-保存记录表
 	)
 }
+
+func initAi() {
+	orm.RegisterModel(
+		new(aimod.AiChatTopic),
+		new(aimod.AiChat),
+	)
+}