jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more clean up

Browse Source
This commit is contained in:
Roy Han 2024-07-12 17:30:30 -07:00
parent 424f3f81a9
commit f08ac71a3b
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/routes.go
View File

@ -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)