Skip to content

Commit ceb3560

Browse files
Remove default model set (#39)
* remove default model set * Update CODEOWNERS
1 parent 3223f85 commit ceb3560

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

CODEOWNERS

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
* @BarakHarariCx @RamiGerassi
1+
* @BarakHarariCx @RamiGerassi @cx-avi-sabzerou @cx-mor-levy @cx-alex-cohen

internal/gpt.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,16 +3,17 @@ package internal
33
import (
44
"errors"
55
"fmt"
6+
"net/url"
7+
68
"github.com/Checkmarx/gen-ai-wrapper/pkg/message"
79
"github.com/Checkmarx/gen-ai-wrapper/pkg/role"
8-
"net/url"
910
)
1011

1112
// const gptByAzure = "https://cxgpt4.openai.azure.com/openai/deployments/gpt-4/chat/completions?api-version=2023-05-15"
1213
// const gptByOpenAi = "https://api.openai.com/v1/chat/completions"
1314

1415
type ChatCompletionRequest struct {
15-
Model string `json:"model"`
16+
Model string `json:"model,omitempty"`
1617
Messages []message.Message `json:"messages"`
1718
}
1819

pkg/wrapper/stateless_wrapper.go

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ import (
77
"github.com/Checkmarx/gen-ai-wrapper/internal/secrets"
88
"github.com/Checkmarx/gen-ai-wrapper/pkg/maskedSecret"
99
"github.com/Checkmarx/gen-ai-wrapper/pkg/message"
10-
"github.com/Checkmarx/gen-ai-wrapper/pkg/models"
1110
"github.com/Checkmarx/gen-ai-wrapper/pkg/role"
1211
)
1312

@@ -31,9 +30,6 @@ type StatelessWrapperImpl struct {
3130
}
3231

3332
func NewStatelessWrapper(endPoint, apiKey, model string, dropLen, limit int) (StatelessWrapper, error) {
34-
if model == "" {
35-
model = models.DefaultModel
36-
}
3733
wrapper, err := internal.NewWrapperFactory(endPoint, apiKey, dropLen)
3834
if err != nil {
3935
return nil, err

0 commit comments

Comments
 (0)