From 555ebad47eb2b8f8c66e0706dd10fcc12174437e Mon Sep 17 00:00:00 2001 From: Josh Reini <60949774+joshreini1@users.noreply.github.com> Date: Thu, 16 Nov 2023 11:16:10 -0700 Subject: [PATCH] add seed parameter (#560) --- trulens_eval/trulens_eval/feedback/provider/openai.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/trulens_eval/trulens_eval/feedback/provider/openai.py b/trulens_eval/trulens_eval/feedback/provider/openai.py index 9a2e52cf8..50925bb84 100644 --- a/trulens_eval/trulens_eval/feedback/provider/openai.py +++ b/trulens_eval/trulens_eval/feedback/provider/openai.py @@ -64,6 +64,9 @@ def _create_chat_completion( if 'temperature' not in kwargs: kwargs['temperature'] = 0.0 + if 'seed' not in kwargs: + kwargs['seed'] = 123 + if prompt is not None: completion = self.endpoint.client.chat.completions.create(messages=[{ "role": "system",