浏览代码

more clean up

Roy Han 9 月之前
父节点
当前提交
f08ac71a3b
共有 1 个文件被更改,包括 2 次插入8 次删除
  1. 2 8
      server/routes.go

+ 2 - 8
server/routes.go

@@ -265,11 +265,6 @@ func (s *Server) EmbedHandler(c *gin.Context) {
 		truncate = false
 	}
 
-	if req.Truncate == nil {
-		truncate := true
-		req.Truncate = &truncate
-	}
-
 	var input []string
 
 	switch i := req.Input.(type) {
@@ -307,7 +302,6 @@ func (s *Server) EmbedHandler(c *gin.Context) {
 		return
 	}
 
-	reqEmbedArray := make([]string, len(input))
 	for i, s := range input {
 		tokens, err := r.Tokenize(c.Request.Context(), s)
 		if err != nil {
@@ -330,9 +324,9 @@ func (s *Server) EmbedHandler(c *gin.Context) {
 			}
 		}
 
-		reqEmbedArray[i] = s
+		input[i] = s
 	}
-	embeddings, err := r.Embed(c.Request.Context(), reqEmbedArray)
+	embeddings, err := r.Embed(c.Request.Context(), input)
 
 	if err != nil {
 		slog.Error("embedding generation failed", "error", err)