Просмотр исходного кода

feat: support Anthropic Claude 3.0 Haiku (close #116)

CaIon 2 лет назад
Родитель
Сommit
98c1f66d61

+ 1 - 0
common/model-ratio.go

@@ -64,6 +64,7 @@ var ModelRatio = map[string]float64{
 	"claude-instant-1":          0.4,    // $0.8 / 1M tokens
 	"claude-2.0":                4,      // $8 / 1M tokens
 	"claude-2.1":                4,      // $8 / 1M tokens
+	"claude-3-haiku-20240307":   0.125,  // $0.25 / 1M tokens
 	"claude-3-sonnet-20240229":  1.5,    // $3 / 1M tokens
 	"claude-3-opus-20240229":    7.5,    // $15 / 1M tokens
 	"ERNIE-Bot":                 0.8572, // ¥0.012 / 1k tokens

+ 7 - 1
relay/channel/claude/constants.go

@@ -1,7 +1,13 @@
 package claude
 
 var ModelList = []string{
-	"claude-instant-1.2", "claude-2", "claude-2.0", "claude-2.1", "claude-3-sonnet-20240229", "claude-3-opus-20240229",
+	"claude-instant-1.2",
+	"claude-2",
+	"claude-2.0",
+	"claude-2.1",
+	"claude-3-sonnet-20240229",
+	"claude-3-opus-20240229",
+	"claude-3-haiku-20240307",
 }
 
 var ChannelName = "claude"

+ 1 - 1
web/src/pages/Channel/EditChannel.js

@@ -63,7 +63,7 @@ const EditChannel = (props) => {
             let localModels = [];
             switch (value) {
                 case 14:
-                    localModels = ["claude-instant-1.2", "claude-2", "claude-2.0", "claude-2.1", "claude-3-sonnet-20240229", "claude-3-opus-20240229"];
+                    localModels = ["claude-instant-1.2", "claude-2", "claude-2.0", "claude-2.1", "claude-3-opus-20240229", "claude-3-sonnet-20240229", "claude-3-haiku-20240307"];
                     break;
                 case 11:
                     localModels = ['PaLM-2'];