39 lines
869 B
Go
39 lines
869 B
Go
|
package llm
|
||
|
|
|
||
|
|
type RequestContext struct {
|
||
|
|
User UserContext
|
||
|
|
Chat ChatContext
|
||
|
|
}
|
||
|
|
|
||
|
|
type UserContext struct {
|
||
|
|
Username string
|
||
|
|
FirstName string
|
||
|
|
LastName string
|
||
|
|
IsPremium bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type ChatContext struct {
|
||
|
|
Title string
|
||
|
|
Description string
|
||
|
|
Type string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c RequestContext) Prompt() string {
|
||
|
|
prompt := "The chat you're in is called \"" + c.Chat.Title + "\". " +
|
||
|
|
"The type of chat is \"" + c.Chat.Type + "\". " +
|
||
|
|
"The chat description is \"" + c.Chat.Description + "\". "
|
||
|
|
if c.User.Username != "" {
|
||
|
|
prompt += "The user who wrote you has username \"@" + c.Chat.Description + "\". "
|
||
|
|
}
|
||
|
|
prompt += "Their first name is \"" + c.User.FirstName + "\". "
|
||
|
|
if c.User.LastName != "" {
|
||
|
|
prompt += "Their last name is \"" + c.User.LastName + "\". "
|
||
|
|
}
|
||
|
|
|
||
|
|
if c.User.IsPremium {
|
||
|
|
prompt += "They have Telegram Premium subscription. "
|
||
|
|
}
|
||
|
|
|
||
|
|
return prompt
|
||
|
|
}
|