mirror of
https://github.com/coaidev/coai.git
synced 2025-05-23 15:00:14 +09:00
33 lines
535 B
Go
33 lines
535 B
Go
package oneapi
|
|
|
|
import (
|
|
"chat/globals"
|
|
)
|
|
|
|
var HitModels = []string{
|
|
globals.Claude2, globals.Claude2100k,
|
|
}
|
|
|
|
func (c *ChatInstance) Process(data string) string {
|
|
return data
|
|
}
|
|
|
|
func (c *ChatInstance) FormatMessage(message []globals.Message) []globals.Message {
|
|
return message
|
|
}
|
|
|
|
func (c *ChatInstance) FormatModel(model string) string {
|
|
return model
|
|
}
|
|
|
|
func (c *ChatInstance) GetToken(model string) int {
|
|
switch model {
|
|
case globals.Claude2:
|
|
return 5000
|
|
case globals.Claude2100k:
|
|
return 50000
|
|
default:
|
|
return 2500
|
|
}
|
|
}
|