From e341f3365cb417455d427cabb9376bc627aa9d18 Mon Sep 17 00:00:00 2001 From: Zaki Ali Date: Fri, 18 Oct 2024 13:40:33 -0700 Subject: [PATCH] fix: update summarizer_exchange model to use default model from input exchange (#139) --- packages/exchange/src/exchange/moderators/summarizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/exchange/src/exchange/moderators/summarizer.py b/packages/exchange/src/exchange/moderators/summarizer.py index a7bb1b0f..7a90ed04 100644 --- a/packages/exchange/src/exchange/moderators/summarizer.py +++ b/packages/exchange/src/exchange/moderators/summarizer.py @@ -23,7 +23,7 @@ class ContextSummarizer(ContextTruncate): summarizer_exchange = exchange.replace( system=Message.load("summarizer.jinja").text, moderator=PassiveModerator(), - model=self.model, + model=self.model if self.model else exchange.model, messages=messages_to_summarize, checkpoint_data=CheckpointData(), )