diff --git a/spring-ai-client-chat/src/main/java/org/springframework/ai/chat/client/advisor/SimpleLoggerAdvisor.java b/spring-ai-client-chat/src/main/java/org/springframework/ai/chat/client/advisor/SimpleLoggerAdvisor.java index c6fc0d794b8..816862125be 100644 --- a/spring-ai-client-chat/src/main/java/org/springframework/ai/chat/client/advisor/SimpleLoggerAdvisor.java +++ b/spring-ai-client-chat/src/main/java/org/springframework/ai/chat/client/advisor/SimpleLoggerAdvisor.java @@ -36,13 +36,13 @@ * A simple logger advisor that logs the request and response messages. * * @author Christian Tzolov + * @author Jonghoon Park */ public class SimpleLoggerAdvisor implements CallAroundAdvisor, StreamAroundAdvisor { - public static final Function DEFAULT_REQUEST_TO_STRING = request -> request.toString(); + public static final Function DEFAULT_REQUEST_TO_STRING = Record::toString; - public static final Function DEFAULT_RESPONSE_TO_STRING = response -> ModelOptionsUtils - .toJsonStringPrettyPrinter(response); + public static final Function DEFAULT_RESPONSE_TO_STRING = ModelOptionsUtils::toJsonStringPrettyPrinter; private static final Logger logger = LoggerFactory.getLogger(SimpleLoggerAdvisor.class); @@ -50,7 +50,7 @@ public class SimpleLoggerAdvisor implements CallAroundAdvisor, StreamAroundAdvis private final Function responseToString; - private int order; + private final int order; public SimpleLoggerAdvisor() { this(DEFAULT_REQUEST_TO_STRING, DEFAULT_RESPONSE_TO_STRING, 0); @@ -60,6 +60,11 @@ public SimpleLoggerAdvisor(int order) { this(DEFAULT_REQUEST_TO_STRING, DEFAULT_RESPONSE_TO_STRING, order); } + public SimpleLoggerAdvisor(Function requestToString, + Function responseToString) { + this(requestToString, responseToString, 0); + } + public SimpleLoggerAdvisor(Function requestToString, Function responseToString, int order) { this.requestToString = requestToString;