chore: add fakeai provider

pkg/ai/ai.go

@@ -87,3 +87,7 @@ func (a *OpenAIClient) Parse(ctx context.Context, prompt []string, nocache bool)
 	}
 	return response, nil
 }
+
+func (a *OpenAIClient) GetName() string {
+	return "fakeai"
+}

pkg/ai/fakeai.go

@@ -0,0 +1,52 @@
+package ai
+
+import (
+	"context"
+	"encoding/base64"
+	"github.com/fatih/color"
+	"github.com/spf13/viper"
+	"strings"
+)
+
+type FakeAIClient struct {
+	client   string
+	language string
+}
+
+func (c *FakeAIClient) Configure(token string, language string) error {
+	c.language = language
+	c.client = "I am a fake client with the token " + token
+	return nil
+}
+
+func (c *FakeAIClient) GetCompletion(ctx context.Context, prompt string) (string, error) {
+	// Create a completion request
+	response := "I am a fake response to the prompt " + prompt
+	return response, nil
+}
+
+func (a *FakeAIClient) Parse(ctx context.Context, prompt []string, nocache bool) (string, error) {
+	// parse the text with the AI backend
+	inputKey := strings.Join(prompt, " ")
+	// Check for cached data
+	sEnc := base64.StdEncoding.EncodeToString([]byte(inputKey))
+
+	response, err := a.GetCompletion(ctx, inputKey)
+	if err != nil {
+		color.Red("error getting completion: %v", err)
+		return "", err
+	}
+
+	if !viper.IsSet(sEnc) {
+		viper.Set(sEnc, base64.StdEncoding.EncodeToString([]byte(response)))
+		if err := viper.WriteConfig(); err != nil {
+			color.Red("error writing config: %v", err)
+			return "", nil
+		}
+	}
+	return response, nil
+}
+
+func (a *FakeAIClient) GetName() string {
+	return "fakeai"
+}

pkg/ai/iai.go

@@ -8,12 +8,15 @@ type IAI interface {
 	Configure(token string, language string) error
 	GetCompletion(ctx context.Context, prompt string) (string, error)
 	Parse(ctx context.Context, prompt []string, nocache bool) (string, error)
+	GetName() string
 }
 
 func NewClient(provider string) IAI {
 	switch provider {
 	case "openai":
 		return &OpenAIClient{}
+	case "fakeai":
+		return &FakeAIClient{}
 	default:
 		return &OpenAIClient{}
 	}