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

clean up

Browse Source
This commit is contained in:
Roy Han 2024-07-15 09:54:19 -07:00
parent f08ac71a3b
commit 766ca1cd7d
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/client.go
View File

@ -356,7 +356,7 @@ func (c *Client) Embed(ctx context.Context, req *EmbedRequest) (*EmbedResponse,
return &resp, nil return &resp, nil
} }
// Embeddings generates embeddings from a model. (Legacy) // Embeddings generates an embedding from a model.
func (c *Client) Embeddings(ctx context.Context, req *EmbeddingRequest) (*EmbeddingResponse, error) { func (c *Client) Embeddings(ctx context.Context, req *EmbeddingRequest) (*EmbeddingResponse, error) {
var resp EmbeddingResponse var resp EmbeddingResponse
if err := c.do(ctx, http.MethodPost, "/api/embeddings", req, &resp); err != nil { if err := c.do(ctx, http.MethodPost, "/api/embeddings", req, &resp); err != nil {

10
server/routes.go
View File

@ -384,7 +384,7 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) {
return return
} }
embedding, err := r.Embed(c.Request.Context(), []string{req.Prompt}) embeddings, err := r.Embed(c.Request.Context(), []string{req.Prompt})
if err != nil { if err != nil {
slog.Info(fmt.Sprintf("embedding generation failed: %v", err)) slog.Info(fmt.Sprintf("embedding generation failed: %v", err))
@ -392,14 +392,14 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) {
return return
} }
embedding64 := make([]float64, len(embedding[0])) embedding := make([]float64, len(embeddings[0]))
for i, v := range embedding[0] { for i, v := range embeddings[0] {
embedding64[i] = float64(v) embedding[i] = float64(v)
} }
resp := api.EmbeddingResponse{ resp := api.EmbeddingResponse{
Embedding: embedding64, Embedding: embedding,
} }
c.JSON(http.StatusOK, resp) c.JSON(http.StatusOK, resp)
} }