k8sgpt/pkg/ai/interactive/interactive.go

package interactive

import (
	"fmt"
	"strings"

	"github.com/fatih/color"
	"github.com/k8sgpt-ai/k8sgpt/pkg/analysis"
	"github.com/pterm/pterm"
)

type INTERACTIVE_STATE int

const (
	prompt = "Given the following context: "
)

const (
	E_RUNNING INTERACTIVE_STATE = iota
	E_EXITED                    = iota
)

type InteractionRunner struct {
	config        *analysis.Analysis
	State         chan INTERACTIVE_STATE
	contextWindow []byte
}

func NewInteractionRunner(config *analysis.Analysis, contextWindow []byte) *InteractionRunner {
	return &InteractionRunner{
		config:        config,
		contextWindow: contextWindow,
		State:         make(chan INTERACTIVE_STATE),
	}
}

func (a *InteractionRunner) StartInteraction() {
	a.State <- E_RUNNING
	pterm.Println("Interactive mode enabled [type exit to close.]")
	for {

		query := pterm.DefaultInteractiveTextInput.WithMultiLine(false)
		queryString, err := query.Show()
		if err != nil {
			fmt.Println(err)
		}
		if queryString == "" {
			continue
		}
		if strings.Contains(queryString, "exit") {
			a.State <- E_EXITED
			continue
		}
		pterm.Println()
		contextWindow := fmt.Sprintf("%s %s %s", prompt, string(a.contextWindow),
			queryString)

		response, err := a.config.AIClient.GetCompletion(a.config.Context,
			contextWindow)
		if err != nil {
			color.Red("Error: %v", err)
			a.State <- E_EXITED
			continue
		}
		pterm.Println(response)
	}
}