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

simplfied logic and added tests

Browse Source
This commit is contained in:
nicole pardal
2025-10-21 13:57:01 -07:00
parent 53add2e0f5
commit c563286e63
2 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
middleware/openai.go
View File

@@ -256,8 +256,7 @@ func (w *EmbedWriter) writeResponse(data []byte) (int, error) {
} }
w.ResponseWriter.Header().Set("Content-Type", "application/json") w.ResponseWriter.Header().Set("Content-Type", "application/json")
converted := openai.ToEmbeddingList(w.model, embedResponse, w.encodingFormat) err = json.NewEncoder(w.ResponseWriter).Encode(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,15 +383,11 @@ 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 encodingFormat == "base64" { if strings.EqualFold(encodingFormat, "base64") {
embedding = convertFloatsToBase64(e) embedding = floatsToBase64(e)
} else { } else {
embedding = e embedding = e
} }