feat: integrate the output language to the system/user prompt

danielmiessler · Sep 19, 2024 · 26fccfe · 26fccfe
1 parent 2b79a05
commit 26fccfe
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/core/fabric.go b/core/fabric.go
@@ -257,6 +257,9 @@ func (o *Chat) BuildChatSession(raw bool) (ret *db.Session, err error) {
 	}
 
 	systemMessage := strings.TrimSpace(o.Context) + strings.TrimSpace(o.Pattern)
+	if o.Language != "" {
+		systemMessage = fmt.Sprintf("%s. Please use the language '%s' for the output.", systemMessage, o.Language)
+	}
 	userMessage := strings.TrimSpace(o.Message)
 
 	if raw {
@@ -274,10 +277,6 @@ func (o *Chat) BuildChatSession(raw bool) (ret *db.Session, err error) {
 		}
 	}
 
-	if o.Language != "" {
-		ret.Append(&common.Message{Role: "system", Content: "please use " + o.Language + " language"})
-	}
-
 	if ret.IsEmpty() {
 		ret = nil
 		err = fmt.Errorf("no session, pattern or user messages provided")