Browse Source

more clean up

Roy Han 9 months ago
parent
commit
f08ac71a3b
1 changed files with 2 additions and 8 deletions
  1. 2 8
      server/routes.go

+ 2 - 8
server/routes.go

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