diff --git a/trulens_eval/trulens_eval/feedback/provider/base.py b/trulens_eval/trulens_eval/feedback/provider/base.py index 6dfda7ba7..da58c5fcb 100644 --- a/trulens_eval/trulens_eval/feedback/provider/base.py +++ b/trulens_eval/trulens_eval/feedback/provider/base.py @@ -57,6 +57,7 @@ def _create_chat_completion(self, prompt, *args, **kwargs): def _find_relevant_string(self, full_source, hypothesis): return self.endpoint.run_me( lambda: self._create_chat_completion( + model=self.model_engine, prompt = str.format( prompts.SYSTEM_FIND_SUPPORTING,