skobkin/telegram-ollama-reply-bot
skobkin/telegram-ollama-reply-bot
1
0
Fork 0
Code Issues 4 Pull requests 1 Releases Activity

Request context prompt fixes and improvements.
All checks were successful
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/pr Build is passing
Details

Browse source
This commit is contained in:
Alexey Skobkin 2024-03-12 06:31:07 +03:00
parent a1debf7908
commit 19da4c6b34
No known key found for this signature in database
GPG key ID: 5D5CEF6F221278E7
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
llm/request_context.go
View file

@ -19,17 +19,22 @@ type ChatContext struct {
} }
func (c RequestContext) Prompt() string { func (c RequestContext) Prompt() string {
prompt := "The chat you're in is called \"" + c.Chat.Title + "\". " + prompt := "The type of chat you're in is \"" + c.Chat.Type + "\". "
"The type of chat is \"" + c.Chat.Type + "\". " +
"The chat description is \"" + c.Chat.Description + "\". " if c.Chat.Title != "" {
prompt += "Chat is called \"" + c.Chat.Title + "\". "
}
if c.Chat.Description != "" {
prompt += "Chat description is \"" + c.Chat.Description + "\". "
}
if c.User.Username != "" { if c.User.Username != "" {
prompt += "The user who wrote you has username \"@" + c.Chat.Description + "\". " prompt += "The user who wrote you has username \"@" + c.User.Username + "\". "
} }
prompt += "Their first name is \"" + c.User.FirstName + "\". " prompt += "Their first name is \"" + c.User.FirstName + "\". "
if c.User.LastName != "" { if c.User.LastName != "" {
prompt += "Their last name is \"" + c.User.LastName + "\". " prompt += "Their last name is \"" + c.User.LastName + "\". "
} }
if c.User.IsPremium { if c.User.IsPremium {
prompt += "They have Telegram Premium subscription. " prompt += "They have Telegram Premium subscription. "
} }