Skip to content

Commit

Permalink
Merge pull request #83 from flystar32/main
Browse files Browse the repository at this point in the history
Fix #76
  • Loading branch information
chickenlj authored Oct 30, 2024
2 parents f772076 + a7ba8df commit b5a2054
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ public class DocumentRetrievalAdvisor implements CallAroundAdvisor, StreamAround

private static final int DEFAULT_ORDER = 0;

public static String RETRIEVED_DOCUMENTS = "qa_retrieved_documents";
public static String RETRIEVED_DOCUMENTS = "question_answer_context";

private final DocumentRetriever retriever;

Expand Down Expand Up @@ -145,9 +145,8 @@ private AdvisedRequest before(AdvisedRequest request) {
advisedUserParams.put("question_answer_context", documentContext);

return AdvisedRequest.from(request)
.withSystemText(this.userTextAdvise)
.withSystemParams(advisedUserParams)
.withUserText(request.userText())
.withUserText(request.userText() + System.lineSeparator() + this.userTextAdvise)
.withUserParams(advisedUserParams)
.withAdviseContext(context)
.build();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ public class DashScopeDocumentRetrievalAdvisor implements CallAroundAdvisor, Str

private static final int DEFAULT_ORDER = 0;

public static String RETRIEVED_DOCUMENTS = "qa_retrieved_documents";
public static String RETRIEVED_DOCUMENTS = "question_answer_context";

private final DocumentRetriever retriever;

Expand Down Expand Up @@ -184,9 +184,8 @@ private AdvisedRequest before(AdvisedRequest request) {
advisedUserParams.put(RETRIEVED_DOCUMENTS, documentContext);

return AdvisedRequest.from(request)
.withSystemText(this.userTextAdvise)
.withSystemParams(advisedUserParams)
.withUserText(request.userText())
.withUserText(request.userText() + System.lineSeparator() + this.userTextAdvise)
.withUserParams(advisedUserParams)
.withAdviseContext(context)
.build();
}
Expand Down

0 comments on commit b5a2054

Please sign in to comment.