data_ai/mainT.go

package main

import (
	"data_ai/prompt"
	"data_ai/ul"
	qu "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
)

func mainT() {
	coll := "ai_41411_100"
	q := map[string]interface{}{
		"_id": mongodb.StringTOBsonId("669e907166cf0db42a653c52"),
	}
	sort := map[string]interface{}{
		"num": 1,
	}
	list, _ := ul.SourceMgo.Find(coll, q, sort, nil)
	for _, l := range list {
		detail := qu.ObjToString(l["detail"])
		title := qu.ObjToString(l["title"])
		s_toptype, s_subtype := prompt.AcquireClassInfo(detail, title)
		qu.Debug("ai:", s_toptype, s_subtype)
		s_toptype, s_subtype = prompt.CheckClassByOtherFileds(s_toptype, s_subtype, l) //根据抽取字段校正
		qu.Debug("check:", l["_id"], s_toptype, s_subtype)
		ul.SourceMgo.UpdateById(coll, mongodb.BsonIdToSId(l["_id"]), map[string]interface{}{
			"$set": map[string]interface{}{
				"top": s_toptype,
				"sub": s_subtype,
			},
		})
	}
}