Havoc412
ae7edb5e8d
- 重构了 rag_controller.go 中的逻辑，使用新的 DocumentHub 结构 - 修改了 encounter.go 中的 Encounter 结构，增加了 explain 标签 - 重写了 rag_websocket.go 中的逻辑，使用新的 DocumentHub 结构 - 新增了 curd_es/encounter_es_curd.go 文件，实现了 Encounter 的 CURD 操作 - 更新了 nlp/func.go 中的 ChatRAG 函数，使用新的 DocumentHub 结构 - 新增了 curd/docs_hub.go 文件，实现了 DocumentHub 的 TopK 方法 - 新增了 utils/data_explain/data_explain_rag.go 文件，实现了结构体到解释字符串的转换
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package test
|
||
|
||
import (
|
||
"fmt"
|
||
"reflect"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// EncounterResult2 结构体定义
|
||
type EncounterResult2 struct {
|
||
Id int64 `json:"id" explain:"路遇笔记ID"`
|
||
Title string `json:"title" explain:"路遇笔记标题"`
|
||
Content string `json:"content" explain:"路遇笔记内容"`
|
||
UpdatedAt *time.Time `json:"updated_at" explain:"最后更新时间"`
|
||
}
|
||
|
||
// StructToString 使用反射将结构体的内容组织为字符串
|
||
func StructToString(v interface{}) string {
|
||
val := reflect.ValueOf(v)
|
||
typ := val.Type()
|
||
|
||
var result []string
|
||
for i := 0; i < val.NumField(); i++ {
|
||
field := typ.Field(i)
|
||
tag := field.Tag.Get("explain")
|
||
value := val.Field(i).Interface()
|
||
result = append(result, fmt.Sprintf("%s：%v", tag, value))
|
||
}
|
||
return strings.Join(result, "；")
|
||
}
|
||
|
||
func TestExplain(t *testing.T) {
|
||
// 示例数据
|
||
updatedAt := time.Date(2023, 10, 1, 12, 0, 0, 0, time.UTC)
|
||
encounter := EncounterResult2{
|
||
Id: 1,
|
||
Title: "遇见小猫",
|
||
Content: "今天在公园遇到了一只可爱的小猫。",
|
||
UpdatedAt: &updatedAt,
|
||
}
|
||
|
||
// 调用 StructToString 函数
|
||
t.Logf("结构体内容：", StructToString(encounter))
|
||
}
|