Skip to content

Commit a2d10e2

Browse files
codemakerai-devcodemakerai-dev
committed
Added endpoint option to all commands
1 parent 34d358d commit a2d10e2

File tree

2 files changed

+27
-27
lines changed

2 files changed

+27
-27
lines changed

‎cli/cli.go‎

Lines changed: 20 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -71,25 +71,26 @@ func (c *Cli) parseGenerateArgs() {
7171

7272
switch os.Args[2] {
7373
case "code":
74-
generateDocsCmd := flag.NewFlagSet("generateCode", flag.ExitOnError)
75-
lang := generateDocsCmd.String("language", "", "Programming language: JavaScript, Java, Kotlin")
76-
replace := generateDocsCmd.Bool("replace", false, "Determines if the existing code is replaced")
77-
codePath := generateDocsCmd.String("codepath", "", "The codepath to match.")
78-
model := generateDocsCmd.String("model", "", "The fine-tuned model name.")
79-
80-
err := generateDocsCmd.Parse(os.Args[3:])
74+
generateCodeCmd := flag.NewFlagSet("generateCode", flag.ExitOnError)
75+
lang := generateCodeCmd.String("language", "", "Programming language: JavaScript, Java, Kotlin")
76+
replace := generateCodeCmd.Bool("replace", false, "Determines if the existing code is replaced")
77+
codePath := generateCodeCmd.String("codepath", "", "The codepath to match.")
78+
model := generateCodeCmd.String("model", "", "The fine-tuned model name.")
79+
endpoint := generateCodeCmd.String("endpoint", "", "The endpoint name.")
80+
81+
err := generateCodeCmd.Parse(os.Args[3:])
8182
if err != nil {
8283
c.logger.Errorf("Could not parse args %v", err)
8384
os.Exit(1)
8485
}
8586

86-
if len(generateDocsCmd.Args()) == 0 {
87+
if len(generateCodeCmd.Args()) == 0 {
8788
c.logger.Errorf("Expected file input")
8889
fmt.Printf("Usage: codemaker generate code <file>\n")
8990
os.Exit(1)
9091
}
9192

92-
config, err := createConfig()
93+
config, err := createConfig(endpoint)
9394
if err != nil {
9495
c.logger.Errorf("No valid api key found: %v", err)
9596
os.Exit(1)
@@ -101,7 +102,7 @@ func (c *Cli) parseGenerateArgs() {
101102
os.Exit(1)
102103
}
103104

104-
files := generateDocsCmd.Args()[0:]
105+
files := generateCodeCmd.Args()[0:]
105106

106107
if err := c.generateCode(cl, lang, replace, codePath, model, files); err != nil {
107108
c.logger.Errorf("Could not generate the code %v", err)
@@ -112,6 +113,7 @@ func (c *Cli) parseGenerateArgs() {
112113
lang := generateDocsCmd.String("language", "", "Programming language: JavaScript, Java, Kotlin")
113114
replace := generateDocsCmd.Bool("replace", false, "Determines if the existing documentations are replaced")
114115
codePath := generateDocsCmd.String("codepath", "", "The codepath to match.")
116+
endpoint := generateDocsCmd.String("endpoint", "", "The endpoint name.")
115117

116118
err := generateDocsCmd.Parse(os.Args[3:])
117119
if err != nil {
@@ -125,7 +127,7 @@ func (c *Cli) parseGenerateArgs() {
125127
os.Exit(1)
126128
}
127129

128-
config, err := createConfig()
130+
config, err := createConfig(endpoint)
129131
if err != nil {
130132
c.logger.Errorf("No valid api key found: %v", err)
131133
os.Exit(1)
@@ -157,21 +159,22 @@ func (c *Cli) parseFixArgs() {
157159

158160
switch os.Args[2] {
159161
case "syntax":
160-
refactorNaming := flag.NewFlagSet("fixSyntax", flag.ExitOnError)
161-
lang := refactorNaming.String("language", "", "Programming language: JavaScript, Java, Kotlin")
162+
correctSyntaxCmd := flag.NewFlagSet("fixSyntax", flag.ExitOnError)
163+
lang := correctSyntaxCmd.String("language", "", "Programming language: JavaScript, Java, Kotlin")
164+
endpoint := correctSyntaxCmd.String("endpoint", "", "The endpoint name.")
162165

163-
err := refactorNaming.Parse(os.Args[3:])
166+
err := correctSyntaxCmd.Parse(os.Args[3:])
164167
if err != nil {
165168
c.logger.Errorf("Could not parse args %v", err)
166169
}
167170

168-
if len(refactorNaming.Args()) == 0 {
171+
if len(correctSyntaxCmd.Args()) == 0 {
169172
c.logger.Errorf("Expected file input")
170173
fmt.Printf("Usage: codemaker fix syntax <file>")
171174
os.Exit(1)
172175
}
173176

174-
config, err := createConfig()
177+
config, err := createConfig(endpoint)
175178
if err != nil {
176179
c.logger.Errorf("No valid api key found %v", err)
177180
os.Exit(1)
@@ -183,7 +186,7 @@ func (c *Cli) parseFixArgs() {
183186
os.Exit(1)
184187
}
185188

186-
input := refactorNaming.Args()[0:]
189+
input := correctSyntaxCmd.Args()[0:]
187190

188191
if err := c.fixSyntax(cl, lang, input); err != nil {
189192
c.logger.Errorf("Could not fix syntax %v", err)
@@ -488,7 +491,6 @@ func (c *Cli) printHelp() {
488491
fmt.Printf("\n")
489492
fmt.Printf("Commands:\n")
490493
fmt.Printf(" * generate\n")
491-
fmt.Printf(" * migrate\n")
492494
fmt.Printf(" * refactor\n")
493495
fmt.Printf(" * fix\n")
494496
fmt.Printf(" * configure\n")
@@ -506,14 +508,6 @@ func (c *Cli) printGenerateHelp() {
506508
os.Exit(1)
507509
}
508510

509-
func (c *Cli) printMigrateHelp() {
510-
fmt.Printf("Usage: codemaker migrate <command>\n")
511-
fmt.Printf("\n")
512-
fmt.Printf("Commands:\n")
513-
fmt.Printf(" * syntax\n")
514-
os.Exit(1)
515-
}
516-
517511
func (c *Cli) printRefactorHelp() {
518512
fmt.Printf("Usage: codemaker refactor <command>\n")
519513
fmt.Printf("\n")

‎cli/config.go‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,12 @@ const (
1515
defaultMaxRetries = 10
1616
)
1717

18-
func createConfig() (*client.Config, error) {
18+
func createConfig(endpoint *string) (*client.Config, error) {
19+
var serviceEndpoint *string = nil
20+
if endpoint != nil && len(*endpoint) > 0 {
21+
serviceEndpoint = endpoint
22+
}
23+
1924
apiKey := os.Getenv(apiKeyEnvironmentVariable)
2025

2126
if len(apiKey) == 0 {
@@ -39,6 +44,7 @@ func createConfig() (*client.Config, error) {
3944
maxRetries := defaultMaxRetries
4045

4146
return &client.Config{
47+
Endpoint: serviceEndpoint,
4248
ApiKey: apiKey,
4349
MaxRetries: &maxRetries,
4450
}, nil

0 commit comments

Comments
 (0)