From e9ca8607e3aec48640d95e305b2cbdbd22176bec Mon Sep 17 00:00:00 2001 From: Zaki Ali Date: Thu, 10 Oct 2024 17:10:46 -0700 Subject: [PATCH] fix: update summarizer_exchange model to use default model from input exchange --- 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 7e2dd558..0634ab8b 100644 --- a/packages/exchange/src/exchange/moderators/summarizer.py +++ b/packages/exchange/src/exchange/moderators/summarizer.py @@ -25,7 +25,7 @@ def rewrite(self, exchange: Type["exchange.exchange.Exchange"]) -> None: # noqa 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(), )