diff --git a/pom.xml b/pom.xml index 9404832..4516554 100755 --- a/pom.xml +++ b/pom.xml @@ -4,7 +4,7 @@ lm-lib de.extio lm-lib - 1.8.7 + 1.8.8 21 diff --git a/src/main/java/de/extio/lmlib/client/ClientService.java b/src/main/java/de/extio/lmlib/client/ClientService.java index 660184a..0456cc7 100755 --- a/src/main/java/de/extio/lmlib/client/ClientService.java +++ b/src/main/java/de/extio/lmlib/client/ClientService.java @@ -83,7 +83,7 @@ public class ClientService { } var completion = this.client.conversation(modelProfile, conversation); for (final var interceptor : this.completionInterceptors) { - completion = interceptor.after(modelProfile, completion); + completion = interceptor.after(modelProfile, conversation, completion); } return completion; } @@ -96,7 +96,7 @@ public class ClientService { } var completion = this.client.streamConversation(modelProfile, conversation, chunkConsumer); for (final var interceptor : this.completionInterceptors) { - completion = interceptor.after(modelProfile, completion); + completion = interceptor.after(modelProfile, conversation, completion); } return completion; } diff --git a/src/main/java/de/extio/lmlib/client/CompletionInterceptor.java b/src/main/java/de/extio/lmlib/client/CompletionInterceptor.java index 9e27d28..47034ab 100644 --- a/src/main/java/de/extio/lmlib/client/CompletionInterceptor.java +++ b/src/main/java/de/extio/lmlib/client/CompletionInterceptor.java @@ -6,6 +6,6 @@ public interface CompletionInterceptor { Conversation before(ModelProfile modelProfile, Conversation conversation); - Completion after(ModelProfile modelProfile, Completion completion); + Completion after(ModelProfile modelProfile, Conversation conversation, Completion completion); }