Skip to content

Commit ea2fa4b

Browse files
committed
fix: fix the promptFile bug
Signed-off-by: Grigoris Pavlidis <[email protected]>
1 parent c3d4ed4 commit ea2fa4b

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

main.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ type Response struct {
4141
} `json:"choices"`
4242
Error struct {
4343
Message string `json:"message"`
44-
Code int `json:"code"`
44+
Code string `json:"code"`
4545
Type string `json:"type"`
4646
} `json:"error"`
4747
}
@@ -64,7 +64,7 @@ func gatherOptions() options {
6464
fs.StringVar(&o.apiKey, "apiKey", "", "OpenAI API key")
6565
fs.StringVar(&o.model, "model", "gpt-3.5-turbo", "OpenAI model ID")
6666
fs.StringVar(&o.prompt, "prompt", "Write me a 100 word paragraph, use h1 and h2 and bold. reply in markdown format", "Text prompt to generate a response to")
67-
fs.StringVar(&o.promptFile, "promptFile", "File", "Text prompt to generate a response to")
67+
fs.StringVar(&o.promptFile, "promptFile", "", "Text prompt to generate a response to")
6868
fs.IntVar(&o.maxTokens, "maxTokens", 50, "Maximum number of tokens to generate in the response")
6969
fs.Float64Var(&o.temperature, "temperature", 0.5, "Sampling temperature for the model")
7070
fs.Float64Var(&o.frequencyPenalty, "frequencyPenalty", 0.5, "Frequency penalty for the model")
@@ -91,15 +91,14 @@ func main() {
9191
logrus.WithError(err).Fatalf("argument error: %v", err)
9292
}
9393

94-
prompt := ""
94+
prompt := o.prompt
95+
9596
if o.promptFile != "" {
9697
data, err := ioutil.ReadFile(o.promptFile)
9798
if err != nil {
9899
logrus.WithError(err).Fatalf("Failed to read prompt file: %v", err)
99100
}
100101
prompt = string(data)
101-
} else {
102-
prompt = o.prompt
103102
}
104103

105104
reqData := Request{

0 commit comments

Comments
 (0)