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);
}