diff --git a/app/model_es/encounter.go b/app/model_es/encounter.go index 3edaf74..82bf726 100644 --- a/app/model_es/encounter.go +++ b/app/model_es/encounter.go @@ -192,6 +192,7 @@ func (e *Encounter) TopK(embedding []float64, k int) ([]Encounter, error) { return nil, err } + // UPDATE 这里无法处理 ES 中 embedding 为 null 的情况。 body := fmt.Sprintf(`{ "size": %d, "query": { @@ -203,7 +204,7 @@ func (e *Encounter) TopK(embedding []float64, k int) ([]Encounter, error) { } } }, - "_source":["id"] + "_source": ["id"] }`, k, string(paramsJSON)) hits, err := model_handler.SearchRequest(body, e.IndexName()) diff --git a/routers/web.go b/routers/web.go index 8d1d305..871ae09 100644 --- a/routers/web.go +++ b/routers/web.go @@ -3,12 +3,10 @@ package routers import ( "catface/app/global/consts" "catface/app/global/variable" - // "catface/app/http/controller/captcha" // 验证码组件 // "catface/app/http/middleware/authorization" "catface/app/http/middleware/cors" validatorFactory "catface/app/http/validator/core/factory" - // TODO validatorFactory "catface/app/http/validator/core/factory" "catface/app/utils/gin_release" "net/http" @@ -16,6 +14,7 @@ import ( "github.com/gin-contrib/pprof" "github.com/gin-gonic/gin" "go.uber.org/zap" + ) // 该路由主要设置 后台管理系统等后端应用路由 diff --git a/test/es/encounter_embedding/main.go b/test/es/encounter_embedding/main.go new file mode 100644 index 0000000..2a9bac8 --- /dev/null +++ b/test/es/encounter_embedding/main.go @@ -0,0 +1,20 @@ +package main + +import ( + "fmt" + "math/rand" + "strings" +) + +func main() { + embedding := make([]float64, 768) + for i := range embedding { + embedding[i] = rand.Float64() + } + + // 将嵌入向量转换为字符串,每个元素之间用逗号隔开 + embeddingStr := strings.Trim(strings.Join(strings.Fields(fmt.Sprint(embedding)), ","), "[]") + + // 打印前几个元素以验证 + fmt.Println(embeddingStr) +}