INNER CODE UNIT · Go

codexToSQL

mergestat/mergestat-lite · cmd/codex.go:16

func codexToSQL(ctx context.Context, prompt string) (string, error) {
	apiKey := os.Getenv("OPENAI_API_KEY")
	if apiKey == "" {
		return "", fmt.Errorf("missing OPENAI_API_KEY environment variable")
	}

	client := gpt3.NewClient(apiKey)
	var temp float32 = 0
	var topP float32 = 1
	var maxTokens = 512
	res, err := client.CompletionWithEngine(ctx, "code-davinci-002", gpt3.CompletionRequest{
		Prompt:           []string{promptPrefix + prompt + "SELECT"},
		Temperature:      &temp,
		TopP:             &topP,
		FrequencyPenalty: 1,
		Stop:             []string{";"},
		MaxTokens:        &maxTokens,
	})

View source record →