From 3d294ea1d6e462069dc3e2e8c0146f65fbadb1ff Mon Sep 17 00:00:00 2001
From: skeris <kotilion.95@gmail.com>
Date: Tue, 8 Apr 2025 17:34:01 +0300
Subject: [PATCH] added ai client for current version ai api

---
 app/app.go               |  1 +
 clients/aiClient.go      | 49 +++++++++++++++++++++++++++++++++-------
 clients/aiClient_test.go | 17 ++++++++++++++
 go.sum                   |  0
 service/service.go       | 11 +++++++--
 5 files changed, 68 insertions(+), 10 deletions(-)
 create mode 100644 clients/aiClient_test.go
 create mode 100644 go.sum

diff --git a/app/app.go b/app/app.go
index 16e2536..1cda1b6 100644
--- a/app/app.go
+++ b/app/app.go
@@ -153,6 +153,7 @@ func New(ctx context.Context, opts interface{}, ver appInit.Version) (appInit.Co
 		WorkerRespondentCh: workerRespondentCh,
 		Encrypt:            encrypt,
 		RedirectURl:        options.RedirectURL,
+		AiClient:           clients.NewAiClient("http://35.206.94.7:80/api/engine/generate_question_by_answer/"),
 	})
 
 	saveRespWcData := savewc.DepsForResp{
diff --git a/clients/aiClient.go b/clients/aiClient.go
index 7fea6bb..4acf52b 100644
--- a/clients/aiClient.go
+++ b/clients/aiClient.go
@@ -1,6 +1,8 @@
 package clients
 
 import (
+	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/gofiber/fiber/v2"
 )
@@ -24,13 +26,44 @@ func NewAiClient(baseURL string) *AIClient {
 	}
 }
 
-func (client *AIClient) SendAnswerer(final bool, tipe, message, session string) error {
-	req := SendAnswerRequest{
-		Tipe:    tipe,
-		Message: message,
-		Final:   final,
-		Session: session,
+func (client *AIClient) SendAnswerer(final bool, tipe, message, session string) (string, error) {
+	//req := SendAnswerRequest{
+	//	Tipe:    tipe,
+	//	Message: message,
+	//	Final:   final,
+	//	Session: session,
+	//}
+
+	clownRequest := struct {
+		Text string `json:"text"`
+	}{
+		Text: message,
 	}
-	fmt.Println(req)
-	return nil
+
+	body, err := json.Marshal(clownRequest)
+	if err != nil {
+		return "", err
+	}
+
+	agent := client.httpClient.Post(client.baseURL)
+	agent.Set("Content-Type", "application/json").Body(body)
+
+	statusCode, respBody, errs := agent.Bytes()
+	if len(errs) > 0 {
+		return "", errors.Join(errs...)
+	}
+
+	if statusCode != fiber.StatusCreated {
+		return "", fmt.Errorf("invalid response status code: %d", statusCode)
+	}
+
+	resp := struct {
+		Text string `json:"text"`
+	}{}
+
+	if err := json.Unmarshal(respBody, &resp); err != nil {
+		return "", err
+	}
+
+	return resp.Text, nil
 }
diff --git a/clients/aiClient_test.go b/clients/aiClient_test.go
new file mode 100644
index 0000000..34975bb
--- /dev/null
+++ b/clients/aiClient_test.go
@@ -0,0 +1,17 @@
+package clients
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test_AiClient(t *testing.T) {
+	aiClient := NewAiClient("http://35.206.94.7:80/api/engine/generate_question_by_answer/")
+	for i := 0; i < 10; i++ {
+		question, err := aiClient.SendAnswerer(false, "type", "Дартвейдер был хорошим.", "1234")
+		if err != nil {
+			t.Error(err)
+		}
+		fmt.Println(question)
+	}
+}
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..e69de29
diff --git a/service/service.go b/service/service.go
index f0a9cb5..3e964bd 100644
--- a/service/service.go
+++ b/service/service.go
@@ -384,12 +384,19 @@ func (s *Service) PutAnswersOnePiece(c *fiber.Ctx) error {
 				return c.Status(fiber.StatusNotFound).SendString("no questions found")
 			}
 
-			err = s.aiClient.SendAnswerer(final, question[0].Type, ans.Content, cs)
+			questionText, err := s.aiClient.SendAnswerer(final, question[0].Type, ans.Content, cs)
 			if err != nil {
 				return c.Status(fiber.StatusInternalServerError).SendString(fmt.Sprintf("can not send answer to ai, err: %s", err.Error()))
 			}
 
-			// todo тут надо создать вопрос по ответу от ai
+			_, err = s.dal.QuestionRepo.CreateQuestion(c.Context(), &model.Question{
+				QuizId:  quiz.Id,
+				Title:   questionText,
+				Session: cs,
+			})
+			if err != nil {
+				return c.Status(fiber.StatusInternalServerError).SendString(fmt.Sprintf("can not create question type ai, err: %s", err.Error()))
+			}
 		}
 
 		ans.DeviceType = deviceType