mirror of
https://github.com/Direct-Dev-Ru/go-lcg.git
synced 2025-11-16 01:29:55 +00:00
feat: update prompt and model
This commit is contained in:
@@ -18,6 +18,7 @@ type Gpt3 struct {
|
||||
HomeDir string
|
||||
ApiKeyFile string
|
||||
ApiKey string
|
||||
Temperature float64
|
||||
}
|
||||
|
||||
type Chat struct {
|
||||
@@ -28,6 +29,7 @@ type Chat struct {
|
||||
type Gpt3Request struct {
|
||||
Model string `json:"model"`
|
||||
Messages []Chat `json:"messages"`
|
||||
Temperature float64 `json:"temperature"`
|
||||
}
|
||||
|
||||
type Gpt3Response struct {
|
||||
@@ -139,6 +141,7 @@ func (gpt3 *Gpt3) Completions(ask string) string {
|
||||
payload := Gpt3Request{
|
||||
gpt3.Model,
|
||||
messages,
|
||||
gpt3.Temperature,
|
||||
}
|
||||
payloadJson, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
|
||||
5
main.go
5
main.go
@@ -16,8 +16,8 @@ import (
|
||||
const (
|
||||
HOST = "https://api.openai.com/v1/"
|
||||
COMPLETIONS = "chat/completions"
|
||||
MODEL = "gpt-3.5-turbo"
|
||||
PROMPT = "I want you to reply with linux command and nothing else. Do not write explanations."
|
||||
MODEL = "gpt-4o-mini"
|
||||
PROMPT = "Reply with linux command and nothing else. No need explanation. No need code blocks"
|
||||
|
||||
// This file is created in the user's home directory
|
||||
// Example: /home/username/.openai_api_key
|
||||
@@ -136,6 +136,7 @@ func main() {
|
||||
Prompt: PROMPT,
|
||||
HomeDir: currentUser.HomeDir,
|
||||
ApiKeyFile: API_KEY_FILE,
|
||||
Temperature: 0.01,
|
||||
}
|
||||
|
||||
if h == CMD_UPDATE {
|
||||
|
||||
Reference in New Issue
Block a user