Skip to content
This repository was archived by the owner on Sep 30, 2024. It is now read-only.

Commit 025a603

Browse files
Add Llama Code 13b to all PLG users and remove unused model conversation (#60332)
1 parent 0d24f22 commit 025a603

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

cmd/frontend/internal/dotcom/productsubscription/codygateway_dotcom_user.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -374,6 +374,7 @@ func allowedModels(scope types.CompletionsFeature, isProUser bool) []string {
374374
"anthropic/claude-instant-1.2-cyan",
375375
"anthropic/claude-instant-1.2",
376376
"fireworks/starcoder",
377+
"fireworks/" + fireworks.Llama213bCode,
377378
// TODO: Remove the specific model identifiers below when Cody Gateway for PLG was updated.
378379
"fireworks/" + fireworks.Starcoder16b,
379380
"fireworks/" + fireworks.Starcoder7b,

internal/completions/httpapi/codecompletion.go

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,9 @@ func NewCodeCompletionsHandler(logger log.Logger, db database.DB, test guardrail
4545

4646
func allowedCustomModel(model string) string {
4747
switch model {
48-
// These virtual model strings allow the server to choose the model.
49-
// TODO: Remove the specific model identifiers below when Cody Gateway for PLG was updated.
50-
case "fireworks/starcoder-16b":
51-
return "fireworks/" + fireworks.Starcoder16b
52-
case "fireworks/starcoder-7b":
53-
return "fireworks/" + fireworks.Starcoder7b
5448
case "fireworks/starcoder",
49+
"fireworks/starcoder-16b",
50+
"fireworks/starcoder-7b",
5551
"fireworks/" + fireworks.Starcoder16b,
5652
"fireworks/" + fireworks.Starcoder7b,
5753
"fireworks/" + fireworks.Llama27bCode,

0 commit comments

Comments
 (0)