artin/sourcegraph
1
0
Fork 0
mirror of https://github.com/sourcegraph/sourcegraph.git synced 2026-02-06 16:51:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Llama Code 13b to all PLG users and remove unused model conversation (#60332)

Browse Source
This commit is contained in:
Philipp Spiess 2024-02-09 11:51:25 +01:00 committed by GitHub
parent 0d24f22744
commit 025a603207
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd/frontend/internal/dotcom/productsubscription/codygateway_dotcom_user.go
View File

@ -374,6 +374,7 @@ func allowedModels(scope types.CompletionsFeature, isProUser bool) []string {
"anthropic/claude-instant-1.2-cyan",
"anthropic/claude-instant-1.2",
"fireworks/starcoder",
"fireworks/" + fireworks.Llama213bCode,
// TODO: Remove the specific model identifiers below when Cody Gateway for PLG was updated.
"fireworks/" + fireworks.Starcoder16b,
"fireworks/" + fireworks.Starcoder7b,

8
internal/completions/httpapi/codecompletion.go
View File

@ -45,13 +45,9 @@ func NewCodeCompletionsHandler(logger log.Logger, db database.DB, test guardrail
func allowedCustomModel(model string) string {
switch model {
// These virtual model strings allow the server to choose the model.
// TODO: Remove the specific model identifiers below when Cody Gateway for PLG was updated.
case "fireworks/starcoder-16b":
return "fireworks/" + fireworks.Starcoder16b
case "fireworks/starcoder-7b":
return "fireworks/" + fireworks.Starcoder7b
case "fireworks/starcoder",
"fireworks/starcoder-16b",
"fireworks/starcoder-7b",
"fireworks/" + fireworks.Starcoder16b,
"fireworks/" + fireworks.Starcoder7b,
"fireworks/" + fireworks.Llama27bCode,