diff --git a/.changeset/gorgeous-rocks-grow.md b/.changeset/gorgeous-rocks-grow.md new file mode 100644 index 0000000000000000000000000000000000000000..ca3eca5b4bba825e5b314134e04f19bdec032294 --- /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 1e8db1ab2d6c32ef6502907c77baa0e9aedc2053..67f1173b2dff982251ceab7826678cfd49ff33af 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(