fix: qa pipeline missing initialization in WithPrompt (#150)

henomis · Nov 8, 2023 · bbad7ea · bbad7ea
1 parent c1a816a
commit bbad7ea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pipeline/qa/qa.go b/pipeline/qa/qa.go
@@ -91,7 +91,10 @@ func (q *QAPipeline) WithPrompt(chat *chat.Chat) *QAPipeline {
 	tube := pipeline.NewTube(llm)
 
 	return &QAPipeline{
-		pipeline: pipeline.New(tube),
+		llmEngine: q.llmEngine,
+		pipeline:  pipeline.New(tube),
+		index:     q.index,
+		mode:      q.mode,
 	}
 }