Skip to content
Snippets Groups Projects
Unverified Commit d903da62 authored by yisding's avatar yisding Committed by GitHub
Browse files

Allow simple response builder prompt change (#505)

parent 177b4462
Branches
Tags
No related merge requests found
---
"llamaindex": patch
---
easier prompt customization for SimpleResponseBuilder
...@@ -33,11 +33,11 @@ enum ResponseMode { ...@@ -33,11 +33,11 @@ enum ResponseMode {
*/ */
export class SimpleResponseBuilder implements ResponseBuilder { export class SimpleResponseBuilder implements ResponseBuilder {
llm: LLM; llm: LLM;
textQATemplate: SimplePrompt; textQATemplate: TextQaPrompt;
constructor(serviceContext: ServiceContext) { constructor(serviceContext: ServiceContext, textQATemplate?: TextQaPrompt) {
this.llm = serviceContext.llm; this.llm = serviceContext.llm;
this.textQATemplate = defaultTextQaPrompt; this.textQATemplate = textQATemplate ?? defaultTextQaPrompt;
} }
getResponse( getResponse(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment