소스 검색

embedding type 64

Roy Han 9 달 전
부모
커밋
fb390b8902
2개의 변경된 파일9개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      api/types.go
  2. 8 2
      server/routes.go

+ 1 - 1
api/types.go

@@ -215,7 +215,7 @@ type EmbeddingRequest struct {
 
 
 // EmbeddingResponse is the response from [Client.Embeddings].
 // EmbeddingResponse is the response from [Client.Embeddings].
 type EmbeddingResponse struct {
 type EmbeddingResponse struct {
-	Embedding []float32 `json:"embedding"`
+	Embedding []float64 `json:"embedding"`
 }
 }
 
 
 // CreateRequest is the request passed to [Client.Create].
 // CreateRequest is the request passed to [Client.Create].

+ 8 - 2
server/routes.go

@@ -504,7 +504,7 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) {
 
 
 	// an empty request loads the model
 	// an empty request loads the model
 	if req.Prompt == "" {
 	if req.Prompt == "" {
-		c.JSON(http.StatusOK, api.EmbeddingResponse{Embedding: []float32{}})
+		c.JSON(http.StatusOK, api.EmbeddingResponse{Embedding: []float64{}})
 		return
 		return
 	}
 	}
 
 
@@ -515,8 +515,14 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) {
 		return
 		return
 	}
 	}
 
 
+	embedding64 := make([]float64, len(embedding[0]))
+
+	for i, v := range embedding[0] {
+		embedding64[i] = float64(v)
+	}
+
 	resp := api.EmbeddingResponse{
 	resp := api.EmbeddingResponse{
-		Embedding: embedding[0],
+		Embedding: embedding64,
 	}
 	}
 	c.JSON(http.StatusOK, resp)
 	c.JSON(http.StatusOK, resp)
 }
 }