|
@@ -29,6 +29,7 @@ func (k *KnowledgeService) KnowledgeAdd(param *knowledgeclient.AddRequest, segme
|
|
|
query := map[string]interface{}{"status": 1, "appid": param.AppId, "ent_id": param.EntId}
|
|
|
datalist := Mysql.Find(util.KNOWLEDGE, query, "id", "", -1, -1)
|
|
|
if datalist != nil && *datalist != nil && len(*datalist) > 0 {
|
|
|
+
|
|
|
//问题进行分词
|
|
|
keywords := ""
|
|
|
keywordsArr := util.HanlpGetNormalWords(param.Question, segment)
|
|
@@ -79,10 +80,12 @@ func (k *KnowledgeService) KnowledgeAdd(param *knowledgeclient.AddRequest, segme
|
|
|
}
|
|
|
b := elastic.Save(Index, Type, knowledge)
|
|
|
if !b {
|
|
|
+ logx.Error("es 保存失败")
|
|
|
return false, "es 保存失败"
|
|
|
}
|
|
|
questionVector, err := util.EncodeVector(param.Question)
|
|
|
if err != nil {
|
|
|
+ logx.Error("获取 向量失败 ",err)
|
|
|
return false, "获取 向量失败 :" + err.Error()
|
|
|
}
|
|
|
// 存入向量库
|
|
@@ -97,6 +100,7 @@ func (k *KnowledgeService) KnowledgeAdd(param *knowledgeclient.AddRequest, segme
|
|
|
}
|
|
|
if !ESV7.Save(ESV7Index, ESV7Type, knowledgeV) {
|
|
|
logx.Error("知识库添加向量失败:", knowledgeV)
|
|
|
+ return false, "知识库添加向量失败"
|
|
|
}
|
|
|
return b, ""
|
|
|
}
|