feat: yi

labring · Dec 30, 2024 · ed1d742 · ed1d742
1 parent db7bc0e
commit ed1d742
Showing 1 changed file with 9 additions and 77 deletions.
diff --git a/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go b/service/aiproxy/relay/adaptor/lingyiwanwu/constants.go
@@ -14,88 +14,20 @@ var ModelList = []*model.ModelConfig{
 		Owner:       model.ModelOwnerLingyiWanwu,
 		InputPrice:  0.00099,
 		OutputPrice: 0.00099,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
+		RPM:         60,
+		Config: model.NewModelConfig(
+			model.WithModelConfigMaxContextTokens(16384),
+		),
 	},
 	{
-		Model:       "yi-large",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.02,
-		OutputPrice: 0.02,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 32768,
-		},
-	},
-	{
-		Model:       "yi-medium",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.0025,
-		OutputPrice: 0.0025,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
-	},
-	{
-		Model:       "yi-vision",
+		Model:       "yi-vision-v2",
 		Type:        relaymode.ChatCompletions,
 		Owner:       model.ModelOwnerLingyiWanwu,
 		InputPrice:  0.006,
 		OutputPrice: 0.006,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
-	},
-	{
-		Model:       "yi-medium-200k",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.012,
-		OutputPrice: 0.012,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 204800,
-		},
-	},
-	{
-		Model:       "yi-spark",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.001,
-		OutputPrice: 0.001,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
-	},
-	{
-		Model:       "yi-large-rag",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.025,
-		OutputPrice: 0.025,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
-	},
-	{
-		Model:       "yi-large-fc",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.02,
-		OutputPrice: 0.02,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 32768,
-		},
-	},
-	{
-		Model:       "yi-large-turbo",
-		Type:        relaymode.ChatCompletions,
-		Owner:       model.ModelOwnerLingyiWanwu,
-		InputPrice:  0.012,
-		OutputPrice: 0.012,
-		Config: map[model.ModelConfigKey]any{
-			model.ModelConfigMaxContextTokensKey: 16384,
-		},
+		RPM:         60,
+		Config: model.NewModelConfig(
+			model.WithModelConfigMaxContextTokens(16384),
+		),
 	},
 }