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

feat: support gemini embeddings (text-embedding-004,embedding-001) (#1475)

Browse Source 
* Refactor Gemini Adaptor to Support Embeddings

* Add new models to ModelList
This commit is contained in:
Wei Tingjiang
2024-05-29 01:17:32 +08:00
committed by GitHub
parent ceea4c6d4a
commit 9321427c6e
4 changed files with 127 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
relay/adaptor/gemini/constants.go
View File

@@ -4,5 +4,5 @@ package gemini
var ModelList = []string{
"gemini-pro", "gemini-1.0-pro-001", "gemini-1.5-pro",
"gemini-pro-vision", "gemini-1.0-pro-vision-001",
"gemini-pro-vision", "gemini-1.0-pro-vision-001", "embedding-001", "text-embedding-004",
}