Merge pull request #43 from basenana/fix/history

fix: get real history
basenana · Apr 22, 2024 · ff7ccfa · ff7ccfa
2 parents 9359a34 + 35f4d97
commit ff7ccfa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/friday/question.go b/pkg/friday/question.go
@@ -42,6 +42,10 @@ func (f *Friday) Question(q string) *Friday {
 	return f
 }
 
+func (f *Friday) GetRealHistory() []map[string]string {
+	return f.statement.history
+}
+
 func (f *Friday) Chat(res *ChatState) *Friday {
 	if len(f.statement.history) == 0 {
 		f.Error = errors.New("history can not be nil")
@@ -156,6 +160,7 @@ func (f *Friday) chat(res *ChatState) *Friday {
 	}
 
 	// go for llm
+	f.statement.history = dialogues // return realHistory
 	_, err := f.LLM.Chat(f.statement.context, true, dialogues, res.Response)
 	if err != nil {
 		f.Error = err