summaryrefslogtreecommitdiff
path: root/llm.go
diff options
context:
space:
mode:
authorGrail Finder <wohilas@gmail.com>2025-08-20 22:45:41 +0300
committerGrail Finder <wohilas@gmail.com>2025-08-20 22:45:41 +0300
commiteee5e83d329fa9a19cf04ba290102ea650aca580 (patch)
tree79b20b29dd6db0f1c4c2ace8a1a3bbebd1378d94 /llm.go
parent6955a5bc3a89e982f0b05d1e8e6f109e3b79d9ca (diff)
Feat: bot to write for any char in chat (completion only)
Diffstat (limited to 'llm.go')
-rw-r--r--llm.go18
1 files changed, 15 insertions, 3 deletions
diff --git a/llm.go b/llm.go
index 05874a1..4cef914 100644
--- a/llm.go
+++ b/llm.go
@@ -92,7 +92,11 @@ func (lcp LlamaCPPeer) FormMsg(msg, role string, resume bool) (io.Reader, error)
prompt := strings.Join(messages, "\n")
// strings builder?
if !resume {
- botMsgStart := "\n" + cfg.AssistantRole + ":\n"
+ botPersona := cfg.AssistantRole
+ if cfg.WriteNextMsgAsCompletionAgent != "" {
+ botPersona = cfg.WriteNextMsgAsCompletionAgent
+ }
+ botMsgStart := "\n" + botPersona + ":\n"
prompt += botMsgStart
}
if cfg.ThinkUse && !cfg.ToolUse {
@@ -234,7 +238,11 @@ func (ds DeepSeekerCompletion) FormMsg(msg, role string, resume bool) (io.Reader
prompt := strings.Join(messages, "\n")
// strings builder?
if !resume {
- botMsgStart := "\n" + cfg.AssistantRole + ":\n"
+ botPersona := cfg.AssistantRole
+ if cfg.WriteNextMsgAsCompletionAgent != "" {
+ botPersona = cfg.WriteNextMsgAsCompletionAgent
+ }
+ botMsgStart := "\n" + botPersona + ":\n"
prompt += botMsgStart
}
if cfg.ThinkUse && !cfg.ToolUse {
@@ -376,7 +384,11 @@ func (or OpenRouterCompletion) FormMsg(msg, role string, resume bool) (io.Reader
prompt := strings.Join(messages, "\n")
// strings builder?
if !resume {
- botMsgStart := "\n" + cfg.AssistantRole + ":\n"
+ botPersona := cfg.AssistantRole
+ if cfg.WriteNextMsgAsCompletionAgent != "" {
+ botPersona = cfg.WriteNextMsgAsCompletionAgent
+ }
+ botMsgStart := "\n" + botPersona + ":\n"
prompt += botMsgStart
}
if cfg.ThinkUse && !cfg.ToolUse {