Update members.py (#142)

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 <[email protected]>
StreetLamb · Sep 13, 2024 · b2a46db · b2a46db
1 parent 20d869c
commit b2a46db
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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"""