From f08ac71a3ba5ba98141c8296bcd856379398479d Mon Sep 17 00:00:00 2001 From: Roy Han Date: Fri, 12 Jul 2024 17:30:30 -0700 Subject: [PATCH] more clean up --- server/routes.go | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/server/routes.go b/server/routes.go index e6be65d04..89284163a 100644 --- a/server/routes.go +++ b/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)