CalciumIon
0725c8aac9
- Introduced `GEMINI_VISION_MAX_IMAGE_NUM` to README files for better user guidance. - Updated `env.go` to retrieve the maximum image number from environment variables, defaulting to 16. - Modified image handling logic in `relay-gemini.go` to respect the new configuration, allowing disabling of the limit by setting it to -1. - Removed hardcoded constant for maximum image number in `constant.go` to streamline configuration management.
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package constant
|
|
|
|
import (
|
|
"fmt"
|
|
"one-api/common"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
var StreamingTimeout = common.GetEnvOrDefault("STREAMING_TIMEOUT", 60)
|
|
var DifyDebug = common.GetEnvOrDefaultBool("DIFY_DEBUG", true)
|
|
|
|
// ForceStreamOption 覆盖请求参数，强制返回usage信息
|
|
var ForceStreamOption = common.GetEnvOrDefaultBool("FORCE_STREAM_OPTION", true)
|
|
|
|
var GetMediaToken = common.GetEnvOrDefaultBool("GET_MEDIA_TOKEN", true)
|
|
|
|
var GetMediaTokenNotStream = common.GetEnvOrDefaultBool("GET_MEDIA_TOKEN_NOT_STREAM", true)
|
|
|
|
var UpdateTask = common.GetEnvOrDefaultBool("UPDATE_TASK", true)
|
|
|
|
var GeminiModelMap = map[string]string{
|
|
"gemini-1.0-pro": "v1",
|
|
}
|
|
|
|
var GeminiVisionMaxImageNum = common.GetEnvOrDefault("GEMINI_VISION_MAX_IMAGE_NUM", 16)
|
|
|
|
func InitEnv() {
|
|
modelVersionMapStr := strings.TrimSpace(os.Getenv("GEMINI_MODEL_MAP"))
|
|
if modelVersionMapStr == "" {
|
|
return
|
|
}
|
|
for _, pair := range strings.Split(modelVersionMapStr, ",") {
|
|
parts := strings.Split(pair, ":")
|
|
if len(parts) == 2 {
|
|
GeminiModelMap[parts[0]] = parts[1]
|
|
} else {
|
|
common.SysError(fmt.Sprintf("invalid model version map: %s", pair))
|
|
}
|
|
}
|
|
}
|
|
|
|
// 是否生成初始令牌，默认关闭。
|
|
var GenerateDefaultToken = common.GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false)
|