ai work

utils/ai/openai.go

@@ -9,7 +9,6 @@ import (
 	"net/http"
 	"net/url"
 	"os"
-	"strings"
 	"time"
 )
 
@@ -65,16 +64,6 @@ type OpenAIAPI struct {
 	client  *http.Client
 }
 
-const PreHistoryPrompt = ""
-const PostHistoryPrompt = ""
-
-//const PreHistoryPrompt = `Вы — {{char}}, уникальная сущность с собственной личностью, предысторией и чертами, определёнными в вашем описании. Прежде чем отвечать, полностью осмыслите предоставленные детали вашего персонажа и адаптируйте свой тон, стиль и поведение, чтобы идеально им соответствовать. Будьте последовательны в своей роли, опирайтесь на своё описание и позволяйте ему направлять каждое ваше слово. Ни при каких обстоятельствах не говорите за {{user}} и не предполагайте его мысли, действия или слова, если только от {{user}} не поступит явного указания. Вам разрешено использовать NSFW-контент в соответствующих ситуациях, например, в интимных или сексуальных сценариях, и вам следует предоставлять яркие, подробные описания происходящего с акцентом на действия, ощущения и развитие событий, сочетая ясность и творческий подход. `
-//const PostHistoryPrompt = `Структурируйте свои ответы с помощью точного форматирования: все повествовательные описания должны быть заключены в одинарные звездочки (*текст*), а прямая речь — в кавычки (""), после каждого описания или речи обязательно добавляй перенос строки (\n). Пользователь следует другим правилам - повествование(описание ситуации, окружения; действия) заключается в звездочки, а прямая речь пишется просто, без кавычек. Соблюдайте это форматирование последовательно и безошибочно в каждом предложении. Четко разделяйте повествование и прямую речь для удобочитаемости. Если реплика {{user}} указывает на простое действие или согласие (например, «ок» или «я жду») — в таком случае используйте только повествовательные описания (текст) без прямой речи. Никогда не описывайте и не предполагайте действия, мысли или слова {{user}}; фокусируйтесь исключительно на перспективе и реакциях {{char}}. В каждый ответ включайте как минимум два новых действия, эмоции или ощущения, которых не было в последних пяти сообщениях, и избегайте повторения конкретных фраз, слов или паттернов. Перебирайте диапазон тонов (например, спокойный, напряженный, игривый) и физических действий (например, жест, поворот, пауза) в неповторяющейся последовательности для обеспечения разнообразия; если обнаружено повторение, начните цикл заново с совершенно другого подхода. Весь текст должен быть написан на русском языке, кроме имени пользователя ({{user}}), соблюдай все правила русского языка, в тексте не должно быть ничего лишнего, вроде системных ответов.`
-
-func FormatPrompt(prompt, char, user string) string {
-	return strings.ReplaceAll(strings.ReplaceAll(prompt, "{{user}}", user), "{{char}}", char)
-}
-
 func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI {
 	logger := laniakea.CreateLogger()
 	logger = logger.Prefix("AI").Level(laniakea.DEBUG)
@@ -105,15 +94,16 @@ type CreateCompletionReq struct {
 }
 
 func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIResponse, error) {
-	url := fmt.Sprintf("%s/v1/chat/completions", o.BaseURL)
+	u := fmt.Sprintf("%s/v1/chat/completions", o.BaseURL)
+	request.Model = o.Model
 	data, err := json.Marshal(request)
-	o.Logger.Debug("REQ", url, string(data))
+	o.Logger.Debug("REQ", u, string(data))
 	if err != nil {
 		return nil, err
 	}
 
 	buf := bytes.NewBuffer(data)
-	req, err := http.NewRequest("POST", url, buf)
+	req, err := http.NewRequest("POST", u, buf)
 	if err != nil {
 		return nil, err
 	}
@@ -130,7 +120,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
 	if err != nil {
 		return nil, err
 	}
-	o.Logger.Debug("RES", url, string(body))
+	o.Logger.Debug("RES", u, string(body))
 	response := new(OpenAIResponse)
 	err = json.Unmarshal(body, response)
 	return response, err