highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-11-11 00:37:42 +01:00
Code Issues Packages Projects Releases Wiki Activity

allowed base64 encoding

Browse Source
This commit is contained in:
nicole pardal
2025-10-20 17:38:42 -07:00
parent 0334e67ffd
commit 53add2e0f5
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
middleware/openai.go
View File

@@ -256,7 +256,8 @@ func (w *EmbedWriter) writeResponse(data []byte) (int, error) {
} }
w.ResponseWriter.Header().Set("Content-Type", "application/json") w.ResponseWriter.Header().Set("Content-Type", "application/json")
err = json.NewEncoder(w.ResponseWriter).Encode(openai.ToEmbeddingList(w.model, embedResponse, w.encodingFormat)) converted := openai.ToEmbeddingList(w.model, embedResponse, w.encodingFormat)
err = json.NewEncoder(w.ResponseWriter).Encode(converted)
if err != nil { if err != nil {
return 0, err return 0, err
} }

8
openai/openai.go
View File

@@ -383,11 +383,15 @@ func ToListCompletion(r api.ListResponse) ListCompletion {
// encodingFormat can be "float", "base64", or empty (defaults to "float") // encodingFormat can be "float", "base64", or empty (defaults to "float")
func ToEmbeddingList(model string, r api.EmbedResponse, encodingFormat string) EmbeddingList { func ToEmbeddingList(model string, r api.EmbedResponse, encodingFormat string) EmbeddingList {
if r.Embeddings != nil { if r.Embeddings != nil {
if encodingFormat == "" {
encodingFormat = "float"
}
var data []Embedding var data []Embedding
for i, e := range r.Embeddings { for i, e := range r.Embeddings {
var embedding any var embedding any
if strings.EqualFold(encodingFormat, "base64") { if encodingFormat == "base64" {
embedding = floatsToBase64(e) embedding = convertFloatsToBase64(e)
} else { } else {
embedding = e embedding = e
} }