ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-19 18:14:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: update Ollama embedding API to latest version with multi-text embedding support (#1715)

Browse Source
This commit is contained in:
SLKun
2024-08-06 23:43:20 +08:00
committed by GitHub
parent 402fcf7f79
commit e7e99e558a
3 changed files with 26 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
relay/adaptor/ollama/model.go
View File

@@ -37,11 +37,15 @@ type ChatResponse struct {
}
type EmbeddingRequest struct {
Model string `json:"model"`
Prompt string `json:"prompt"`
Model string `json:"model"`
Input []string `json:"input"`
// Truncate bool `json:"truncate,omitempty"`
Options *Options `json:"options,omitempty"`
// KeepAlive string `json:"keep_alive,omitempty"`
}
type EmbeddingResponse struct {
Error string `json:"error,omitempty"`
Embedding []float64 `json:"embedding,omitempty"`
Error string `json:"error,omitempty"`
Model string `json:"model"`
Embeddings [][]float64 `json:"embeddings"`
}