From d903da626f9cd8f467f6fbfd9234a77fa4db52a4 Mon Sep 17 00:00:00 2001
From: yisding <yi.s.ding@gmail.com>
Date: Thu, 1 Feb 2024 14:12:38 -0800
Subject: [PATCH] Allow simple response builder prompt change (#505)
---
.changeset/gorgeous-rocks-grow.md | 5 +++++
packages/core/src/synthesizers/builders.ts | 6 +++---
2 files changed, 8 insertions(+), 3 deletions(-)
create mode 100644 .changeset/gorgeous-rocks-grow.md
diff --git a/.changeset/gorgeous-rocks-grow.md b/.changeset/gorgeous-rocks-grow.md
new file mode 100644
index 000000000..ca3eca5b4
--- /dev/null
+++ b/.changeset/gorgeous-rocks-grow.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+easier prompt customization for SimpleResponseBuilder
diff --git a/packages/core/src/synthesizers/builders.ts b/packages/core/src/synthesizers/builders.ts
index 1e8db1ab2..67f1173b2 100644
--- a/packages/core/src/synthesizers/builders.ts
+++ b/packages/core/src/synthesizers/builders.ts
@@ -33,11 +33,11 @@ enum ResponseMode {
*/
export class SimpleResponseBuilder implements ResponseBuilder {
llm: LLM;
- textQATemplate: SimplePrompt;
+ textQATemplate: TextQaPrompt;
- constructor(serviceContext: ServiceContext) {
+ constructor(serviceContext: ServiceContext, textQATemplate?: TextQaPrompt) {
this.llm = serviceContext.llm;
- this.textQATemplate = defaultTextQaPrompt;
+ this.textQATemplate = textQATemplate ?? defaultTextQaPrompt;
}
getResponse(
--
GitLab