From 221da36663a92c13d244f7e5e7628f846905dc6e Mon Sep 17 00:00:00 2001 From: Arun Cherla Date: Fri, 13 Sep 2024 01:03:15 -0400 Subject: [PATCH] Update members.py fix final_answer_model to use the model configuration provided by user to resolve issues with configuration inconsistency between members & final model Signed-off-by: Arun Cherla --- backend/app/core/graph/members.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/backend/app/core/graph/members.py b/backend/app/core/graph/members.py index 1d1e5f5..ee99de3 100644 --- a/backend/app/core/graph/members.py +++ b/backend/app/core/graph/members.py @@ -174,9 +174,7 @@ def __init__( self.model = init_chat_model( model, model_provider=provider, temperature=0, streaming=True ) - self.final_answer_model = init_chat_model( - model, model_provider=provider, temperature=0, streaming=True - ) + self.final_answer_model = self.model def tag_with_name(self, ai_message: AIMessage, name: str) -> AIMessage: """Tag a name to the AI message"""