Compare commits

..

3 Commits

55
main.go
View File

@@ -24,23 +24,32 @@ type OpenAIPrompter struct {
log *slog.Logger
}
func composeSysPromptWithContext(systemPrompt string, req PromptRequest) string {
var b strings.Builder
b.WriteString(systemPrompt)
b.WriteString("\nИмя пользователя: ")
b.WriteString(req.Username)
if req.ReplyToContent != nil {
b.WriteString("\nПользователь отсылается на текст сообщения: ")
b.WriteString(*req.ReplyToContent)
}
return b.String()
}
func (p *OpenAIPrompter) Prompt(ctx context.Context, req PromptRequest) (*HighlyTrustedResponse, error) {
p.log.Info("new prompt request",
"req", req)
sysPrompt := composeSysPromptWithContext(p.cfg.SystemPrompt, req)
input := []responses.ResponseInputItemUnionParam{
{
OfMessage: &responses.EasyInputMessageParam{
Content: responses.EasyInputMessageContentUnionParam{
OfString: openai.String(p.cfg.SystemPrompt),
},
Role: responses.EasyInputMessageRoleSystem,
},
},
{
OfMessage: &responses.EasyInputMessageParam{
Content: responses.EasyInputMessageContentUnionParam{
OfString: openai.String("Пользователя зовут " + req.Username),
OfString: openai.String(sysPrompt),
},
Role: responses.EasyInputMessageRoleSystem,
},
@@ -55,23 +64,7 @@ func (p *OpenAIPrompter) Prompt(ctx context.Context, req PromptRequest) (*Highly
},
}
if req.OriginalPostContent != nil {
message := "Пользователь может ссылаться на текст поста, который он комментирует. Вот он: " +
*req.OriginalPostContent
input = append(input, responses.ResponseInputItemUnionParam{
OfMessage: &responses.EasyInputMessageParam{
Content: responses.EasyInputMessageContentUnionParam{
OfString: openai.String(message),
},
Role: responses.EasyInputMessageRoleDeveloper,
Type: responses.EasyInputMessageTypeMessage,
},
})
}
resp, err := p.cli.Responses.New(ctx, responses.ResponseNewParams{
Instructions: openai.String(p.cfg.SystemPrompt),
Input: responses.ResponseNewParamsInputUnion{
OfInputItemList: input,
},
@@ -105,7 +98,7 @@ type HighlyTrustedResponse struct {
type PromptRequest struct {
Username string
Question string
OriginalPostContent *string
ReplyToContent *string
}
type Proompter interface {
@@ -146,10 +139,10 @@ func (a *App) handleMessage(ctx context.Context, msg *tgbotapi.Message) error {
"transformed_text", question,
)
var ogPostContent *string
if msg.ReplyToMessage != nil && msg.ReplyToMessage.SenderChat.ID == a.config.ChannelID {
a.log.Info("message was a reply to channel post")
ogPostContent = &msg.ReplyToMessage.Text
var repliedToContent *string
if msg.ReplyToMessage != nil {
a.log.Info("message was a reply")
repliedToContent = &msg.ReplyToMessage.Text
}
var username string
@@ -159,7 +152,7 @@ func (a *App) handleMessage(ctx context.Context, msg *tgbotapi.Message) error {
response, err := a.proompter.Prompt(ctx, PromptRequest{
Question: question,
OriginalPostContent: ogPostContent,
ReplyToContent: repliedToContent,
Username: username,
})
if err != nil {