56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package chatmodel
|
|
|
|
import "github.com/coze-dev/coze-studio/backend/api/model/ocean/cloud/developer_api"
|
|
|
|
type Protocol string
|
|
|
|
const (
|
|
ProtocolOpenAI Protocol = "openai"
|
|
ProtocolClaude Protocol = "claude"
|
|
ProtocolDeepseek Protocol = "deepseek"
|
|
ProtocolGemini Protocol = "gemini"
|
|
ProtocolArk Protocol = "ark"
|
|
ProtocolOllama Protocol = "ollama"
|
|
ProtocolQwen Protocol = "qwen"
|
|
ProtocolErnie Protocol = "ernie"
|
|
)
|
|
|
|
func (p Protocol) TOModelClass() developer_api.ModelClass {
|
|
switch p {
|
|
case ProtocolArk:
|
|
return developer_api.ModelClass_SEED
|
|
case ProtocolOpenAI:
|
|
return developer_api.ModelClass_GPT
|
|
case ProtocolDeepseek:
|
|
return developer_api.ModelClass_DeekSeek
|
|
case ProtocolClaude:
|
|
return developer_api.ModelClass_Claude
|
|
case ProtocolGemini:
|
|
return developer_api.ModelClass_Gemini
|
|
case ProtocolOllama:
|
|
return developer_api.ModelClass_Llama
|
|
case ProtocolQwen:
|
|
return developer_api.ModelClass_QWen
|
|
case ProtocolErnie:
|
|
return developer_api.ModelClass_Ernie
|
|
default:
|
|
return developer_api.ModelClass_Other
|
|
}
|
|
}
|